update

pool_predict.py

@@ -146,15 +146,16 @@ def predict(app_type, video_info_list, flow_pool_id_list):
     """
     try:
         redis_helper = RedisHelper()
+        flow_pool_key_name = f"{config_.FLOWPOOL_KEY_NAME_PREFIX}{app_type}"
+        quick_flow_pool_key_name = f"{config_.QUICK_FLOWPOOL_KEY_NAME_PREFIX}{app_type}:{config_.QUICK_FLOW_POOL_ID}"
         # 从流量池获取数据
         videos = get_videos_from_flow_pool(app_type=app_type)
         if len(videos) <= 0:
             log_.info('流量池中无需分发的视频')
             # 清空数据
-            flow_pool_key_name = f"{config_.FLOWPOOL_KEY_NAME_PREFIX}{app_type}"
-            # 如果key已存在，删除key
-            if redis_helper.key_exists(flow_pool_key_name):
-                redis_helper.del_keys(flow_pool_key_name)
+            for key in [flow_pool_key_name, quick_flow_pool_key_name]:
+                if redis_helper.key_exists(key):
+                    redis_helper.del_keys(key)
             return video_info_list
         # video_id 与 flow_pool 进行mapping
         video_ids = set()
@@ -186,6 +187,10 @@ def predict(app_type, video_info_list, flow_pool_id_list):
 
         if not filtered_videos:
             log_.info('流量池中视频状态不符合分发')
+            # 清空数据
+            for key in [flow_pool_key_name, quick_flow_pool_key_name]:
+                if redis_helper.key_exists(key):
+                    redis_helper.del_keys(key)
             return video_info_list
 
         # 预测