浏览代码

add video rank

linfan 1 年之前
父节点
当前提交
3e69b3e6fd
共有 1 个文件被更改,包括 6 次插入1 次删除
  1. 6 1
      video_rank.py

+ 6 - 1
video_rank.py

@@ -164,7 +164,12 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
         return []
 
     redisObj = RedisHelper()
-    video_scores = redisObj.get_batch_key(videoIds)
+    vidKeys = []
+    for vid in videoIds:
+        vidKeys.append("k_p:"+str(vid))
+    print("vidKeys:", vidKeys)
+    video_scores = redisObj.get_batch_key(vidKeys)
+    print(video_scores)
     video_items = []
     for i in range(len(video_scores)):
         try: