mirror of
https://github.com/fawney19/Aether.git
synced 2026-10-11 03:39:49 +08:00
refactor: 大规模模块拆分与代码精简，新增 ai-pipeline/data-contracts 独立 crate
- 新增 aether-ai-pipeline 和 aether-data-contracts crate，将 pipeline 逻辑与数据契约从 gateway 中解耦 - 重构 admin handlers：拆分单体模块为 auth/billing/endpoint/features/model/observability/provider/system 等独立子模块 - 合并 chat/cli 重复代码路径：精简 conversion、finalize、planner 中的 sync/chat/cli 分支 - 重构 scheduler/executor/data 层，引入 facade 模式降低模块间耦合 - 移除冗余的 intent 模块，将 plan_fallback/policy/stream_path/sync_path 迁移至 executor - 前端适配：调整 admin API 调用和 provider 模型测试对话框
This commit is contained in:
fawney19
@@ -0,0 +1,304 @@
|
||||
use super::activity::{
|
||||
build_admin_monitoring_audit_logs_response,
|
||||
build_admin_monitoring_suspicious_activities_response,
|
||||
build_admin_monitoring_system_status_response, build_admin_monitoring_user_behavior_response,
|
||||
};
|
||||
use super::cache::{
|
||||
build_admin_monitoring_cache_config_response, build_admin_monitoring_cache_metrics_response,
|
||||
build_admin_monitoring_cache_stats_response,
|
||||
};
|
||||
use super::cache_affinity_reads::{
|
||||
build_admin_monitoring_cache_affinities_response,
|
||||
build_admin_monitoring_cache_affinity_response,
|
||||
};
|
||||
use super::cache_model_mapping::{
|
||||
build_admin_monitoring_model_mapping_stats_response,
|
||||
build_admin_monitoring_redis_cache_categories_response,
|
||||
};
|
||||
use super::cache_mutations::{
|
||||
build_admin_monitoring_cache_affinity_delete_response,
|
||||
build_admin_monitoring_cache_flush_response,
|
||||
build_admin_monitoring_cache_provider_delete_response,
|
||||
build_admin_monitoring_cache_users_delete_response,
|
||||
build_admin_monitoring_model_mapping_delete_model_response,
|
||||
build_admin_monitoring_model_mapping_delete_provider_response,
|
||||
build_admin_monitoring_model_mapping_delete_response,
|
||||
build_admin_monitoring_redis_keys_delete_response,
|
||||
};
|
||||
use super::resilience::{
|
||||
build_admin_monitoring_reset_error_stats_response,
|
||||
build_admin_monitoring_resilience_circuit_history_response,
|
||||
build_admin_monitoring_resilience_status_response,
|
||||
};
|
||||
use super::trace::{
|
||||
build_admin_monitoring_trace_provider_stats_response,
|
||||
build_admin_monitoring_trace_request_response,
|
||||
};
|
||||
use crate::control::GatewayPublicRequestContext;
|
||||
use crate::handlers::admin::shared::attach_admin_audit_response;
|
||||
use crate::{AppState, GatewayError};
|
||||
use axum::{body::Body, http, response::Response};
|
||||
|
||||
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
|
||||
pub(crate) enum AdminMonitoringRoute {
|
||||
AuditLogs,
|
||||
SystemStatus,
|
||||
SuspiciousActivities,
|
||||
UserBehavior,
|
||||
ResilienceStatus,
|
||||
ResilienceErrorStats,
|
||||
ResilienceCircuitHistory,
|
||||
TraceRequest,
|
||||
TraceProviderStats,
|
||||
CacheStats,
|
||||
CacheAffinity,
|
||||
CacheAffinities,
|
||||
CacheUsersDelete,
|
||||
CacheAffinityDelete,
|
||||
CacheFlush,
|
||||
CacheProviderDelete,
|
||||
CacheConfig,
|
||||
CacheMetrics,
|
||||
CacheModelMappingStats,
|
||||
CacheModelMappingDelete,
|
||||
CacheModelMappingDeleteModel,
|
||||
CacheModelMappingDeleteProvider,
|
||||
CacheRedisKeys,
|
||||
CacheRedisKeysDelete,
|
||||
}
|
||||
|
||||
pub(crate) async fn maybe_build_local_admin_monitoring_response(
|
||||
state: &AppState,
|
||||
request_context: &GatewayPublicRequestContext,
|
||||
) -> Result<Option<Response<Body>>, GatewayError> {
|
||||
let Some(route) = match_admin_monitoring_route(
|
||||
&request_context.request_method,
|
||||
request_context.request_path.as_str(),
|
||||
) else {
|
||||
return Ok(None);
|
||||
};
|
||||
|
||||
match route {
|
||||
AdminMonitoringRoute::AuditLogs => Ok(Some(attach_admin_audit_response(
|
||||
build_admin_monitoring_audit_logs_response(state, request_context).await?,
|
||||
"admin_monitoring_audit_logs_viewed",
|
||||
"view_admin_audit_logs",
|
||||
"audit_log",
|
||||
&admin_monitoring_audit_target_id(request_context),
|
||||
))),
|
||||
AdminMonitoringRoute::ResilienceStatus => Ok(Some(
|
||||
build_admin_monitoring_resilience_status_response(state).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::ResilienceErrorStats => Ok(Some(
|
||||
build_admin_monitoring_reset_error_stats_response(state, request_context).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::ResilienceCircuitHistory => Ok(Some(
|
||||
build_admin_monitoring_resilience_circuit_history_response(state, request_context)
|
||||
.await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheStats => Ok(Some(
|
||||
build_admin_monitoring_cache_stats_response(state).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheAffinities => Ok(Some(attach_admin_audit_response(
|
||||
build_admin_monitoring_cache_affinities_response(state, request_context).await?,
|
||||
"admin_monitoring_cache_affinities_viewed",
|
||||
"view_cache_affinities",
|
||||
"cache_affinity",
|
||||
&admin_monitoring_audit_target_id(request_context),
|
||||
))),
|
||||
AdminMonitoringRoute::CacheAffinity => Ok(Some(attach_admin_audit_response(
|
||||
build_admin_monitoring_cache_affinity_response(state, request_context).await?,
|
||||
"admin_monitoring_cache_affinity_viewed",
|
||||
"view_cache_affinity",
|
||||
"cache_affinity",
|
||||
&admin_monitoring_audit_target_id(request_context),
|
||||
))),
|
||||
AdminMonitoringRoute::CacheUsersDelete => Ok(Some(
|
||||
build_admin_monitoring_cache_users_delete_response(state, request_context).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheAffinityDelete => Ok(Some(
|
||||
build_admin_monitoring_cache_affinity_delete_response(state, request_context).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheFlush => Ok(Some(
|
||||
build_admin_monitoring_cache_flush_response(state).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheProviderDelete => Ok(Some(
|
||||
build_admin_monitoring_cache_provider_delete_response(state, request_context).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheModelMappingDelete => Ok(Some(
|
||||
build_admin_monitoring_model_mapping_delete_response(state).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheModelMappingDeleteModel => Ok(Some(
|
||||
build_admin_monitoring_model_mapping_delete_model_response(state, request_context)
|
||||
.await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheModelMappingDeleteProvider => Ok(Some(
|
||||
build_admin_monitoring_model_mapping_delete_provider_response(state, request_context)
|
||||
.await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheRedisKeysDelete => Ok(Some(
|
||||
build_admin_monitoring_redis_keys_delete_response(state, request_context).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheMetrics => Ok(Some(
|
||||
build_admin_monitoring_cache_metrics_response(state).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheConfig => {
|
||||
Ok(Some(build_admin_monitoring_cache_config_response().await?))
|
||||
}
|
||||
AdminMonitoringRoute::CacheModelMappingStats => Ok(Some(
|
||||
build_admin_monitoring_model_mapping_stats_response(state).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::CacheRedisKeys => Ok(Some(
|
||||
build_admin_monitoring_redis_cache_categories_response(state).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::SystemStatus => Ok(Some(
|
||||
build_admin_monitoring_system_status_response(state).await?,
|
||||
)),
|
||||
AdminMonitoringRoute::SuspiciousActivities => Ok(Some(attach_admin_audit_response(
|
||||
build_admin_monitoring_suspicious_activities_response(state, request_context).await?,
|
||||
"admin_monitoring_suspicious_activities_viewed",
|
||||
"view_suspicious_activities",
|
||||
"suspicious_activity",
|
||||
&admin_monitoring_audit_target_id(request_context),
|
||||
))),
|
||||
AdminMonitoringRoute::UserBehavior => Ok(Some(attach_admin_audit_response(
|
||||
build_admin_monitoring_user_behavior_response(state, request_context).await?,
|
||||
"admin_monitoring_user_behavior_viewed",
|
||||
"view_user_behavior",
|
||||
"user",
|
||||
&admin_monitoring_audit_target_id(request_context),
|
||||
))),
|
||||
AdminMonitoringRoute::TraceRequest => Ok(Some(attach_admin_audit_response(
|
||||
build_admin_monitoring_trace_request_response(state, request_context).await?,
|
||||
"admin_monitoring_request_trace_viewed",
|
||||
"view_request_trace",
|
||||
"request_trace",
|
||||
&admin_monitoring_audit_target_id(request_context),
|
||||
))),
|
||||
AdminMonitoringRoute::TraceProviderStats => Ok(Some(attach_admin_audit_response(
|
||||
build_admin_monitoring_trace_provider_stats_response(state, request_context).await?,
|
||||
"admin_monitoring_provider_trace_stats_viewed",
|
||||
"view_provider_trace_stats",
|
||||
"provider",
|
||||
&admin_monitoring_audit_target_id(request_context),
|
||||
))),
|
||||
}
|
||||
}
|
||||
|
||||
fn admin_monitoring_audit_target_id(request_context: &GatewayPublicRequestContext) -> String {
|
||||
match request_context.request_query_string.as_deref() {
|
||||
Some(query) if !query.trim().is_empty() => {
|
||||
format!("{}?{query}", request_context.request_path)
|
||||
}
|
||||
_ => request_context.request_path.clone(),
|
||||
}
|
||||
}
|
||||
|
||||
pub(crate) fn match_admin_monitoring_route(
|
||||
method: &http::Method,
|
||||
path: &str,
|
||||
) -> Option<AdminMonitoringRoute> {
|
||||
let path = normalize_admin_monitoring_path(path);
|
||||
|
||||
match *method {
|
||||
http::Method::GET => match path {
|
||||
"/api/admin/monitoring/audit-logs" => Some(AdminMonitoringRoute::AuditLogs),
|
||||
"/api/admin/monitoring/system-status" => Some(AdminMonitoringRoute::SystemStatus),
|
||||
"/api/admin/monitoring/suspicious-activities" => {
|
||||
Some(AdminMonitoringRoute::SuspiciousActivities)
|
||||
}
|
||||
"/api/admin/monitoring/resilience-status" => {
|
||||
Some(AdminMonitoringRoute::ResilienceStatus)
|
||||
}
|
||||
"/api/admin/monitoring/resilience/circuit-history" => {
|
||||
Some(AdminMonitoringRoute::ResilienceCircuitHistory)
|
||||
}
|
||||
"/api/admin/monitoring/cache/stats" => Some(AdminMonitoringRoute::CacheStats),
|
||||
"/api/admin/monitoring/cache/affinities" => Some(AdminMonitoringRoute::CacheAffinities),
|
||||
"/api/admin/monitoring/cache/config" => Some(AdminMonitoringRoute::CacheConfig),
|
||||
"/api/admin/monitoring/cache/metrics" => Some(AdminMonitoringRoute::CacheMetrics),
|
||||
"/api/admin/monitoring/cache/model-mapping/stats" => {
|
||||
Some(AdminMonitoringRoute::CacheModelMappingStats)
|
||||
}
|
||||
"/api/admin/monitoring/cache/redis-keys" => Some(AdminMonitoringRoute::CacheRedisKeys),
|
||||
_ if matches_dynamic_segments(path, "/api/admin/monitoring/user-behavior/", 1) => {
|
||||
Some(AdminMonitoringRoute::UserBehavior)
|
||||
}
|
||||
_ if matches_dynamic_segments(
|
||||
path,
|
||||
"/api/admin/monitoring/trace/stats/provider/",
|
||||
1,
|
||||
) =>
|
||||
{
|
||||
Some(AdminMonitoringRoute::TraceProviderStats)
|
||||
}
|
||||
_ if matches_dynamic_segments(path, "/api/admin/monitoring/trace/", 1) => {
|
||||
Some(AdminMonitoringRoute::TraceRequest)
|
||||
}
|
||||
_ if matches_dynamic_segments(path, "/api/admin/monitoring/cache/affinity/", 1) => {
|
||||
Some(AdminMonitoringRoute::CacheAffinity)
|
||||
}
|
||||
_ => None,
|
||||
},
|
||||
http::Method::DELETE => match path {
|
||||
"/api/admin/monitoring/resilience/error-stats" => {
|
||||
Some(AdminMonitoringRoute::ResilienceErrorStats)
|
||||
}
|
||||
"/api/admin/monitoring/cache" => Some(AdminMonitoringRoute::CacheFlush),
|
||||
"/api/admin/monitoring/cache/model-mapping" => {
|
||||
Some(AdminMonitoringRoute::CacheModelMappingDelete)
|
||||
}
|
||||
_ if matches_dynamic_segments(path, "/api/admin/monitoring/cache/users/", 1) => {
|
||||
Some(AdminMonitoringRoute::CacheUsersDelete)
|
||||
}
|
||||
_ if matches_dynamic_segments(path, "/api/admin/monitoring/cache/providers/", 1) => {
|
||||
Some(AdminMonitoringRoute::CacheProviderDelete)
|
||||
}
|
||||
_ if matches_dynamic_segments(path, "/api/admin/monitoring/cache/redis-keys/", 1) => {
|
||||
Some(AdminMonitoringRoute::CacheRedisKeysDelete)
|
||||
}
|
||||
_ if matches_dynamic_segments(
|
||||
path,
|
||||
"/api/admin/monitoring/cache/model-mapping/provider/",
|
||||
2,
|
||||
) =>
|
||||
{
|
||||
Some(AdminMonitoringRoute::CacheModelMappingDeleteProvider)
|
||||
}
|
||||
_ if matches_dynamic_segments(
|
||||
path,
|
||||
"/api/admin/monitoring/cache/model-mapping/",
|
||||
1,
|
||||
) =>
|
||||
{
|
||||
Some(AdminMonitoringRoute::CacheModelMappingDeleteModel)
|
||||
}
|
||||
_ if matches_dynamic_segments(path, "/api/admin/monitoring/cache/affinity/", 4) => {
|
||||
Some(AdminMonitoringRoute::CacheAffinityDelete)
|
||||
}
|
||||
_ => None,
|
||||
},
|
||||
_ => None,
|
||||
}
|
||||
}
|
||||
|
||||
fn normalize_admin_monitoring_path(path: &str) -> &str {
|
||||
let normalized = path.trim_end_matches('/');
|
||||
if normalized.is_empty() {
|
||||
"/"
|
||||
} else {
|
||||
normalized
|
||||
}
|
||||
}
|
||||
|
||||
fn matches_dynamic_segments(path: &str, prefix: &str, dynamic_segments: usize) -> bool {
|
||||
let Some(suffix) = path.strip_prefix(prefix) else {
|
||||
return false;
|
||||
};
|
||||
|
||||
let segments = suffix
|
||||
.split('/')
|
||||
.filter(|segment| !segment.is_empty())
|
||||
.collect::<Vec<_>>();
|
||||
segments.len() == dynamic_segments
|
||||
}
|
||||
Reference in New Issue
Block a user