牧安云哨-后端

bbit_ai/app/config/redis.py

@@ -0,0 +1,24 @@
+import redis
+
+
+class RedisClient:
+
+    def __init__(self, config_path="config.yaml"):
+        self.redis = redis.Redis(
+            "10.10.12.101",
+            6379,
+            0,
+            decode_responses=True,
+        )
+
+    def set_online(self, device_id: str):
+        key = f"device:online:{device_id}"
+        self.redis.set(key, 1)
+
+    def set_offline(self, device_id: str):
+        key = f"device:online:{device_id}"
+        self.redis.delete(key)
+
+    def is_device_online(self, device_id: str) -> bool:
+        key = f"device:online:{device_id}"
+        return self.redis.exists(key) == 1