cobweb-launcher 1.2.51__py3-none-any.whl → 1.2.52__py3-none-any.whl

cobweb/db/redis_db.py

@@ -6,60 +6,82 @@ class RedisDB:
 
     def __init__(self, **kwargs):
         redis_config = kwargs or setting.REDIS_CONFIG
-        pool = redis.ConnectionPool(**redis_config)
-        self._client = redis.Redis(connection_pool=pool)
+        # pool = redis.ConnectionPool(**redis_config)
+        self.pool = redis.ConnectionPool(
+            max_connections=25,
+            **redis_config
+        )
+        
+    def get_connection(self):
+        return redis.StrictRedis(connection_pool=self.pool)
+        # self._client = redis.Redis(connection_pool=pool)
 
     def setnx(self, name, value=""):
-        return self._client.setnx(name, value)
+        with self.get_connection() as client:
+            client.setnx(name, value)
 
     def setex(self, name, t, value=""):
-        return self._client.setex(name, t, value)
+        with self.get_connection() as client:
+            client.setex(name, t, value)
 
     def expire(self, name, t, nx: bool = False, xx: bool = False, gt: bool = False, lt: bool = False):
-        return self._client.expire(name, t, nx, xx, gt, lt)
+        with self.get_connection() as client:
+            client.expire(name, t, nx, xx, gt, lt)
 
     def ttl(self, name):
-        return self._client.ttl(name)
+        with self.get_connection() as client:
+            return client.ttl(name)
 
     def delete(self, name):
-        return self._client.delete(name)
+        with self.get_connection() as client:
+            return client.delete(name)
 
     def exists(self, *name) -> bool:
-        return self._client.exists(*name)
+        with self.get_connection() as client:
+            return client.exists(*name)
 
     def sadd(self, name, value):
-        return self._client.sadd(name, value)
+        with self.get_connection() as client:
+            return client.sadd(name, value)
 
     def zcard(self, name) -> bool:
-        return self._client.zcard(name)
+        with self.get_connection() as client:
+            return client.zcard(name)
 
     def zadd(self, name, item: dict, **kwargs):
-        return self._client.zadd(name, item, **kwargs)
+        with self.get_connection() as client:
+            return client.zadd(name, item, **kwargs)
 
     def zrem(self, name, *value):
-        return self._client.zrem(name, *value)
+        with self.get_connection() as client:
+            return client.zrem(name, *value)
 
     def zcount(self, name, _min, _max):
-        return self._client.zcount(name, _min, _max)
+        with self.get_connection() as client:
+            return client.zcount(name, _min, _max)
 
     # def zrangebyscore(self, name, _min, _max, start, num, withscores: bool = False, *args):
-    #     return self._client.zrangebyscore(name, _min, _max, start, num, withscores, *args)
+    #     with self.get_connection() as client:
+    #        return client.zrangebyscore(name, _min, _max, start, num, withscores, *args)
 
     def lua(self, script: str, keys: list = None, args: list = None):
         keys = keys or []
         args = args or []
         keys_count = len(keys)
-        return self._client.eval(script, keys_count, *keys, *args)
+        with self.get_connection() as client:
+            return client.eval(script, keys_count, *keys, *args)
 
     def lua_sha(self, sha1: str, keys: list = None, args: list = None):
         keys = keys or []
         args = args or []
         keys_count = len(keys)
-        return self._client.evalsha(sha1, keys_count, *keys, *args)
+        with self.get_connection() as client:
+            return client.evalsha(sha1, keys_count, *keys, *args)
 
     def execute_lua(self, lua_script: str, keys: list, *args):
