comparison mercurial/sshserver.py @ 11585:5d907fbb9703

protocol: unify stream_out command
author Matt Mackall <mpm@selenic.com>
date Wed, 14 Jul 2010 16:19:27 -0500
parents 1af96b090116
children d054cc5c7737
comparison
equal deleted inserted replaced
11584:1af96b090116 11585:5d907fbb9703
63 d = changegroup.read(4096) 63 d = changegroup.read(4096)
64 if not d: 64 if not d:
65 break 65 break
66 self.fout.write(d) 66 self.fout.write(d)
67 67
68 self.fout.flush()
69
70 def sendstream(self, source):
71 for chunk in source:
72 self.fout.write(chunk)
68 self.fout.flush() 73 self.fout.flush()
69 74
70 def serve_forever(self): 75 def serve_forever(self):
71 try: 76 try:
72 while self.serve_one(): 77 while self.serve_one():
175 self.lock.release() 180 self.lock.release()
176 self.lock = None 181 self.lock = None
177 finally: 182 finally:
178 fp.close() 183 fp.close()
179 os.unlink(tempname) 184 os.unlink(tempname)
180
181 def do_stream_out(self):
182 try:
183 for chunk in streamclone.stream_out(self.repo):
184 self.fout.write(chunk)
185 self.fout.flush()
186 except streamclone.StreamException, inst:
187 self.fout.write(str(inst))
188 self.fout.flush()