liqian 3 years ago
parent
commit
b5e30c7319
1 changed files with 10 additions and 7 deletions
  1. 10 7
      video_recall.py

+ 10 - 7
video_recall.py

@@ -133,10 +133,6 @@ class PoolRecall(object):
                 ge.join()
                 check_result, error_flag = ge.get()
                 #check_result, error_flag = self.check_video_counts(video_ids=filtered_result, flow_pool_mapping=video_mapping)
-                # 判断错误标记, True为错误
-                if error_flag:
-                    # 结束流量池召回
-                    break
 
                 for item in check_result:
                     if item[0] not in flow_pool_recall_videos:
@@ -154,6 +150,12 @@ class PoolRecall(object):
                 et_check = time.time()
                 log_.info('check result: result = {}, execute time = {}ms'.format(
                     check_result, (et_check - st_check) * 1000))
+
+                # 判断错误标记, True为错误
+                if error_flag:
+                    # 结束流量池召回
+                    break
+
             idx += get_size
 
         return flow_pool_recall_result[:size]
@@ -193,10 +195,11 @@ class PoolRecall(object):
 
         # 判断返回的错误标记,True为错误
         if error_flag:
+            return check_result, error_flag
             # 从流量召回池移除视频videos
-            for item in videos:
-                value = '{}-{}'.format(item['videoId'], item['flowPool'])
-                self.redis_helper.remove_value_from_zset(key_name=flow_pool_key, value=value)
+            # for item in videos:
+            #     value = '{}-{}'.format(item['videoId'], item['flowPool'])
+            #     self.redis_helper.remove_value_from_zset(key_name=flow_pool_key, value=value)
 
         redis_helper = RedisHelper()
         for item in view_count_result: