向爬虫而生---Redis 实战篇1 动态Redis连接管理在Python中的实践

import redis
 
class RedisClient:
    """
    Redis客户端，用于管理Redis连接。
    """
    def __init__(self, host='localhost', port=6379, db=0):
        self.host = host
        self.port = port
        self.db = db
        self._connection = None
 
    def connect(self):
        """
        建立到Redis服务器的连接。
        """
        self._connection = redis.Redis(host=self.host, port=self.port, db=self.db)
 
    def disconnect(self):
        """
        关闭与Redis服务器的连接。
        """
        if self._connection:
            self._connection.close()
            self._connection = None
 
    def set(self, name, value, ex=None, px=None, nx=False, xx=False):
        """
        设置键值对，使用与redis.Redis.set相同的参数。
        """
        self._connect_if_needed()
        return self._connection.set(name, value, ex, px, nx, xx)
 
    def get(self, name):
        """
        获取键的值，使用与redis.Redis.get相同的参数。
        """
        self._connect_if_needed()
        return self._connection.get(name)
 
    def _connect_if_needed(self):
        """
        如果尚未连接，则建立连接。
        """
        if not self._connection:
            self.connect()
 
# 使用示例
client = RedisClient()
client.set('key', 'value')
print(client.get('key'))
client.disconnect()

这段代码定义了一个名为RedisClient的类，用于管理与Redis服务器的连接。它提供了连接和断开连接的方法，并且在需要时才建立连接。此外，它还提供了set和get方法，以便动态地存储和检索数据。这个类的实例可以用来进行基本的Redis操作，而无需手动管理连接的打开和关闭。