Skip to content

Commit

Permalink
remote-bzr: reuse bzrlib transports when possible
Browse files Browse the repository at this point in the history
Pass a list of open bzrlib.transport.Transport objects to each bzrlib
function that might create a transport.  This enables bzrlib to reuse
existing transports when possible, avoiding multiple concurrent
connections to the same remote server.

If the remote server is accessed via ssh, this fixes a couple of
problems:
  * If the user does not have keys loaded into an ssh agent, the user
    may be prompted for a password multiple times.
  * If the user is using OpenSSH and the ControlMaster setting is set
    to auto, git-remote-bzr might hang.  This is because bzrlib closes
    the multiple ssh sessions in an undefined order and might try to
    close the master ssh session before the other sessions.  The
    master ssh process will not exit until the other sessions have
    exited, causing a deadlock.  (The ssh sessions are closed in an
    undefined order because bzrlib relies on the Python garbage
    collector to trigger ssh session termination.)

Signed-off-by: Richard Hansen <rhansen@bbn.com>
Acked-by: Felipe Contreras <felipe.contreras@gmail.com>
Signed-off-by: Junio C Hamano <gitster@pobox.com>
  • Loading branch information
Richard Hansen authored and Junio C Hamano committed Sep 8, 2013
1 parent d2dbd39 commit bd5424f
Showing 1 changed file with 21 additions and 12 deletions.
33 changes: 21 additions & 12 deletions contrib/remote-helpers/git-remote-bzr
Original file line number Diff line number Diff line change
Expand Up @@ -674,7 +674,7 @@ def parse_reset(parser):
parsed_refs[ref] = mark_to_rev(from_mark)

def do_export(parser):
global parsed_refs, dirname
global parsed_refs, dirname, transports

parser.next()

Expand All @@ -699,7 +699,8 @@ def do_export(parser):
branch.generate_revision_history(revid, marks.get_tip(name))

if name in peers:
peer = bzrlib.branch.Branch.open(peers[name])
peer = bzrlib.branch.Branch.open(peers[name],
possible_transports=transports)
try:
peer.bzrdir.push_branch(branch, revision_id=revid)
except bzrlib.errors.DivergedBranches:
Expand Down Expand Up @@ -769,25 +770,28 @@ def do_list(parser):
print

def clone(path, remote_branch):
global transports
try:
bdir = bzrlib.bzrdir.BzrDir.create(path)
bdir = bzrlib.bzrdir.BzrDir.create(path, possible_transports=transports)
except bzrlib.errors.AlreadyControlDirError:
bdir = bzrlib.bzrdir.BzrDir.open(path)
bdir = bzrlib.bzrdir.BzrDir.open(path, possible_transports=transports)
repo = bdir.find_repository()
repo.fetch(remote_branch.repository)
return remote_branch.sprout(bdir, repository=repo)

def get_remote_branch(name):
global dirname, branches
global dirname, branches, transports

remote_branch = bzrlib.branch.Branch.open(branches[name])
remote_branch = bzrlib.branch.Branch.open(branches[name],
possible_transports=transports)
if isinstance(remote_branch.user_transport, bzrlib.transport.local.LocalTransport):
return remote_branch

branch_path = os.path.join(dirname, 'clone', name)

try:
branch = bzrlib.branch.Branch.open(branch_path)
branch = bzrlib.branch.Branch.open(branch_path,
possible_transports=transports)
except bzrlib.errors.NotBranchError:
# clone
branch = clone(branch_path, remote_branch)
Expand Down Expand Up @@ -821,17 +825,19 @@ def find_branches(repo):
yield name, branch.base

def get_repo(url, alias):
global dirname, peer, branches
global dirname, peer, branches, transports

normal_url = bzrlib.urlutils.normalize_url(url)
origin = bzrlib.bzrdir.BzrDir.open(url)
origin = bzrlib.bzrdir.BzrDir.open(url, possible_transports=transports)
is_local = isinstance(origin.transport, bzrlib.transport.local.LocalTransport)

shared_path = os.path.join(gitdir, 'bzr')
try:
shared_dir = bzrlib.bzrdir.BzrDir.open(shared_path)
shared_dir = bzrlib.bzrdir.BzrDir.open(shared_path,
possible_transports=transports)
except bzrlib.errors.NotBranchError:
shared_dir = bzrlib.bzrdir.BzrDir.create(shared_path)
shared_dir = bzrlib.bzrdir.BzrDir.create(shared_path,
possible_transports=transports)
try:
shared_repo = shared_dir.open_repository()
except bzrlib.errors.NoRepositoryPresent:
Expand All @@ -844,7 +850,8 @@ def get_repo(url, alias):
else:
# check and remove old organization
try:
bdir = bzrlib.bzrdir.BzrDir.open(clone_path)
bdir = bzrlib.bzrdir.BzrDir.open(clone_path,
possible_transports=transports)
bdir.destroy_repository()
except bzrlib.errors.NotBranchError:
pass
Expand Down Expand Up @@ -897,6 +904,7 @@ def main(args):
global files_cache
global is_tmp
global branches, peers
global transports

alias = args[1]
url = args[2]
Expand All @@ -909,6 +917,7 @@ def main(args):
marks = None
branches = {}
peers = {}
transports = []

if alias[5:] == url:
is_tmp = True
Expand Down

0 comments on commit bd5424f

Please sign in to comment.