liqian 1 年之前
父节点
当前提交
5336cc20fb
共有 1 个文件被更改,包括 6 次插入1 次删除
  1. 6 1
      pool_predict.py

+ 6 - 1
pool_predict.py

@@ -145,10 +145,16 @@ def predict(app_type, video_info_list, flow_pool_id_list):
     :return: None
     """
     try:
+        redis_helper = RedisHelper()
         # 从流量池获取数据
         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)
             return video_info_list
         # video_id 与 flow_pool 进行mapping
         video_ids = set()
@@ -219,7 +225,6 @@ def predict(app_type, video_info_list, flow_pool_id_list):
                 video_info_data['flow_pool'][video_id] = flow_pool_list
 
         # 快速曝光流量池视频写入redis
-        redis_helper = RedisHelper()
         quick_flow_pool_key_name = f"{config_.QUICK_FLOWPOOL_KEY_NAME_PREFIX}{app_type}:{config_.QUICK_FLOW_POOL_ID}"
         # 如果key已存在,删除key
         if redis_helper.key_exists(quick_flow_pool_key_name):