mirror of
https://github.com/HKUDS/AI-Trader
synced 2026-04-21 13:37:41 +00:00
a2347431f5
workers (#182) * Separate API service from background workers * Update frontend and environment defaults * Update frontend and environment defaults
93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
"""
|
|
AI-Trader Backend Server
|
|
|
|
项目结构：
|
|
- config.py : 配置和环境变量
|
|
- database.py : 数据库初始化和连接
|
|
- utils.py : 通用工具函数
|
|
- tasks.py : 后台任务
|
|
- services.py : 业务逻辑服务
|
|
- routes.py : API路由定义
|
|
- main.py : 应用入口
|
|
"""
|
|
|
|
import secrets
|
|
import logging
|
|
import os
|
|
from logging.handlers import RotatingFileHandler
|
|
|
|
# Setup logging
|
|
LOG_DIR = os.path.join(os.path.dirname(__file__), "logs")
|
|
os.makedirs(LOG_DIR, exist_ok=True)
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s - %(levelname)s - %(message)s",
|
|
handlers=[
|
|
RotatingFileHandler(
|
|
os.path.join(LOG_DIR, "server.log"),
|
|
maxBytes=10 * 1024 * 1024, # 10MB
|
|
backupCount=5
|
|
),
|
|
logging.StreamHandler()
|
|
]
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
from cache import get_cache_status
|
|
from database import init_database, get_database_status
|
|
from routes import create_app
|
|
from tasks import (
|
|
_update_trending_cache,
|
|
background_tasks_enabled_for_api,
|
|
start_background_tasks,
|
|
)
|
|
|
|
# Initialize database
|
|
init_database()
|
|
|
|
# Create app
|
|
app = create_app()
|
|
|
|
|
|
# ==================== Startup ====================
|
|
|
|
@app.on_event("startup")
|
|
async def startup_event():
|
|
"""Startup event - schedule background tasks."""
|
|
db_status = get_database_status()
|
|
logger.info(
|
|
"Database ready: backend=%s details=%s",
|
|
db_status.get("backend"),
|
|
{key: value for key, value in db_status.items() if key != "backend"},
|
|
)
|
|
cache_status = get_cache_status()
|
|
logger.info(
|
|
"Cache ready: enabled=%s configured=%s available=%s prefix=%s client_installed=%s error=%s",
|
|
cache_status.get("enabled"),
|
|
cache_status.get("configured"),
|
|
cache_status.get("available"),
|
|
cache_status.get("prefix"),
|
|
cache_status.get("client_installed"),
|
|
cache_status.get("last_error"),
|
|
)
|
|
# Initialize trending cache
|
|
logger.info("Initializing trending cache...")
|
|
_update_trending_cache()
|
|
if not background_tasks_enabled_for_api():
|
|
logger.info(
|
|
"API background tasks disabled. Run `python service/server/worker.py` "
|
|
"to process prices, profit history, settlements, and market intel."
|
|
)
|
|
return
|
|
|
|
started = start_background_tasks(logger)
|
|
logger.info("Background tasks started: %s", len(started))
|
|
|
|
|
|
# ==================== Run ====================
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run(app, host="0.0.0.0", port=8000)
|