From: Kevan Date: Sun, 15 Jan 2012 02:50:37 +0000 (-0800) Subject: Add _record_allocated_shares, which records the results of share X-Git-Url: https://git.rkrishnan.org/%5B/%5D%20/uri/provisioning?a=commitdiff_plain;h=640e42845b4d3d34aa6b49858a40e945d863f818;p=tahoe-lafs%2Ftahoe-lafs.git Add _record_allocated_shares, which records the results of share allocations. --- diff --git a/src/allmydata/immutable/upload.py b/src/allmydata/immutable/upload.py index 32a2d7be..82441002 100644 --- a/src/allmydata/immutable/upload.py +++ b/src/allmydata/immutable/upload.py @@ -538,6 +538,11 @@ class Tahoe2ServerSelector(log.PrefixingLogMixin): self.preexisting_shares.setdefault(share, set()).add(serverid) self.homeless_shares.discard(share) + def _record_allocated_shares(self, tracker, shares): + # the ServerTracker will remember which shares were allocated on + # that peer. We just have to remember to use them. + self.use_trackers.add(tracker) + def _got_response(self, res, tracker, shares_to_ask, put_tracker_here): if isinstance(res, failure.Failure): # This is unusual, and probably indicates a bug or a network @@ -573,11 +578,9 @@ class Tahoe2ServerSelector(log.PrefixingLogMixin): progress = True self._record_existing_shares(tracker, alreadygot) - # the ServerTracker will remember which shares were allocated on - # that peer. We just have to remember to use them. if allocated: - self.use_trackers.add(tracker) progress = True + self._record_allocated_shares(tracker, allocated) if allocated or alreadygot: self.serverids_with_shares.add(tracker.get_serverid())