Administrator
2023-10-30 fb47d36048e94b9a506d5c153e3dd19a01e37df1
code_attribute/gpcode_manager.py
@@ -2,6 +2,7 @@
股票代码管理器
"""
import json
import random
import time
import constant
@@ -13,9 +14,11 @@
from ths import l2_listen_pos_health_manager, client_manager
__redisManager = redis_manager.RedisManager(0)
__db = 0
class CodesNameManager:
    __db = 0
    redisManager = redis_manager.RedisManager(0)
    __gp_list_names_first_cache = []
@@ -83,7 +86,7 @@
    # 设置首板代码名称
    @classmethod
    def set_first_code_names(cls, datas):
        RedisUtils.set(cls.__get_redis(), "gp_list_names_first", json.dumps(datas))
        RedisUtils.set_async(cls.__db, "gp_list_names_first", json.dumps(datas))
    # 删除首板代码名称
    @classmethod
@@ -130,6 +133,7 @@
# 首板代码管理
class FirstCodeManager:
    __instance = None
    __db = 0
    def __new__(cls, *args, **kwargs):
        if not cls.__instance:
@@ -151,11 +155,11 @@
        hasChanged = False
        for code in codes:
            if code not in self.__first_code_record_cache:
                RedisUtils.sadd(self.__get_redis(), "first_code_record", code)
                RedisUtils.sadd_async(self.__db, "first_code_record", code)
                hasChanged = True
            self.__first_code_record_cache.add(code)
        if hasChanged:
            RedisUtils.expire(self.__get_redis(), "first_code_record", tool.get_expire())
            RedisUtils.expire_async(self.__db, "first_code_record", tool.get_expire())
    def is_in_first_record(self, code):
        if RedisUtils.sismember(self.__get_redis(), "first_code_record", code):
@@ -171,11 +175,11 @@
        hasChanged = False
        for code in codes:
            if code not in self.__first_code_limited_up_record_cache:
                RedisUtils.sadd(self.__get_redis(), "first_code_limited_up_record", code)
                RedisUtils.sadd_async(self.__db, "first_code_limited_up_record", code)
                hasChanged = True
            self.__first_code_limited_up_record_cache.add(code)
        if hasChanged:
            RedisUtils.expire(self.__get_redis(), "first_code_limited_up_record", tool.get_expire())
            RedisUtils.expire_async(self.__db, "first_code_limited_up_record", tool.get_expire())
    # 是否涨停过
    def is_limited_up(self, code):
@@ -397,11 +401,11 @@
            del_set = old_codes_set - codes_set
            add_codes = codes_set - old_codes_set
            for code in add_codes:
                RedisUtils.sadd(redis_instance, "gp_list_first", code, auto_free=False)
                RedisUtils.sadd_async(self.__db, "gp_list_first", code, auto_free=False)
            for code in del_set:
                RedisUtils.srem(redis_instance, "gp_list_first", code, auto_free=False)
                RedisUtils.srem_async(self.__db, "gp_list_first", code, auto_free=False)
            if add_codes or del_set:
                RedisUtils.expire(redis_instance, "gp_list_first", tool.get_expire(), auto_free=False)
                RedisUtils.expire_async(self.__db, "gp_list_first", tool.get_expire(), auto_free=False)
            # 更新缓存
            self.__gp_list_first_codes_cache.clear()
            self.__gp_list_first_codes_cache |= codes_set
@@ -508,12 +512,13 @@
def is_in_gp_pool(code):
    return RedisUtils.sismember(__redisManager.getRedis(), "gp_list", code) or FirstGPCodesManager().is_in_first_gp_codes_cache(code)
    return RedisUtils.sismember(__redisManager.getRedis(), "gp_list",
                                code) or FirstGPCodesManager().is_in_first_gp_codes_cache(code)
def get_gp_list():
    codes = RedisUtils.smembers(__redisManager.getRedis(), "gp_list")
    first_codes = FirstGPCodesManager().get_first_gp_codes()
    first_codes = FirstGPCodesManager().get_first_gp_codes_cache()
    return set.union(codes, first_codes)
@@ -583,6 +588,14 @@
    return limit_up_price
def get_limit_up_price_cache(code):
    if code in __limit_up_price_dict:
        return __limit_up_price_dict[code]
    return None
def get_limit_up_price_by_preprice(price):
    if price is None:
        return None
@@ -618,14 +631,12 @@
# datas:[(code,price)]
def set_prices(datas):
    pipe = __redisManager.getRedis().pipeline()
    for d in datas:
        code, price = d[0], d[1]
        if code in __current_price_cache and __current_price_cache[code] == price:
            continue
        __current_price_cache[code] = price
        RedisUtils.setex(pipe, "price-{}".format(code), tool.get_expire(), price)
    pipe.execute()
        RedisUtils.setex_async(__db, "price-{}".format(code), tool.get_expire(), price)
# 获取正在监听的代码