-        execute = self._client.register_script(lua_script)
-        return execute(keys=keys, args=args)
+        with self.get_connection() as client:
+            execute = client.register_script(lua_script)
+            return execute(keys=keys, args=args)
 
     def lock(self, key, t=15) -> bool:
         lua_script = """

{cobweb_launcher-1.2.51.dist-info → cobweb_launcher-1.2.52.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cobweb-launcher
-Version: 1.2.51
+Version: 1.2.52
 Summary: spider_hole
 Home-page: https://github.com/Juannie-PP/cobweb
 Author: Juannie-PP

{cobweb_launcher-1.2.51.dist-info → cobweb_launcher-1.2.52.dist-info}/RECORD

@@ -17,7 +17,7 @@ cobweb/crawlers/crawler.py,sha256=pePDGroD6JJAht5QTU51L7MkFIY4ob9TKpznZ2wUmsw,89
 cobweb/crawlers/file_crawler.py,sha256=2Sjbdgxzqd41WykKUQE3QQlGai3T8k-pmHNmPlTchjQ,4454
 cobweb/db/__init__.py,sha256=uZwSkd105EAwYo95oZQXAfofUKHVIAZZIPpNMy-hm2Q,56
 cobweb/db/api_db.py,sha256=bDc5dJQxq4z04h70KUTHd0OqUOEY7Cm3wcNJZtTvJIM,3015
-cobweb/db/redis_db.py,sha256=fumNZJiio-uQqRcSrymx8eJ1PqsdOwITe_Y-9JOXxrQ,4298
+cobweb/db/redis_db.py,sha256=mEWTBWSBXOhf3qxU9fdXYrhNJ44O8pS9tvqEAvZNhoI,5156
 cobweb/exceptions/__init__.py,sha256=E9SHnJBbhD7fOgPFMswqyOf8SKRDrI_i25L0bSpohvk,32
 cobweb/exceptions/oss_db_exception.py,sha256=iP_AImjNHT3-Iv49zCFQ3rdLnlvuHa3h2BXApgrOYpA,636
 cobweb/launchers/__init__.py,sha256=qMuVlQcjErVK67HyKFZEsXf_rfZD5ODjx1QucSCKMOM,114
@@ -37,8 +37,8 @@ cobweb/utils/bloom.py,sha256=vng-YbKgh9HbtpAWYf_nkUSbfVTOj40aqUUejRYlsCU,1752
 cobweb/utils/dotting.py,sha256=mVICaa26R-dQ4JGmPK-kkR6QjX38QiRewXZnGb2DCIc,1784
 cobweb/utils/oss.py,sha256=gyt8-UB07tVphZLQXMOf-JTJwU-mWq8KZkOXKkAf3uk,3513
 cobweb/utils/tools.py,sha256=5JEaaAwYoV9Sdla2UBIJn6faUBuXmxUMagm9ck6FVqs,1253
-cobweb_launcher-1.2.51.dist-info/LICENSE,sha256=z1rxSIGOyzcSb3orZxFPxzx-0C1vTocmswqBNxpKfEk,1063
-cobweb_launcher-1.2.51.dist-info/METADATA,sha256=xzLg7vLTsWurDo9L8jjmnlq-8B6UE1bT2itXPHt4a8o,6510
-cobweb_launcher-1.2.51.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
-cobweb_launcher-1.2.51.dist-info/top_level.txt,sha256=4GETBGNsKqiCUezmT-mJn7tjhcDlu7nLIV5gGgHBW4I,7
-cobweb_launcher-1.2.51.dist-info/RECORD,,
+cobweb_launcher-1.2.52.dist-info/LICENSE,sha256=z1rxSIGOyzcSb3orZxFPxzx-0C1vTocmswqBNxpKfEk,1063
+cobweb_launcher-1.2.52.dist-info/METADATA,sha256=lUniniJ1G9qDV9zA4XtI2p5bZPwDkMI6fKPSGnWjb_g,6510
+cobweb_launcher-1.2.52.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
+cobweb_launcher-1.2.52.dist-info/top_level.txt,sha256=4GETBGNsKqiCUezmT-mJn7tjhcDlu7nLIV5gGgHBW4I,7
+cobweb_launcher-1.2.52.dist-info/RECORD,,