redis封装

Administrator

2023-08-02 a3c05bddc4c61d63b804ee541cf6dd01717ba94d

 trade/l2_trade_util.py

@@ -1,5 +1,6 @@
# 是否在禁止交易代码中
from db import redis_manager
from db.redis_manager import RedisUtils
from utils import tool

__redis_manager = redis_manager.RedisManager(2)
@@ -14,8 +15,8 @@

    @classmethod
    def add_code(cls, code):
        cls.__get_redis().sadd("white_list_codes", code)
        cls.__get_redis().expire("white_list_codes", tool.get_expire())
        RedisUtils.sadd(cls.__get_redis(), "white_list_codes", code)
        RedisUtils.expire(cls.__get_redis(), "white_list_codes", tool.get_expire())

    @classmethod
    def remove_code(cls, code):
@@ -27,11 +28,11 @@

    @classmethod
    def list_codes(cls):
        return cls.__get_redis().smembers("white_list_codes")
        return RedisUtils.smembers(cls.__get_redis(), "white_list_codes")

    @classmethod
    def clear(cls):
        cls.__get_redis().delete("white_list_codes")
        RedisUtils.delete( cls.__get_redis(), "white_list_codes")


class BlackListCodeManager:
@@ -43,8 +44,8 @@

    @classmethod
    def add_code(cls, code):
        cls.__get_redis().sadd("forbidden-trade-codes", code)
        cls.__get_redis().expire("forbidden-trade-codes", tool.get_expire())
        RedisUtils.sadd( cls.__get_redis(), "forbidden-trade-codes", code)
        RedisUtils.expire(cls.__get_redis(), "forbidden-trade-codes", tool.get_expire())

    @classmethod
    def remove_code(cls, code):
@@ -56,11 +57,11 @@

    @classmethod
    def list_codes(cls):
        return cls.__get_redis().smembers("forbidden-trade-codes")
        return RedisUtils.smembers(cls.__get_redis(), "forbidden-trade-codes")

    @classmethod
    def clear(cls):
        cls.__get_redis().delete("forbidden-trade-codes")
        RedisUtils.delete(cls.__get_redis(), "forbidden-trade-codes")


#  初始化禁止交易代码库