102 lines
3.0 KiB
Python
102 lines
3.0 KiB
Python
# main.py - 修复 reload 问题
|
|
#!/usr/bin/env python3
|
|
"""
|
|
FastAPI chongming 示例
|
|
主入口文件，适配 chongming 打包系统
|
|
"""
|
|
import os
|
|
import sys
|
|
import platform
|
|
from utils.launch import launch
|
|
from multiprocessing import freeze_support
|
|
import server
|
|
|
|
use_shell = platform.system() == "Windows"
|
|
|
|
|
|
def app_run(app_config: dict, default_config: dict):
|
|
"""
|
|
应用运行函数 - 适配模块银行环境
|
|
|
|
Args:
|
|
app_config: 应用配置字典，包含服务器配置、环境设置等
|
|
default_config: 默认配置字典，包含应用名称、版本等基础信息
|
|
|
|
Returns:
|
|
None: 此函数不返回值，直接启动服务器并阻塞运行
|
|
"""
|
|
from app.core.logger import get_logger # type: ignore
|
|
|
|
logger = get_logger("app")
|
|
|
|
# 获取服务器配置
|
|
server_config = app_config.get("server", {})
|
|
host = server_config.get("host", "0.0.0.0")
|
|
port = server_config.get("port", 8000)
|
|
timeout_keep_alive = server_config.get("timeout_keep_alive", 5)
|
|
|
|
# 打包环境强制禁用重载
|
|
is_frozen = getattr(sys, "frozen", False)
|
|
reload = False if is_frozen else server_config.get("reload", False)
|
|
# 支持通过环境变量 APP_WORKERS 覆盖 worker 数量（方便嵌入式设备调优）
|
|
default_workers = server_config.get("workers", 4)
|
|
workers = int(os.environ.get("APP_WORKERS", str(default_workers)))
|
|
|
|
# 显示启动信息
|
|
env = app_config.get("env", "development")
|
|
logger.info("=" * 50)
|
|
logger.info(f"🚀 启动 {default_config.get('app.name', 'FastAPI 应用')}")
|
|
logger.info(f"📦 版本: {default_config.get('app.version', '1.0.0')}")
|
|
logger.info(f"🌍 环境: {env}")
|
|
logger.info(f"📍 地址: http://{host}:{port}")
|
|
logger.info(f"📚 文档: http://{host}:{port}/docs")
|
|
logger.info(f"🔧 调试: {'启用' if app_config.get('debug', False) else '禁用'}")
|
|
logger.info(f"🔄 重载: {'启用' if reload else '禁用'}")
|
|
logger.info(f"👥 Workers: {workers}")
|
|
logger.info("=" * 50)
|
|
|
|
# 根据操作系统选择不同的服务器启动方式
|
|
if use_shell:
|
|
import uvicorn
|
|
|
|
uvicorn.run(
|
|
"server:app",
|
|
host=host,
|
|
port=port,
|
|
reload=reload,
|
|
workers=workers,
|
|
log_level="debug" if app_config.get("debug", False) else "info",
|
|
access_log=False,
|
|
timeout_keep_alive=timeout_keep_alive,
|
|
)
|
|
else:
|
|
from gunicorn.app.wsgiapp import run
|
|
|
|
sys.argv = [
|
|
"gunicorn",
|
|
"server:app",
|
|
"-k",
|
|
"uvicorn.workers.UvicornWorker",
|
|
"--preload",
|
|
"-w",
|
|
str(workers),
|
|
"-b",
|
|
f"{host}:{port}",
|
|
]
|
|
run()
|
|
|
|
|
|
def main():
|
|
"""
|
|
主函数入口 - 初始化多进程支持并启动应用
|
|
|
|
Returns:
|
|
None: 此函数不返回值，调用launch函数启动应用
|
|
"""
|
|
freeze_support()
|
|
launch(app_run, "production")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|