|
|
"""主应用入口 - 对应Go的cmd/server/main.go"""
|
|
|
import os
|
|
|
from contextlib import asynccontextmanager
|
|
|
|
|
|
from fastapi import FastAPI
|
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
from fastapi.responses import HTMLResponse
|
|
|
from fastapi.staticfiles import StaticFiles
|
|
|
|
|
|
from app.api import router, admin_router
|
|
|
from app.websocket import WebSocketServer
|
|
|
from app.core.config import get_config, get_settings
|
|
|
from app.core.logger import info, error, setup_logging
|
|
|
from app.repositories.database import init_db
|
|
|
|
|
|
|
|
|
# 获取配置
|
|
|
config = get_config()
|
|
|
settings = get_settings()
|
|
|
|
|
|
# 设置日志
|
|
|
setup_logging()
|
|
|
|
|
|
|
|
|
@asynccontextmanager
|
|
|
async def lifespan(app: FastAPI):
|
|
|
"""应用生命周期管理"""
|
|
|
# 启动时执行
|
|
|
info("Starting Market Data Service...")
|
|
|
|
|
|
# 初始化数据库
|
|
|
try:
|
|
|
init_db()
|
|
|
info("Database initialized")
|
|
|
except Exception as e:
|
|
|
error(f"Database initialization failed: {e}")
|
|
|
|
|
|
yield
|
|
|
|
|
|
# 关闭时执行
|
|
|
info("Shutting down Market Data Service...")
|
|
|
|
|
|
|
|
|
# 创建FastAPI应用
|
|
|
app = FastAPI(
|
|
|
title="统一行情数据服务",
|
|
|
description="提供股票和期货的标准化行情数据查询服务",
|
|
|
version="1.0.0",
|
|
|
lifespan=lifespan
|
|
|
)
|
|
|
|
|
|
# 添加CORS中间件
|
|
|
app.add_middleware(
|
|
|
CORSMiddleware,
|
|
|
allow_origins=["*"],
|
|
|
allow_credentials=True,
|
|
|
allow_methods=["*"],
|
|
|
allow_headers=["*"],
|
|
|
)
|
|
|
|
|
|
# 注册API路由
|
|
|
app.include_router(router, prefix="/v1")
|
|
|
app.include_router(admin_router, prefix="/v1")
|
|
|
|
|
|
# WebSocket服务器
|
|
|
ws_server = WebSocketServer()
|
|
|
|
|
|
|
|
|
@app.websocket("/v1/stream")
|
|
|
async def websocket_endpoint(websocket):
|
|
|
"""WebSocket端点"""
|
|
|
import uuid
|
|
|
client_id = str(uuid.uuid4())
|
|
|
await ws_server.handle(websocket, client_id)
|
|
|
|
|
|
|
|
|
# 管理后台页面HTML（简化版）
|
|
|
ADMIN_HTML = """<!DOCTYPE html>
|
|
|
<html lang="zh-CN">
|
|
|
<head>
|
|
|
<meta charset="UTF-8">
|
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|
|
<title>行情数据服务 - 管理后台</title>
|
|
|
<style>
|
|
|
* { margin: 0; padding: 0; box-sizing: border-box; }
|
|
|
body {
|
|
|
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'PingFang SC', sans-serif;
|
|
|
background: #f0f2f5;
|
|
|
color: #333;
|
|
|
}
|
|
|
.layout { display: flex; min-height: 100vh; }
|
|
|
.sidebar {
|
|
|
width: 200px;
|
|
|
background: #001529;
|
|
|
color: #fff;
|
|
|
position: fixed;
|
|
|
height: 100vh;
|
|
|
overflow-y: auto;
|
|
|
}
|
|
|
.logo {
|
|
|
padding: 16px;
|
|
|
font-size: 18px;
|
|
|
font-weight: bold;
|
|
|
border-bottom: 1px solid rgba(255,255,255,0.1);
|
|
|
}
|
|
|
.nav-menu { padding: 16px 0; }
|
|
|
.nav-item {
|
|
|
padding: 12px 24px;
|
|
|
cursor: pointer;
|
|
|
transition: all 0.3s;
|
|
|
display: flex;
|
|
|
align-items: center;
|
|
|
gap: 8px;
|
|
|
}
|
|
|
.nav-item:hover { background: rgba(255,255,255,0.05); }
|
|
|
.nav-item.active { background: #1890ff; }
|
|
|
.main-content {
|
|
|
flex: 1;
|
|
|
margin-left: 200px;
|
|
|
padding: 24px;
|
|
|
}
|
|
|
.header {
|
|
|
background: #fff;
|
|
|
padding: 16px 24px;
|
|
|
margin: -24px -24px 24px -24px;
|
|
|
box-shadow: 0 1px 4px rgba(0,0,0,0.1);
|
|
|
display: flex;
|
|
|
justify-content: space-between;
|
|
|
align-items: center;
|
|
|
}
|
|
|
.page-title { font-size: 20px; font-weight: 500; }
|
|
|
.card {
|
|
|
background: #fff;
|
|
|
border-radius: 8px;
|
|
|
padding: 24px;
|
|
|
margin-bottom: 24px;
|
|
|
box-shadow: 0 1px 2px rgba(0,0,0,0.05);
|
|
|
}
|
|
|
.card-title {
|
|
|
font-size: 16px;
|
|
|
font-weight: 500;
|
|
|
margin-bottom: 16px;
|
|
|
padding-bottom: 12px;
|
|
|
border-bottom: 1px solid #f0f0f0;
|
|
|
}
|
|
|
.stats-grid {
|
|
|
display: grid;
|
|
|
grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
|
|
|
gap: 16px;
|
|
|
margin-bottom: 24px;
|
|
|
}
|
|
|
.stat-card {
|
|
|
background: #fff;
|
|
|
border-radius: 8px;
|
|
|
padding: 20px;
|
|
|
box-shadow: 0 1px 2px rgba(0,0,0,0.05);
|
|
|
}
|
|
|
.stat-value {
|
|
|
font-size: 32px;
|
|
|
font-weight: bold;
|
|
|
color: #1890ff;
|
|
|
}
|
|
|
.stat-label { color: #666; margin-top: 4px; }
|
|
|
.btn {
|
|
|
padding: 8px 16px;
|
|
|
border: none;
|
|
|
border-radius: 4px;
|
|
|
font-size: 14px;
|
|
|
cursor: pointer;
|
|
|
transition: all 0.3s;
|
|
|
}
|
|
|
.btn-primary { background: #1890ff; color: #fff; }
|
|
|
.btn-primary:hover { background: #40a9ff; }
|
|
|
.btn-success { background: #52c41a; color: #fff; }
|
|
|
.btn-danger { background: #ff4d4f; color: #fff; }
|
|
|
.hidden { display: none; }
|
|
|
.page { display: none; }
|
|
|
.page.active { display: block; }
|
|
|
</style>
|
|
|
</head>
|
|
|
<body>
|
|
|
<div class="layout">
|
|
|
<aside class="sidebar">
|
|
|
<div class="logo">📊 行情数据服务</div>
|
|
|
<nav class="nav-menu">
|
|
|
<div class="nav-item active" onclick="showPage('dashboard')">
|
|
|
<span>📈</span> 系统概览
|
|
|
</div>
|
|
|
<div class="nav-item" onclick="showPage('config')">
|
|
|
<span>⚙️</span> 配置管理
|
|
|
</div>
|
|
|
<div class="nav-item" onclick="showPage('adapters')">
|
|
|
<span>🔌</span> 数据源适配
|
|
|
</div>
|
|
|
<div class="nav-item" onclick="showPage('tests')">
|
|
|
<span>🧪</span> 接口测试
|
|
|
</div>
|
|
|
</nav>
|
|
|
</aside>
|
|
|
|
|
|
<main class="main-content">
|
|
|
<div class="header">
|
|
|
<h1 class="page-title">系统概览</h1>
|
|
|
<div>
|
|
|
<button class="btn btn-success" onclick="reloadConfig()">🔄 热加载配置</button>
|
|
|
<button class="btn btn-danger" onclick="restartService()">🔁 重启服务</button>
|
|
|
</div>
|
|
|
</div>
|
|
|
|
|
|
<!-- 系统概览页面 -->
|
|
|
<div id="dashboard" class="page active">
|
|
|
<div class="stats-grid">
|
|
|
<div class="stat-card">
|
|
|
<div class="stat-value" id="stat-status">运行中</div>
|
|
|
<div class="stat-label">运行状态</div>
|
|
|
</div>
|
|
|
<div class="stat-card">
|
|
|
<div class="stat-value" id="stat-uptime">-</div>
|
|
|
<div class="stat-label">运行时长</div>
|
|
|
</div>
|
|
|
<div class="stat-card">
|
|
|
<div class="stat-value">1.0.0</div>
|
|
|
<div class="stat-label">系统版本</div>
|
|
|
</div>
|
|
|
<div class="stat-card">
|
|
|
<div class="stat-value" id="stat-threads">-</div>
|
|
|
<div class="stat-label">线程数量</div>
|
|
|
</div>
|
|
|
</div>
|
|
|
|
|
|
<div class="card">
|
|
|
<div class="card-title">API文档</div>
|
|
|
<p>访问 <a href="/docs">/docs</a> 查看Swagger API文档</p>
|
|
|
<p>访问 <a href="/redoc">/redoc</a> 查看ReDoc API文档</p>
|
|
|
</div>
|
|
|
</div>
|
|
|
|
|
|
<!-- 配置管理页面 -->
|
|
|
<div id="config" class="page">
|
|
|
<div class="card">
|
|
|
<div class="card-title">配置管理</div>
|
|
|
<p>配置管理功能开发中...</p>
|
|
|
</div>
|
|
|
</div>
|
|
|
|
|
|
<!-- 数据源适配页面 -->
|
|
|
<div id="adapters" class="page">
|
|
|
<div class="card">
|
|
|
<div class="card-title">数据源适配</div>
|
|
|
<p>适配器管理功能开发中...</p>
|
|
|
</div>
|
|
|
</div>
|
|
|
|
|
|
<!-- 接口测试页面 -->
|
|
|
<div id="tests" class="page">
|
|
|
<div class="card">
|
|
|
<div class="card-title">接口测试</div>
|
|
|
<p>接口测试功能开发中...</p>
|
|
|
</div>
|
|
|
</div>
|
|
|
</main>
|
|
|
</div>
|
|
|
|
|
|
<script>
|
|
|
function showPage(pageName) {
|
|
|
document.querySelectorAll('.page').forEach(p => p.classList.remove('active'));
|
|
|
document.querySelectorAll('.nav-item').forEach(n => n.classList.remove('active'));
|
|
|
document.getElementById(pageName).classList.add('active');
|
|
|
event.target.classList.add('active');
|
|
|
}
|
|
|
|
|
|
async function reloadConfig() {
|
|
|
try {
|
|
|
const response = await fetch('/v1/admin/system/reload', {
|
|
|
method: 'POST',
|
|
|
headers: {'Content-Type': 'application/json'}
|
|
|
});
|
|
|
const data = await response.json();
|
|
|
alert(data.message || '热加载完成');
|
|
|
} catch (e) {
|
|
|
alert('热加载失败: ' + e.message);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
async function restartService() {
|
|
|
if (confirm('确定要重启服务吗？')) {
|
|
|
try {
|
|
|
const response = await fetch('/v1/admin/system/restart', {
|
|
|
method: 'POST',
|
|
|
headers: {'Content-Type': 'application/json'}
|
|
|
});
|
|
|
const data = await response.json();
|
|
|
alert(data.message || '重启命令已发送');
|
|
|
} catch (e) {
|
|
|
alert('重启失败: ' + e.message);
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 加载系统状态
|
|
|
async function loadSystemStatus() {
|
|
|
try {
|
|
|
const response = await fetch('/v1/admin/system/status');
|
|
|
const data = await response.json();
|
|
|
if (data.data) {
|
|
|
document.getElementById('stat-uptime').textContent = data.data.uptime;
|
|
|
document.getElementById('stat-threads').textContent = data.data.threads;
|
|
|
}
|
|
|
} catch (e) {
|
|
|
console.error('Failed to load system status:', e);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 页面加载时获取状态
|
|
|
loadSystemStatus();
|
|
|
setInterval(loadSystemStatus, 30000); // 每30秒刷新
|
|
|
</script>
|
|
|
</body>
|
|
|
</html>
|
|
|
"""
|
|
|
|
|
|
|
|
|
@app.get("/admin", response_class=HTMLResponse)
|
|
|
async def admin_page():
|
|
|
"""管理后台页面"""
|
|
|
return ADMIN_HTML
|
|
|
|
|
|
|
|
|
@app.get("/")
|
|
|
async def root():
|
|
|
"""根路径重定向到管理后台"""
|
|
|
return {"message": "Market Data Service API", "docs": "/docs", "admin": "/admin"}
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
import uvicorn
|
|
|
|
|
|
# 从环境变量或配置获取端口
|
|
|
port = settings.port or config.server.port
|
|
|
|
|
|
uvicorn.run(
|
|
|
"app.main:app",
|
|
|
host="0.0.0.0",
|
|
|
port=port,
|
|
|
reload=config.server.mode == "debug"
|
|
|
)
|
