feat(tunnel/usage): proxy writer 双优先级队列、hub 连接压力感知选择、usage 请求记录级别控制及 trace 页面 proxy timing 增强

apps/aether-gateway/src/handlers/admin/observability/monitoring/tests/trace.rs

@@ -1,5 +1,5 @@
 use super::super::test_support::{
-    request_context, sample_candidate, sample_endpoint, sample_key, sample_provider,
+    request_context, sample_candidate, sample_endpoint, sample_key, sample_provider, sample_usage,
 };
 use super::local_monitoring_response;
 use crate::AppState;
@@ -10,6 +10,7 @@ use std::sync::Arc;
 
 use aether_data::repository::candidates::InMemoryRequestCandidateRepository;
 use aether_data::repository::provider_catalog::InMemoryProviderCatalogReadRepository;
+use aether_data::repository::usage::InMemoryUsageReadRepository;
 
 #[tokio::test]
 async fn admin_monitoring_trace_request_returns_local_payload() {
@@ -156,6 +157,88 @@ async fn admin_monitoring_trace_request_keeps_format_conversion_disabled_candida
     );
 }
 
+#[tokio::test]
+async fn admin_monitoring_trace_request_enriches_proxy_timing_from_usage_audit() {
+    let mut candidate = sample_candidate(
+        "cand-used",
+        "request-1",
+        1,
+        RequestCandidateStatus::Success,
+        Some(101),
+        Some(33),
+        Some(200),
+    );
+    candidate.extra_data = Some(json!({
+        "proxy": {
+            "node_id": "proxy-node-1",
+            "node_name": "edge-1",
+            "source": "provider"
+        }
+    }));
+
+    let request_candidates = Arc::new(InMemoryRequestCandidateRepository::seed(vec![candidate]));
+    let provider_catalog = Arc::new(InMemoryProviderCatalogReadRepository::seed(
+        vec![sample_provider()],
+        vec![sample_endpoint()],
+        vec![sample_key()],
+    ));
+    let mut usage = sample_usage(
+        "request-1",
+        "provider-1",
+        "OpenAI",
+        40,
+        0.02,
+        "completed",
+        Some(200),
+        100,
+    );
+    usage.candidate_id = Some("cand-used".to_string());
+    usage.response_headers = Some(json!({
+        "x-proxy-timing": "{\"connection_acquire_ms\":125,\"response_wait_ms\":475,\"ttfb_ms\":600}"
+    }));
+    let usage_repository = Arc::new(InMemoryUsageReadRepository::seed(vec![usage]));
+    let data_state =
+        crate::data::GatewayDataState::with_request_candidate_and_usage_repository_for_tests(
+            request_candidates,
+            usage_repository,
+        )
+        .with_provider_catalog_reader(provider_catalog);
+    let state = AppState::new()
+        .expect("state should build")
+        .with_data_state_for_tests(data_state);
+    let context = request_context(
+        http::Method::GET,
+        "/api/admin/monitoring/trace/request-1?attempted_only=true",
+    );
+
+    let response = local_monitoring_response(&state, &context)
+        .await
+        .expect("handler should not error")
+        .expect("route should be handled locally");
+
+    assert_eq!(response.status(), http::StatusCode::OK);
+    let body = to_bytes(response.into_body(), usize::MAX)
+        .await
+        .expect("body should read");
+    let payload: serde_json::Value = serde_json::from_slice(&body).expect("json body should parse");
+    assert_eq!(
+        payload["candidates"][0]["extra_data"]["first_byte_time_ms"],
+        json!(30)
+    );
+    assert_eq!(
+        payload["candidates"][0]["extra_data"]["proxy"]["ttfb_ms"],
+        json!(600)
+    );
+    assert_eq!(
+        payload["candidates"][0]["extra_data"]["proxy"]["timing"]["connection_acquire_ms"],
+        json!(125)
+    );
+    assert_eq!(
+        payload["candidates"][0]["extra_data"]["proxy"]["timing"]["response_wait_ms"],
+        json!(475)
+    );
+}
+
 #[tokio::test]
 async fn admin_monitoring_trace_provider_stats_returns_local_payload() {
     let request_candidates = Arc::new(InMemoryRequestCandidateRepository::seed(vec![

apps/aether-gateway/src/handlers/admin/observability/monitoring/trace.rs

@@ -51,9 +51,15 @@ pub(super) async fn build_admin_monitoring_trace_request_response(
             attempted_only,
         ));
     };
+    let usage = state
+        .data
+        .read_request_usage_audit(&request_id)
+        .await
+        .map_err(|err| GatewayError::Internal(err.to_string()))?;
 
     Ok(build_admin_monitoring_trace_request_payload_response(
         &trace,
+        usage.as_ref(),
     ))
 }