properly update inc/dec functions

pallets-eco · Jul 22, 2023 · 8141bfb · 8141bfb
1 parent 543cbd5
commit 8141bfb
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/cachelib/memcached.py b/src/cachelib/memcached.py
@@ -202,12 +202,20 @@ def clear(self) -> bool:
 
  def inc(self, key: str, delta: int = 1) -> _t.Optional[int]:
  key = self._normalize_key(key)
- value = (self._client.get(key) or 0) + delta
+ if self.pylibmc_used:
+ with self._client.reserve(block=self.blocking) as mc:
+ value = (mc.get(key) or 0) + delta
+ else:
+ value = (self._client.get(key) or 0) + delta
  return value if self.set(key, value) else None
 
  def dec(self, key: str, delta: int = 1) -> _t.Optional[int]:
  key = self._normalize_key(key)
- value = (self._client.get(key) or 0) - delta
+ if self.pylibmc_used:
+ with self._client.reserve(block=self.blocking) as mc:
+ value = (mc.get(key) or 0) - delta
+ else:
+ value = (self._client.get(key) or 0) - delta
  return value if self.set(key, value) else None
 
  def import_preferred_memcache_lib(self, servers: _t.Any, threads: int) -> _t.Any: