You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.4 KiB
77 lines
2.4 KiB
from redis import asyncio as aioredis
|
|
from redis.exceptions import AuthenticationError, TimeoutError, RedisError
|
|
from config.database import AsyncSessionLocal
|
|
from config.env import RedisConfig
|
|
from module_admin.service.config_service import ConfigService
|
|
from module_admin.service.dict_service import DictDataService
|
|
from utils.log_util import logger
|
|
|
|
|
|
class RedisUtil:
|
|
"""
|
|
Redis相关方法
|
|
"""
|
|
|
|
@classmethod
|
|
async def create_redis_pool(cls) -> aioredis.Redis:
|
|
"""
|
|
应用启动时初始化redis连接
|
|
|
|
:return: Redis连接对象
|
|
"""
|
|
logger.info('开始连接redis...')
|
|
redis = await aioredis.from_url(
|
|
url=f'redis://{RedisConfig.redis_host}',
|
|
port=RedisConfig.redis_port,
|
|
username=RedisConfig.redis_username,
|
|
password=RedisConfig.redis_password,
|
|
db=RedisConfig.redis_database,
|
|
encoding='utf-8',
|
|
decode_responses=True,
|
|
)
|
|
try:
|
|
connection = await redis.ping()
|
|
if connection:
|
|
logger.info('redis连接成功')
|
|
else:
|
|
logger.error('redis连接失败')
|
|
except AuthenticationError as e:
|
|
logger.error(f'redis用户名或密码错误，详细错误信息：{e}')
|
|
except TimeoutError as e:
|
|
logger.error(f'redis连接超时，详细错误信息：{e}')
|
|
except RedisError as e:
|
|
logger.error(f'redis连接错误，详细错误信息：{e}')
|
|
return redis
|
|
|
|
@classmethod
|
|
async def close_redis_pool(cls, app):
|
|
"""
|
|
应用关闭时关闭redis连接
|
|
|
|
:param app: fastapi对象
|
|
:return:
|
|
"""
|
|
await app.state.redis.close()
|
|
logger.info('关闭redis连接成功')
|
|
|
|
@classmethod
|
|
async def init_sys_dict(cls, redis):
|
|
"""
|
|
应用启动时缓存字典表
|
|
|
|
:param redis: redis对象
|
|
:return:
|
|
"""
|
|
async with AsyncSessionLocal() as session:
|
|
await DictDataService.init_cache_sys_dict_services(session, redis)
|
|
|
|
@classmethod
|
|
async def init_sys_config(cls, redis):
|
|
"""
|
|
应用启动时缓存参数配置表
|
|
|
|
:param redis: redis对象
|
|
:return:
|
|
"""
|
|
async with AsyncSessionLocal() as session:
|
|
await ConfigService.init_cache_sys_config_services(session, redis)
|
|
|