From 02557f16b05fb733509e906b9a4224dcb2ca755b Mon Sep 17 00:00:00 2001 From: Martin Hostettler Date: Fri, 5 Aug 2016 21:51:01 +0200 Subject: [PATCH] RemoteRepository: Rename cache to chunkid_to_msgids. Also fetch_from_cache to pop_preload_msgid. --- src/borg/remote.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/src/borg/remote.py b/src/borg/remote.py index 029933672..824161f04 100644 --- a/src/borg/remote.py +++ b/src/borg/remote.py @@ -216,7 +216,7 @@ def __init__(self, location, create=False, exclusive=False, lock_wait=None, lock self.preload_ids = [] self.msgid = 0 self.to_send = b'' - self.cache = {} + self.chunkid_to_msgids = {} self.ignore_responses = set() self.responses = {} self.ratelimit = SleepingBandwidthLimiter(args.remote_ratelimit * 1024 if args and args.remote_ratelimit else 0) @@ -350,10 +350,10 @@ def call_many(self, cmd, calls, wait=True, is_preloaded=False): if not calls: return - def fetch_from_cache(args): - msgid = self.cache[args].pop(0) - if not self.cache[args]: - del self.cache[args] + def pop_preload_msgid(chunkid): + msgid = self.chunkid_to_msgids[chunkid].pop(0) + if not self.chunkid_to_msgids[chunkid]: + del self.chunkid_to_msgids[chunkid] return msgid def handle_error(error, res): @@ -425,12 +425,12 @@ def handle_error(error, res): if calls: if is_preloaded: assert cmd == "get", "is_preload is only supported for 'get'" - if calls[0][0] in self.cache: - waiting_for.append(fetch_from_cache(calls.pop(0)[0])) + if calls[0][0] in self.chunkid_to_msgids: + waiting_for.append(pop_preload_msgid(calls.pop(0)[0])) else: args = calls.pop(0) - if cmd == 'get' and args[0] in self.cache: - waiting_for.append(fetch_from_cache(args[0])) + if cmd == 'get' and args[0] in self.chunkid_to_msgids: + waiting_for.append(pop_preload_msgid(args[0])) else: self.msgid += 1 waiting_for.append(self.msgid) @@ -439,7 +439,7 @@ def handle_error(error, res): chunk_id = self.preload_ids.pop(0) args = (chunk_id,) self.msgid += 1 - self.cache.setdefault(chunk_id, []).append(self.msgid) + self.chunkid_to_msgids.setdefault(chunk_id, []).append(self.msgid) self.to_send = msgpack.packb((1, self.msgid, 'get', args)) if self.to_send: