97 lines
3.2 KiB
Python
97 lines
3.2 KiB
Python
|
import redis
|
|||
|
|
from loguru import logger
|
|||
|
|
|
|||
|
|
|
|||
|
|
class RedisClient:
|
|||
|
|
def __init__(self, host: str = 'localhost', port: int = 6379, password: str = None):
|
|||
|
|
self.host = host
|
|||
|
|
self.port = port
|
|||
|
|
self.password = password
|
|||
|
|
self.browser_client = None
|
|||
|
|
self.task_client = None
|
|||
|
|
self.cache_client = None
|
|||
|
|
self.ok_client = None
|
|||
|
|
self.init()
|
|||
|
|
|
|||
|
|
# 初始化
|
|||
|
|
def init(self):
|
|||
|
|
"""
|
|||
|
|
初始化Redis客户端
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
if self.browser_client is None:
|
|||
|
|
self.browser_client = redis.Redis(host=self.host, port=self.port, password=self.password, db=0,
|
|||
|
|
decode_responses=True)
|
|||
|
|
|
|||
|
|
if self.task_client is None:
|
|||
|
|
self.task_client = redis.Redis(host=self.host, port=self.port, password=self.password, db=1,
|
|||
|
|
decode_responses=True)
|
|||
|
|
|
|||
|
|
if self.cache_client is None:
|
|||
|
|
self.cache_client = redis.Redis(host=self.host, port=self.port, password=self.password, db=2,
|
|||
|
|
decode_responses=True)
|
|||
|
|
|
|||
|
|
if self.ok_client is None:
|
|||
|
|
self.ok_client = redis.Redis(host=self.host, port=self.port, password=self.password, db=3,
|
|||
|
|
decode_responses=True)
|
|||
|
|
|
|||
|
|
logger.info("Redis连接已初始化")
|
|||
|
|
|
|||
|
|
# 关闭连接
|
|||
|
|
def close(self):
|
|||
|
|
self.browser_client.close()
|
|||
|
|
self.task_client.close()
|
|||
|
|
self.cache_client.close()
|
|||
|
|
self.ok_client.close()
|
|||
|
|
logger.info("Redis连接已关闭")
|
|||
|
|
|
|||
|
|
"""browser_client"""
|
|||
|
|
|
|||
|
|
# 写入浏览器信息
|
|||
|
|
async def set_browser(self, browser_id: str, data: dict):
|
|||
|
|
try:
|
|||
|
|
# 处理None值，将其转换为空字符串
|
|||
|
|
processed_data = {}
|
|||
|
|
for key, value in data.items():
|
|||
|
|
if value is None:
|
|||
|
|
processed_data[key] = ""
|
|||
|
|
else:
|
|||
|
|
processed_data[key] = value
|
|||
|
|
|
|||
|
|
self.browser_client.hset(browser_id, mapping=processed_data)
|
|||
|
|
logger.info(f"写入浏览器信息: {browser_id} - {processed_data}")
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"写入浏览器信息失败: {browser_id} - {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 获取浏览器信息
|
|||
|
|
async def get_browser(self, browser_id: str = None):
|
|||
|
|
try:
|
|||
|
|
if browser_id is None:
|
|||
|
|
# 获取全部数据
|
|||
|
|
data = self.browser_client.hgetall()
|
|||
|
|
else:
|
|||
|
|
data = self.browser_client.hgetall(browser_id)
|
|||
|
|
logger.info(f"获取浏览器信息: {browser_id} - {data}")
|
|||
|
|
return data
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"获取浏览器信息失败: {browser_id} - {e}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def main():
|
|||
|
|
host = '183.66.27.14'
|
|||
|
|
port = 50086
|
|||
|
|
password = 'redis_AdJsBP'
|
|||
|
|
redis_client = RedisClient(host, port, password)
|
|||
|
|
# await redis_client.set_browser('9eac7f95ca2d47359ace4083a566e119', {'status': 'online', 'current_task_id': None})
|
|||
|
|
await redis_client.get_browser('9eac7f95ca2d47359ace4083a566e119')
|
|||
|
|
# 关闭连接
|
|||
|
|
redis_client.close()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
import asyncio
|
|||
|
|
|
|||
|
|
asyncio.run(main())
|