Forráskód Böngészése

update 不同的redis队列,名字也相应的修改

lierqiang 2 éve
szülő
commit
9a93a9f400
1 módosított fájl, 5 hozzáadás és 5 törlés
  1. 5 5
      common/scheduling_db.py

+ 5 - 5
common/scheduling_db.py

@@ -126,19 +126,19 @@ class RedisHelper:
         return redis_conn
 
     @classmethod
-    def redis_push(cls, env, machine, data):
+    def redis_push(cls, env, task_key, data):
         redis_conn = cls.connect_redis(env)
         # print("开始写入数据")
-        redis_conn.lpush(machine, data)
+        redis_conn.lpush(task_key, data)
         # print("数据写入完成")
 
     @classmethod
-    def redis_pop(cls, env, machine):
+    def redis_pop(cls, env, task_key):
         redis_conn = cls.connect_redis(env)
-        if redis_conn.llen(machine) == 0:
+        if redis_conn.llen(task_key) == 0:
             return None
         else:
-            return redis_conn.rpop(machine)
+            return redis_conn.rpop(task_key)
 
 
 if __name__ == "__main__":