126 lines
2.5 KiB
Python
126 lines
2.5 KiB
Python
"""数据库模块
|
||
|
||
重构后的数据库模块，提供：
|
||
- 核心层：引擎、会话、模型、迁移
|
||
- 优化层：缓存、预加载、批处理
|
||
- API层：CRUD、查询构建器、业务API
|
||
- Utils层：装饰器、监控
|
||
- 兼容层：向后兼容的API
|
||
"""
|
||
|
||
# ===== 核心层 =====
|
||
# ===== API层 =====
|
||
from src.common.database.api import (
|
||
AggregateQuery,
|
||
CRUDBase,
|
||
QueryBuilder,
|
||
# ChatStreams API
|
||
get_active_streams,
|
||
# Messages API
|
||
get_chat_history,
|
||
get_message_count,
|
||
# PersonInfo API
|
||
get_or_create_person,
|
||
# ActionRecords API
|
||
get_recent_actions,
|
||
# LLMUsage API
|
||
get_usage_statistics,
|
||
record_llm_usage,
|
||
# 业务API
|
||
save_message,
|
||
store_action_info,
|
||
update_person_affinity,
|
||
)
|
||
|
||
# ===== 兼容层（向后兼容旧API）=====
|
||
from src.common.database.compatibility import (
|
||
MODEL_MAPPING,
|
||
build_filters,
|
||
db_get,
|
||
db_query,
|
||
db_save,
|
||
)
|
||
from src.common.database.core import (
|
||
Base,
|
||
check_and_migrate_database,
|
||
get_db_session,
|
||
get_engine,
|
||
get_session_factory,
|
||
)
|
||
|
||
# ===== 优化层 =====
|
||
from src.common.database.optimization import (
|
||
AdaptiveBatchScheduler,
|
||
DataPreloader,
|
||
MultiLevelCache,
|
||
get_batch_scheduler,
|
||
get_cache,
|
||
get_preloader,
|
||
)
|
||
|
||
# ===== Utils层 =====
|
||
from src.common.database.utils import (
|
||
cached,
|
||
db_operation,
|
||
get_monitor,
|
||
measure_time,
|
||
print_stats,
|
||
record_cache_hit,
|
||
record_cache_miss,
|
||
record_operation,
|
||
reset_stats,
|
||
retry,
|
||
timeout,
|
||
transactional,
|
||
)
|
||
|
||
__all__ = [
|
||
# 兼容层
|
||
"MODEL_MAPPING",
|
||
"AdaptiveBatchScheduler",
|
||
"AggregateQuery",
|
||
# 核心层
|
||
"Base",
|
||
# API层 - 基础类
|
||
"CRUDBase",
|
||
"DataPreloader",
|
||
# 优化层
|
||
"MultiLevelCache",
|
||
"QueryBuilder",
|
||
"build_filters",
|
||
"cached",
|
||
"check_and_migrate_database",
|
||
"db_get",
|
||
"db_operation",
|
||
"db_query",
|
||
"db_save",
|
||
"get_active_streams",
|
||
"get_batch_scheduler",
|
||
"get_cache",
|
||
"get_chat_history",
|
||
"get_db_session",
|
||
"get_engine",
|
||
"get_message_count",
|
||
"get_monitor",
|
||
"get_or_create_person",
|
||
"get_preloader",
|
||
"get_recent_actions",
|
||
"get_session_factory",
|
||
"get_usage_statistics",
|
||
"measure_time",
|
||
"print_stats",
|
||
"record_cache_hit",
|
||
"record_cache_miss",
|
||
"record_llm_usage",
|
||
"record_operation",
|
||
"reset_stats",
|
||
# Utils层
|
||
"retry",
|
||
"save_message",
|
||
# API层 - 业务API
|
||
"store_action_info",
|
||
"timeout",
|
||
"transactional",
|
||
"update_person_affinity",
|
||
]
|