84 lines
2.4 KiB
Python
84 lines
2.4 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
数据库配置模块
|
||
仅包含数据库连接的配置信息
|
||
"""
|
||
|
||
from sqlalchemy.ext.declarative import declarative_base
|
||
from config.settings import settings
|
||
|
||
# 创建Model基类
|
||
Base = declarative_base()
|
||
|
||
# 数据库连接配置
|
||
class DBConfig:
|
||
"""数据库配置类"""
|
||
# 数据库连接URL
|
||
DATABASE_URL = settings.DATABASE_URL
|
||
|
||
# 数据库连接参数
|
||
DATABASE_ARGS = settings.DATABASE_ARGS
|
||
|
||
# 异步数据库连接参数
|
||
ASYNC_DATABASE_ARGS = {
|
||
"pool_size": settings.DB_POOL_SIZE,
|
||
"max_overflow": settings.DB_MAX_OVERFLOW,
|
||
"pool_recycle": settings.DB_POOL_RECYCLE,
|
||
"echo": settings.DB_ECHO,
|
||
"pool_pre_ping": True
|
||
}
|
||
|
||
@classmethod
|
||
def create_db_url(cls, include_db_name=True):
|
||
"""
|
||
创建数据库连接URL
|
||
|
||
Args:
|
||
include_db_name: 是否包含数据库名称
|
||
|
||
Returns:
|
||
str: 数据库连接URL
|
||
"""
|
||
if include_db_name:
|
||
return settings.DATABASE_URL
|
||
else:
|
||
# 根据数据库类型创建不同的连接URL
|
||
if settings.DB_DIALECT == 'sqlite':
|
||
return "sqlite:///:memory:"
|
||
return f"{settings.DB_DIALECT}+{settings.DB_DRIVER}://{settings.DB_USER}:{settings.DB_PASSWORD}@{settings.DB_HOST}:{settings.DB_PORT}/mysql"
|
||
|
||
|
||
# 缓存配置
|
||
class CacheConfig:
|
||
"""缓存配置类，定义Redis缓存配置"""
|
||
# Redis连接配置
|
||
REDIS_URL = settings.REDIS_URL
|
||
REDIS_HOST = settings.REDIS_HOST
|
||
REDIS_PORT = settings.REDIS_PORT
|
||
REDIS_DB = settings.REDIS_DB
|
||
REDIS_PASSWORD = settings.REDIS_PASSWORD
|
||
REDIS_PREFIX = settings.REDIS_PREFIX
|
||
REDIS_SOCKET_TIMEOUT = settings.REDIS_SOCKET_TIMEOUT
|
||
REDIS_SOCKET_CONNECT_TIMEOUT = settings.REDIS_SOCKET_CONNECT_TIMEOUT
|
||
REDIS_DECODE_RESPONSES = settings.REDIS_DECODE_RESPONSES
|
||
|
||
@classmethod
|
||
def get_redis_connection_args(cls):
|
||
"""
|
||
获取Redis连接参数
|
||
|
||
Returns:
|
||
dict: Redis连接参数字典
|
||
"""
|
||
return {
|
||
"host": cls.REDIS_HOST,
|
||
"port": cls.REDIS_PORT,
|
||
"db": cls.REDIS_DB,
|
||
"password": cls.REDIS_PASSWORD,
|
||
"socket_timeout": cls.REDIS_SOCKET_TIMEOUT,
|
||
"socket_connect_timeout": cls.REDIS_SOCKET_CONNECT_TIMEOUT,
|
||
"decode_responses": cls.REDIS_DECODE_RESPONSES
|
||
}
|