单例+缓存优化

Administrator

2023-08-09 b1e51554a384917e1fdcf7e5011f2b99a0e80f99

 code_attribute/gpcode_manager.py

@@ -342,20 +342,56 @@
        RedisUtils.realse(redis_instance)


__gp_list_first_codes_cache = set()


# -------------------------------首板代码管理-------------------------------
class FirstGPCodesManager:
    __db = 0
    __redisManager = redis_manager.RedisManager(0)
    __gp_list_first_codes_cache = set()
    __instance = None

    def __new__(cls, *args, **kwargs):
        if not cls.__instance:
            cls.__instance = super(FirstGPCodesManager, cls).__new__(cls, *args, **kwargs)
            cls.__load_datas()
        return cls.__instance

    @classmethod
    def __get_redis(cls):
        return cls.__redisManager.getRedis()

    @classmethod
    def __load_datas(cls):
        __redis = cls.__get_redis()
        try:
            cls.__gp_list_first_codes_cache = RedisUtils.smembers(__redis, "gp_list_first")
        finally:
            RedisUtils.realse(__redis)

    @classmethod
    def __parse_codes_data(cls, code_datas):
        codes = []
        name_codes = {}
        for _data in code_datas:
            # 正常的股票
            if _data["sec_type"] == 1 and _data["sec_level"] == 1:
                code = _data["symbol"].split(".")[1]
                if code.find("30") != 0 and code.find("68") != 0:
                    name = _data["sec_name"]
                    codes.append(code)
                    # 保存代码对应的名称
                    name_codes[name] = code
        return codes, name_codes

# 添加首板代码
# code_datas 掘金返回的数据
def set_first_gp_codes_with_data(code_datas):
    redis_instance = __redisManager.getRedis()
    def set_first_gp_codes_with_data(self, code_datas):
        redis_instance = self.__get_redis()
    try:
        codes, name_codes = __parse_codes_data(code_datas)
            codes, name_codes = self.__parse_codes_data(code_datas)
        codes_set = set()
        for code in codes:
            codes_set.add(code)
        old_codes_set = __gp_list_first_codes_cache
            old_codes_set = self.__gp_list_first_codes_cache
        if old_codes_set is None:
            old_codes_set = set()
        del_set = old_codes_set - codes_set
@@ -367,9 +403,8 @@
        if add_codes or del_set:
            RedisUtils.expire(redis_instance, "gp_list_first", tool.get_expire(), auto_free=False)
        # 更新缓存
        __gp_list_first_codes_cache.clear()
        for code in codes_set:
            __gp_list_first_codes_cache.add(code)
            self.__gp_list_first_codes_cache.clear()
            self.__gp_list_first_codes_cache |= codes_set

        old_name_codes = CodesNameManager.list_first_code_name_dict()
        if old_name_codes is None:
@@ -381,25 +416,30 @@
    finally:
        RedisUtils.realse(redis_instance)


# 移除首板代码
def remove_first_gp_code(codes):
    redis_instance = __redisManager.getRedis()
    def remove_first_gp_code(self, codes):
        redis_instance = self.__get_redis()
    try:
        for code in codes:
                self.__gp_list_first_codes_cache.discard(code)
            RedisUtils.srem(redis_instance, "gp_list_first", code, auto_free=False)
    finally:
        RedisUtils.realse(redis_instance)


# 获取首板代码
def get_first_gp_codes():
    return RedisUtils.smembers(__redisManager.getRedis(), "gp_list_first")
    def get_first_gp_codes(self):
        return RedisUtils.smembers(self.__get_redis(), "gp_list_first")

    def get_first_gp_codes_cache(self):
        return self.__gp_list_first_codes_cache

# 是否在首板里面
def is_in_first_gp_codes(code):
    return RedisUtils.sismember(__redisManager.getRedis(), "gp_list_first", code)
    def is_in_first_gp_codes(self, code):
        return RedisUtils.sismember(self.__get_redis(), "gp_list_first", code)

    # 是否在首板里面
    def is_in_first_gp_codes_cache(self, code):
        return code in self.__gp_list_first_codes_cache


# 获取名称对应的代码
@@ -464,16 +504,16 @@

def rm_gp(code):
    RedisUtils.srem(__redisManager.getRedis(), "gp_list", code)
    remove_first_gp_code([code])
    FirstGPCodesManager().remove_first_gp_code([code])


def is_in_gp_pool(code):
    return RedisUtils.sismember(__redisManager.getRedis(), "gp_list", code) or is_in_first_gp_codes(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 = get_first_gp_codes()
    first_codes = FirstGPCodesManager().get_first_gp_codes()
    return set.union(codes, first_codes)