RedisServer-Queue 0.0.1__py3-none-any.whl

RedisServer/RedisServer.py

@@ -0,0 +1,115 @@
+#!/user/bin/env python3
+# -*- coding: UTF-8 -*-
+# @Time : 2024/6/6 上午6:16
+# @Author : 龙翔
+# @File    :RedisServer.py
+# @Software: PyCharm
+import json
+import os
+import sys
+import time
+import uuid
+
+import redis
+
+# 将当前文件夹添加到环境变量
+if os.path.basename(__file__) in ['run.py', 'main.py', '__main__.py']:
+    if '.py' in __file__:
+        sys.path.append(os.path.abspath(os.path.dirname(__file__)))
+    else:
+        sys.path.append(os.path.abspath(__file__))
+
+r = redis.Redis(host='localhost', port=6379, db=0)
+
+
+class RedisQueue:
+    def __init__(self, topic=None):
+        self.topic = topic
+
+    def get(self):
+        # 弹出指定数量的数据
+        data = r.rpop(self.topic)
+        return data if data else None
+
+    def put(self, value):
+        if isinstance(value, list):
+            r.lpush(self.topic, *value)
+        else:
+            r.lpush(self.topic, *[value])
+
+    def clear(self):
+        r.delete(self.topic)
+
+    def size(self):
+        return r.llen(self.topic)
+
+    def qsize(self):
+        return self.size()
+
+    def get_mul(self, num):
+        return [self.get() for _ in range(num)]
+
+    def re_data(self):
+        ch_keys = r.keys(f"ack_{self.topic}_*")
+        for key in ch_keys:
+            data = r.get(key)
+            if data:
+                q = RedisQueue(self.topic)
+                t, _data = json.loads(data)
+                r.delete(key)
+                q.put(_data)
+        return len(ch_keys)
+
+    def get_all(self):
+        return r.lrange(self.topic, 0, -1)
+
+
+class RedisMQ:
+    def __init__(self):
+        self.switch = 1
+
+    def start_receive(self, topic, callback, count=-1):
+        while self.switch:
+            data = r.rpop(topic)
+            if data:
+                ch = RedisCh(topic, data)
+                callback(ch, data)
+                if count == 1:
+                    return
+                continue
+            ch_keys = r.keys(f"ack_{topic}_*")
+            for key in ch_keys:
+                data = r.get(key)
+                if data:
+                    q = RedisQueue(topic)
+                    t, _data = json.loads(data)
+                    if time.time() - t > 10 * 60:
+                        r.delete(key)
+                        q.put(_data)
+                        del q
+            if len(ch_keys) == 0:
+                time.sleep(10)
+            time.sleep(1)
+
+    def stop(self):
+        self.switch = 0
+
+
+class RedisCh:
+    def __init__(self, topic, data):
+        self.topic = topic
+        self.id = uuid.uuid4()
+        r.set(f"ack_{topic}_{self.id}", json.dumps([time.time(), data.decode()]))
+
+    def basic_ack(self):
+        r.delete(f"ack_{self.topic}_{self.id}")
+
+
+if __name__ == '__main__':
+    r.delete('test')
+    # a = RedisQueue("test")
+    # a.put(1)
+    # print(type(a.get()))
+    # r.set('test', 1)
+    # print(r.get('test'), type(r.get('test').decode('utf8')))
+    # RedisQueue.put(value=1, topic="test")

RedisServer_Queue-0.0.1.dist-info/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2018 The Python Packaging Authority
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

RedisServer_Queue-0.0.1.dist-info/METADATA

@@ -0,0 +1,14 @@
+Metadata-Version: 2.1
+Name: RedisServer-Queue
+Version: 0.0.1
+Summary: RedisServer-Queue is a simple redis server
+Author-email: NelsonLongXiang <1169207670@qq.com>
+Project-URL: Homepage, https://github.com/NelsonLongxiang/RedisServer
+Project-URL: Issues, https://github.com/NelsonLongxiang/RedisServer/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+

RedisServer_Queue-0.0.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+RedisServer/RedisServer.py,sha256=-RwVSFKx5sN7XCZcJpip8PeWCmoCmr_VHg5ApUW-1zI,3049
+RedisServer_Queue-0.0.1.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
+RedisServer_Queue-0.0.1.dist-info/METADATA,sha256=XqxOyPvjUtW5lkEWBHWn504leIrfSiIylAJvB6iY4eA,554
+RedisServer_Queue-0.0.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+RedisServer_Queue-0.0.1.dist-info/top_level.txt,sha256=KqY5fDKu1qjJQn9S9SA9hyn4IkuL07UHTsc45MmCB58,12
+RedisServer_Queue-0.0.1.dist-info/RECORD,,

RedisServer_Queue-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.43.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

RedisServer_Queue-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+RedisServer