diff --git a/ad_interface_tornado.py b/ad_interface_tornado.py index ef2e328..4c4414a 100644 --- a/ad_interface_tornado.py +++ b/ad_interface_tornado.py @@ -154,12 +154,16 @@ class DispatchHandler(tornado.web.RequestHandler): # 如果取得的记录条数大于预设,扔掉多余的记录,当前采用的是随机选择,以后可能需要添加加权选择 id_list = [] if limit < len(ids): - for i in range(limit): - while 1: - new = my_redis.srandmember(key) - if new not in id_list: - id_list.append(new) - break + nums=limit + else: + nums=len(ids) + + for i in range(nums): + while 1: + new = my_redis.srandmember(key) + if new not in id_list: + id_list.append(new) + break for id in id_list: temp = my_redis.hgetall(f"ad::{id}::info") info.append(temp)