From 1e607e90c08fa2b7a765360665cc392539e3ddec Mon Sep 17 00:00:00 2001 From: Administrator <admin@example.com> Date: 星期五, 04 八月 2023 13:20:55 +0800 Subject: [PATCH] redis连接池归还 --- third_data/code_plate_key_manager.py | 40 +++++++++++++++++++++------------------- 1 files changed, 21 insertions(+), 19 deletions(-) diff --git a/third_data/code_plate_key_manager.py b/third_data/code_plate_key_manager.py index a34dc57..80dbcb7 100644 --- a/third_data/code_plate_key_manager.py +++ b/third_data/code_plate_key_manager.py @@ -29,7 +29,7 @@ if blocks is None: return # 淇濆瓨鍓�2鏉℃暟鎹� - RedisUtils.setex(self.__get_redis(),f"kpl_jx_blocks-{code}", tool.get_expire(), json.dumps(blocks)) + RedisUtils.setex(self.__get_redis(), f"kpl_jx_blocks-{code}", tool.get_expire(), json.dumps(blocks)) self.__code_blocks[code] = blocks # 鑾峰彇绮鹃�夋澘鍧� @@ -92,7 +92,7 @@ # 浠婃棩娑ㄥ仠鍘熷洜鍙樺寲 def set_today_limit_up_reason_change(self, code, from_reason, to_reason): RedisUtils.sadd(self.__get_redis(), f"kpl_limit_up_reason_his-{code}", from_reason) - self.__get_redis().expire(f"kpl_limit_up_reason_his-{code}", tool.get_expire()) + RedisUtils.expire(self.__get_redis(), f"kpl_limit_up_reason_his-{code}", tool.get_expire()) self.__set_total_keys(code) # 璁剧疆浠g爜鐨勪粖鏃ユ定鍋滃師鍥� @@ -259,7 +259,8 @@ def set_history_limit_up_reason(self, code, reasons): self.__history_limit_up_reason_dict[code] = set(reasons) - RedisUtils.setex(self.__get_redis(),f"kpl_his_limit_up_reason-{code}", tool.get_expire(), json.dumps(list(reasons))) + RedisUtils.setex(self.__get_redis(), f"kpl_his_limit_up_reason-{code}", tool.get_expire(), + json.dumps(list(reasons))) logger_kpl_debug.debug(f"璁剧疆鍘嗗彶娑ㄥ仠鍘熷洜锛歿code}-{reasons}") # 濡傛灉杩斿洖鍊间笉涓篘one琛ㄧず宸茬粡鍔犺浇杩囧巻鍙插師鍥犱簡 @@ -267,7 +268,7 @@ reasons = self.__history_limit_up_reason_dict.get(code) if reasons is None: # 浠庡唴瀛樹腑鍔犺浇 - val = self.__get_redis().get(f"kpl_his_limit_up_reason-{code}") + val = RedisUtils.get(self.__get_redis(), f"kpl_his_limit_up_reason-{code}") if val is not None: val = set(json.loads(val)) self.__history_limit_up_reason_dict[code] = val @@ -280,13 +281,13 @@ def set_blocks(self, code, blocks): self.__blocks_dict[code] = set(blocks) - RedisUtils.setex(self.__get_redis(),f"kpl_blocks-{code}", tool.get_expire(), json.dumps(list(blocks))) + RedisUtils.setex(self.__get_redis(), f"kpl_blocks-{code}", tool.get_expire(), json.dumps(list(blocks))) def get_blocks(self, code): reasons = self.__blocks_dict.get(code) if reasons is None: # 浠庡唴瀛樹腑鍔犺浇 - val = self.__get_redis().get(f"kpl_blocks-{code}") + val = RedisUtils.get(self.__get_redis(), f"kpl_blocks-{code}") if val is not None: val = set(json.loads(val)) self.__blocks_dict[code] = val @@ -397,7 +398,8 @@ for k in k2: if k not in constant.KPL_INVALID_BLOCKS: keys.add(k) - if not keys: + # 濮嬬粓鑾峰彇绮鹃�夋澘鍧� + if True: # 鑾峰彇 jx_blocks = cls.__KPLCodeJXBlockManager.get_jx_blocks(code) if jx_blocks: @@ -406,7 +408,7 @@ # 娑ㄥ仠鍒楄〃涓尮閰嶅叧閿瘝锛岃繑鍥烇紙鏉垮潡:浠g爜闆嗗悎锛夛紝浠g爜闆嗗悎涓凡缁忔帓闄よ嚜韬� if not keys: - return cls.BLOCK_TYPE_NONE, None, "灏氭湭鎵惧埌娑ㄥ仠鍘熷洜" + return None, "灏氭湭鎵惧埌娑ㄥ仠鍘熷洜" code_limit_up_reason_dict = {} load_code_block() msg_list = [] @@ -421,11 +423,11 @@ msg_list.append(f"\n瀹炴椂top10: {top_4_current}(娑ㄥ仠鏁伴噺锛歿len(current_limit_up_datas)})") msg_list.append(f"鍘嗗彶top20: {top_8_record}") - # 鑾峰彇涓绘澘瀹炴椂韬綅 + # 鑾峰彇涓绘澘瀹炴椂韬綅,鍓旈櫎楂樹綅鏉� current_shsz_rank = kpl_block_util.get_code_current_rank(code, block, current_limit_up_datas, - code_limit_up_reason_dict, shsz=True) + code_limit_up_reason_dict, yesterday_current_limit_up_codes, shsz=True) record_shsz_rank = kpl_block_util.get_code_record_rank(code, block, limit_up_record_datas, - code_limit_up_reason_dict, shsz=True) + code_limit_up_reason_dict, yesterday_current_limit_up_codes, shsz=True) # 鑾峰彇涓绘澘鍘嗗彶韬綅 if is_top_4: pen_limit_up_codes = kpl_block_util.get_shsz_open_limit_up_codes(code, block, limit_up_record_datas, @@ -436,7 +438,7 @@ pen_limit_up_codes) + 1: # 灞炰簬榫�1,榫�2 can_buy_blocks.append((block, - f"{block}锛歵op4娑ㄥ仠鏉垮潡锛屼富鏉垮紑1({pen_limit_up_codes}),灞炰簬涓绘澘鍓嶉緳{len(pen_limit_up_codes) + 1}(瀹炴椂韬綅-{current_shsz_rank})")) + f"{block}锛歵op10娑ㄥ仠鏉垮潡锛屼富鏉垮紑1({pen_limit_up_codes}),灞炰簬涓绘澘鍓嶉緳{len(pen_limit_up_codes) + 1}(瀹炴椂韬綅-{current_shsz_rank})")) continue else: msg_list.append( @@ -467,7 +469,7 @@ f"鏉垮潡-{block}: 涓嶆槸top4娑ㄥ仠鏉垮潡锛屾弧瓒崇簿閫�/琛屼笟娴佸叆瑕佹眰,涓嶄负涓绘澘榫�1锛堝疄鏃惰韩浣�-{current_shsz_rank},鍘嗗彶韬綅-{record_shsz_rank}锛�") continue if len(can_buy_blocks) == len(keys): - blocks = "/".join([x[0] for x in can_buy_blocks]) + blocks = [x[0] for x in can_buy_blocks] blocks_msg = "\n".join([x[1] for x in can_buy_blocks]) return blocks, blocks_msg @@ -481,10 +483,10 @@ if constant.TEST: return True, "", cls.BLOCK_TYPE_NONE - block, block_msg = cls.get_can_buy_block(code, current_limit_up_datas, - limit_up_record_datas, yesterday_current_limit_up_codes, - before_blocks_dict) - if block is None: + blocks, block_msg = cls.get_can_buy_block(code, current_limit_up_datas, + limit_up_record_datas, yesterday_current_limit_up_codes, + before_blocks_dict) + if not blocks: return False, block_msg # ---------------------------------鍒ゆ柇鐩爣浠g爜鐨勬澘鍧�-------------------start------------ @@ -565,11 +567,11 @@ # ---------------------------------鍔犺浇宸茬粡涓嬪崟/鎴愪氦鐨勪唬鐮佷俊鎭�------------end------------- msg_list = [] - for key in [block]: + for key in blocks: # 鏉垮潡涓凡缁忔湁鎴愪氦鐨勫氨涓嶄笅鍗曚簡 if key in trade_success_blocks_count: success_codes_count = len(trade_success_blocks_count[key]) - if success_codes_count >= 2: + if success_codes_count >= 1: msg_list.append(f"銆恵key}銆戜腑宸茬粡鏈墈success_codes_count}涓垚浜や唬鐮�") log.logger_kpl_debug.debug(f"{code}锛氭澘鍧楋紙{key}锛夊凡缁忔湁鎴愪氦銆恵trade_success_blocks_count[key]}銆�") continue -- Gitblit v1.8.0