【python】使用redis pool的一种单例实现方式

最新推荐文章于 2025-06-23 19:20:32 发布

原创最新推荐文章于 2025-06-23 19:20:32 发布 · 1.3w 阅读

4 ·

CC 4.0 BY-SA版权

Python 同时被 2 个专栏收录

44 篇文章

订阅专栏

Redis

15 篇文章

订阅专栏

本文介绍了一种使用Python实现的Redis连接池单例模式，该模式允许多个Redis实例共享同一连接池，从而提高资源利用效率。通过定义静态配置类`RedisDBConfig`设置Redis的基本参数，并创建`RedisCache`类来管理连接池，实现了如设置、获取和删除缓存等基本操作。

为适应多个redis实例共享同一个连接池的场景，可以类似于以下单例方式实现:

import redis

class RedisDBConfig:
    HOST = '127.0.0.1'
    PORT = 6379
    DBID = 0

def operator_status(func):
    '''get operatoration status
    '''
    def gen_status(*args, **kwargs):
        error, result = None, None
        try:
            result = func(*args, **kwargs)
        except Exception as e:
            error = str(e)

        return {'result': result, 'error':  error}

    return gen_status

class RedisCache(object):
    def __init__(self):
        if not hasattr(RedisCache, 'pool'):
            RedisCache.create_pool()
        self._connection = redis.Redis(connection_pool = RedisCache.pool)

    @staticmethod
    def create_pool():
        RedisCache.pool = redis.ConnectionPool(
                host = RedisDBConfig.HOST,
                port = RedisDBConfig.PORT,
                db   = RedisDBConfig.DBID)

    @operator_status
    def set_data(self, key, value):
        '''set data with (key, value)
        '''
        return self._connection.set(key, value)

    @operator_status
    def get_data(self, key):
        '''get data by key
        '''
        return self._connection.get(key)

    @operator_status
    def del_data(self, key):
        '''delete cache by key
        '''
        return self._connection.delete(key)


if __name__ == '__main__':
    print RedisCache().set_data('Testkey', "Simple Test")
    print RedisCache().get_data('Testkey')
    print RedisCache().del_data('Testkey')
    print RedisCache().get_data('Testkey')