From a933bf8e8e7f0c5a5131a95ed5d0f0b58bf9ac34 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 22 八月 2023 14:29:35 +0800
Subject: [PATCH] 添加托管交易通道测试

---
 third_data/code_plate_key_manager.py |   52 +++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 37 insertions(+), 15 deletions(-)

diff --git a/third_data/code_plate_key_manager.py b/third_data/code_plate_key_manager.py
index 05b72dc..74497b6 100644
--- a/third_data/code_plate_key_manager.py
+++ b/third_data/code_plate_key_manager.py
@@ -21,34 +21,54 @@
 class KPLCodeJXBlockManager:
     __redisManager = redis_manager.RedisManager(3)
     __code_blocks = {}
+    # 澶囩敤
+    __code_by_blocks = {}
 
     def __get_redis(self):
         return self.__redisManager.getRedis()
 
-    def save_jx_blocks(self, code, blocks):
+    def save_jx_blocks(self, code, blocks, by=False):
         if blocks is None:
             return
-        if len(blocks) >2:
+        if len(blocks) > 2:
             blocks = blocks[:2]
 
         # 淇濆瓨鍓�2鏉℃暟鎹�
-        RedisUtils.setex(self.__get_redis(), f"kpl_jx_blocks-{code}", tool.get_expire(), json.dumps(blocks))
-        self.__code_blocks[code] = blocks
+        if by:
+            RedisUtils.setex(self.__get_redis(), f"kpl_jx_blocks_by-{code}", tool.get_expire(), json.dumps(blocks))
+            self.__code_by_blocks[code] = blocks
+        else:
+            RedisUtils.setex(self.__get_redis(), f"kpl_jx_blocks-{code}", tool.get_expire(), json.dumps(blocks))
+            self.__code_blocks[code] = blocks
 
     # 鑾峰彇绮鹃�夋澘鍧�
-    def get_jx_blocks(self, code):
-        if code in self.__code_blocks:
-            return self.__code_blocks[code]
-        val = RedisUtils.get(self.__get_redis(), f"kpl_jx_blocks-{code}")
-        if val is None:
-            return None
+    def get_jx_blocks(self, code, by=False):
+        if by:
+            if code in self.__code_by_blocks:
+                return self.__code_by_blocks[code]
+            val = RedisUtils.get(self.__get_redis(), f"kpl_jx_blocks_by-{code}")
+            if val is None:
+                return None
+            else:
+                val = json.loads(val)
+                self.__code_by_blocks[code] = val
+            return self.__code_by_blocks[code]
         else:
-            val = json.loads(val)
-            self.__code_blocks[code] = val
-        return self.__code_blocks[code]
+            if code in self.__code_blocks:
+                return self.__code_blocks[code]
+            val = RedisUtils.get(self.__get_redis(), f"kpl_jx_blocks-{code}")
+            if val is None:
+                return None
+            else:
+                val = json.loads(val)
+                self.__code_blocks[code] = val
+            return self.__code_blocks[code]
 
-    def get_jx_blocks_cache(self, code):
-        return self.__code_blocks.get(code)
+    def get_jx_blocks_cache(self, code, by=False):
+        if by:
+            return self.__code_by_blocks.get(code)
+        else:
+            return self.__code_blocks.get(code)
 
 
 # 寮�鐩樺暒绂佹浜ゆ槗鏉垮潡绠$悊
@@ -367,6 +387,8 @@
 
         k4 = set()
         jingxuan_blocks = self.__KPLCodeJXBlockManager.get_jx_blocks_cache(code)
+        if not jingxuan_blocks:
+            jingxuan_blocks = self.__KPLCodeJXBlockManager.get_jx_blocks_cache(code, by=True)
         if jingxuan_blocks:
             jingxuan_blocks = jingxuan_blocks[:2]
             k4 |= set([x[1] for x in jingxuan_blocks])

--
Gitblit v1.8.0