redis上级缓存添加

Administrator

2023-08-03 90886c62a62ecbd64c9b229b48e4ef2cf600ae43

 code_attribute/gpcode_manager.py

@@ -170,7 +170,7 @@
    def add_limited_up_record(self, codes):
        hasChanged = False
        for code in codes:
            if code not in  self.__first_code_limited_up_record_cache:
            if code not in self.__first_code_limited_up_record_cache:
                RedisUtils.sadd(self.__get_redis(), "first_code_limited_up_record", code)
                hasChanged = True
            self.__first_code_limited_up_record_cache.add(code)
@@ -190,47 +190,53 @@

# 想要买的代码
class WantBuyCodesManager:
    redisManager = redis_manager.RedisManager(0)
    __redis_key = "want_buy_codes"
    __instance = None

    def __new__(cls, *args, **kwargs):
        if not cls.__instance:
            cls.__instance = super(WantBuyCodesManager, cls).__new__(cls, *args, **kwargs)
            # 初始化设置
            # 获取交易窗口的锁
            cls.__instance.redisManager = redis_manager.RedisManager(0)
            cls.__instance.__redis_key = "want_buy_codes"
            cls.__instance.__want_buy_codes_cache = RedisUtils.smembers(cls.__instance.__get_redis(),
                                                                        cls.__instance.__redis_key)
        return cls.__instance

    __want_buy_codes_cache = set()

    @classmethod
    def __get_redis(cls):
        return cls.redisManager.getRedis()
    def __get_redis(self):
        return self.redisManager.getRedis()

    @classmethod
    def clear(cls):
        RedisUtils.delete(cls.__get_redis(), cls.__redis_key)
    def clear(self):
        RedisUtils.delete(self.__get_redis(), self.__redis_key)

    @classmethod
    def add_code(cls, code):
        cls.__want_buy_codes_cache.add(code)
        RedisUtils.sadd(cls.__get_redis(), cls.__redis_key, code)
        RedisUtils.expire(cls.__get_redis(), cls.__redis_key, tool.get_expire())
    def add_code(self, code):
        self.__want_buy_codes_cache.add(code)
        RedisUtils.sadd(self.__get_redis(), self.__redis_key, code)
        RedisUtils.expire(self.__get_redis(), self.__redis_key, tool.get_expire())

    @classmethod
    def remove_code(cls, code):
        cls.__want_buy_codes_cache.discard(code)
        RedisUtils.srem(cls.__get_redis(), cls.__redis_key, code)
    def remove_code(self, code):
        self.__want_buy_codes_cache.discard(code)
        RedisUtils.srem(self.__get_redis(), self.__redis_key, code)

    @classmethod
    def sync(cls):
        cls.__want_buy_codes_cache.clear()
        codes = RedisUtils.smembers(cls.__get_redis(), cls.__redis_key)
    def sync(self):
        self.__want_buy_codes_cache.clear()
        codes = RedisUtils.smembers(self.__get_redis(), self.__redis_key)
        if codes:
            cls.__want_buy_codes_cache += set(codes)
            self.__want_buy_codes_cache += set(codes)

    @classmethod
    def is_in(cls, code):
        return RedisUtils.sismember(cls.__get_redis(), cls.__redis_key, code)
    def is_in(self, code):
        return RedisUtils.sismember(self.__get_redis(), self.__redis_key, code)

    @classmethod
    def is_in_cache(cls, code):
        return code in cls.__want_buy_codes_cache
    def is_in_cache(self, code):
        return code in self.__want_buy_codes_cache

    @classmethod
    def list_code(cls):
        return RedisUtils.smembers(cls.__get_redis(), cls.__redis_key)
    def list_code(self):
        return RedisUtils.smembers(self.__get_redis(), self.__redis_key)

    def list_code_cache(self):
        return self.__want_buy_codes_cache


# 暂停下单代码管理
@@ -472,7 +478,7 @@
    @classmethod
    def get_price_pre_cache(cls, code):
        if code in cls.__price_pre_cache:
            return cls.__price_pre_cache[code]
            return float(cls.__price_pre_cache[code])
        val = cls.get_price_pre(code)
        if val:
            cls.__price_pre_cache[code] = val
@@ -486,7 +492,7 @@
            return
        redis_instance = cls.__redisManager.getRedis()
        RedisUtils.setex(redis_instance, "price-pre-{}".format(code), tool.get_expire(), str(price))
        cls.__price_pre_cache[code] = price
        cls.__price_pre_cache[code] = float(price)


__limit_up_price_dict = {}