use axum::{body::Body, extract::Request, routing::get, Router}; use library::{config, task}; use tasks::get_tasks; use tower::ServiceBuilder; use tower_http::trace::TraceLayer; mod controller; mod service; mod tasks; /// 启动服务 pub async fn serve() { let addr = format!("0.0.0.0:{}", config!().server.port); let listener = tokio::net::TcpListener::bind(&addr).await.unwrap(); tracing::info!("服务监听地址: {}", addr); // 启动任务 task::start(get_tasks()).await; // 启动应用服务 axum::serve(listener, init()).await.unwrap(); } /// 初始化router，包括router中间件和数据 fn init() -> Router { let trace_layer = TraceLayer::new_for_http().make_span_with(|request: &Request| { let req_id = match request .headers() .get("x-request-id") .and_then(|value| value.to_str().ok()) { Some(v) => v.to_string(), None => String::from("unknown"), }; tracing::error_span!("request_id", id = req_id) }); let auth: Router = controller::init(); Router::new() .route("/", get(|| async { "hello" })) .nest("/gm/v1", auth) .layer( ServiceBuilder::new() .layer(axum::middleware::from_fn( library::middleware::req_ctx::authenticate_ctx, )) .layer(axum::middleware::from_fn( library::middleware::req_log::handle, )) .layer(axum::middleware::from_fn(library::middleware::cors::handle)) .layer(axum::middleware::from_fn( library::middleware::req_id::handle, )) ) }