fix: instance register

Signed-off-by: binaryYuki <60097976+binaryYuki@users.noreply.github.com>
binaryYuki · Oct 1, 2024 · 1c5a7b3 · 1c5a7b3
1 parent 5acf4a2
commit 1c5a7b3
Showing 1 changed file with 22 additions and 9 deletions.
diff --git a/app.py b/app.py
@@ -37,32 +37,45 @@
 
 
 async def registerInstance():
+    """
+    注册实例
+    :return:
+    """
     redis_connection = redis.from_url(
         f"redis://default:{os.getenv('REDIS_PASSWORD', '')}@{os.getenv('REDIS_HOST', 'localhost')}:{os.getenv('REDIS_PORT', 6379)}")
     try:
         f = await redis_connection.get("InstanceRegister")
-        if not f:
+        if f:
+            f = f.decode('utf-8')
+            f = json.loads(f)  # Assume JSON format
+        else:
             f = []
-        f = list(f)
+
         if instanceID not in f:
             f.append(instanceID)
-            f = str(f)
-            await redis_connection.set("InstanceRegister", f)
+            await redis_connection.set("InstanceRegister", json.dumps(f))
+        else:
+            print(f)
     except Exception as e:
         logger.error(f"Failed to register instance: {e}", exc_info=True)
         exit(-1)
 
 
 async def unregisterInstance():
+    """
+    注销实例
+    :return:
+    """
     redis_connection = redis.from_url(
         f"redis://default:{os.getenv('REDIS_PASSWORD', '')}@{os.getenv('REDIS_HOST', 'localhost')}:{os.getenv('REDIS_PORT', 6379)}")
     try:
         f = await redis_connection.get("InstanceRegister")
-        f = list(f)
-        if instanceID in f:
-            f.remove(instanceID)
-            f = str(f)
-            await redis_connection.set("InstanceRegister", f)
+        if f:
+            f = f.decode('utf-8')
+            f = json.loads(f)  # Assume JSON format
+            if instanceID in f:
+                f.remove(instanceID)
+                await redis_connection.set("InstanceRegister", json.dumps(f))
     except Exception as e:
         logger.error(f"Failed to unregister instance: {e}", exc_info=True)
         exit(-1)