83 lines
2.3 KiB
Rust
83 lines
2.3 KiB
Rust
|
//! 充电柜矩阵系统 — 后端入口
|
|||
|
|
|
|||
|
|
mod config;
|
|||
|
|
mod db;
|
|||
|
|
mod error;
|
|||
|
|
mod middleware;
|
|||
|
|
mod redis;
|
|||
|
|
mod routes;
|
|||
|
|
mod tcp;
|
|||
|
|
|
|||
|
|
use std::net::SocketAddr;
|
|||
|
|
use tracing_subscriber::EnvFilter;
|
|||
|
|
|
|||
|
|
use tcp::commands::AppState;
|
|||
|
|
use tcp::connection::ConnectionPool;
|
|||
|
|
|
|||
|
|
#[tokio::main]
|
|||
|
|
async fn main() {
|
|||
|
|
// 初始化日志
|
|||
|
|
tracing_subscriber::fmt()
|
|||
|
|
.with_env_filter(EnvFilter::try_from_default_env().unwrap_or_else(|_| "info".into()))
|
|||
|
|
.init();
|
|||
|
|
|
|||
|
|
// 加载配置
|
|||
|
|
let cfg = config::Config::from_env();
|
|||
|
|
cfg.validate();
|
|||
|
|
tracing::info!("配置加载完成");
|
|||
|
|
|
|||
|
|
// 初始化 MySQL 连接池
|
|||
|
|
let db_pool = db::init_pool(&cfg.database_url)
|
|||
|
|
.await
|
|||
|
|
.expect("MySQL 连接失败");
|
|||
|
|
tracing::info!("数据库连接成功");
|
|||
|
|
|
|||
|
|
// 执行自动迁移
|
|||
|
|
db::migrate::run(&db_pool)
|
|||
|
|
.await
|
|||
|
|
.expect("数据库迁移失败");
|
|||
|
|
tracing::info!("数据库迁移完成");
|
|||
|
|
|
|||
|
|
// 初始化 Redis 连接（ConnectionManager 内部管理重连，可直接 Clone 共享）
|
|||
|
|
let redis_conn = redis::init_connection(&cfg.redis_url)
|
|||
|
|
.await
|
|||
|
|
.expect("Redis 连接失败");
|
|||
|
|
tracing::info!("Redis 连接成功");
|
|||
|
|
|
|||
|
|
// 初始化设备连接池
|
|||
|
|
let device_pool = ConnectionPool::new();
|
|||
|
|
|
|||
|
|
// 构建共享状态
|
|||
|
|
let state = AppState {
|
|||
|
|
device_pool: device_pool.clone(),
|
|||
|
|
redis: redis_conn.clone(),
|
|||
|
|
mysql: db_pool.clone(),
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 启动 TCP 设备服务（后台）
|
|||
|
|
let tcp_addr = format!("{}:{}", cfg.host, cfg.tcp_port);
|
|||
|
|
tracing::info!("TCP 设备服务启动于 {}", tcp_addr);
|
|||
|
|
let tcp_pool = device_pool.clone();
|
|||
|
|
let tcp_redis = redis_conn.clone();
|
|||
|
|
let tcp_mysql = db_pool.clone();
|
|||
|
|
tokio::spawn(async move {
|
|||
|
|
tcp::run_tcp_server(tcp_addr, tcp_pool, tcp_redis, tcp_mysql).await;
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 构建 HTTP 路由（包含设备指令下发端点和认证中间件）
|
|||
|
|
let app = routes::build(state);
|
|||
|
|
|
|||
|
|
// 启动 HTTP 服务
|
|||
|
|
let addr: SocketAddr = format!("{}:{}", cfg.host, cfg.port)
|
|||
|
|
.parse()
|
|||
|
|
.expect("无效的监听地址");
|
|||
|
|
tracing::info!("HTTP 服务器启动于 http://{}", addr);
|
|||
|
|
|
|||
|
|
let listener = tokio::net::TcpListener::bind(addr)
|
|||
|
|
.await
|
|||
|
|
.expect("端口绑定失败");
|
|||
|
|
axum::serve(listener, app)
|
|||
|
|
.await
|
|||
|
|
.expect("服务器运行失败");
|
|||
|
|
}
|