Improve Proxy Perf

QuangTung97 · Dec 1, 2023 · 7e31456 · 7e31456
1 parent 156f583
commit 7e31456
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/memproxy/proxy/proxy.py b/memproxy/proxy/proxy.py
@@ -154,6 +154,10 @@ def result(self) -> LeaseGetResponse:
             conf.execute()
 
         resp = self.resp
+
+        if len(_P) < 4096:
+            _P.append(self)
+
         return resp
 
 
@@ -208,6 +212,9 @@ def return_func(self) -> DeleteResponse:
         return self.resp
 
 
+_P: List[_LeaseGetState] = []
+
+
 class ProxyPipeline:
     __slots__ = '_conf'
 
@@ -217,7 +224,10 @@ def __init__(self, conf: _ClientConfig, sess: Optional[Session]):
         self._conf = _PipelineConfig(conf=conf, sess=sess)
 
     def lease_get(self, key: str) -> LeaseGetResult:
-        state = _LeaseGetState()
+        try:
+            state = _P.pop()
+        except IndexError:
+            state = _LeaseGetState()
 
         # do init get state
         conf = self._conf