mirror of
https://github.com/fawney19/Aether.git
synced 2026-10-09 10:57:03 +08:00
fawney19
4fc95adfb9
- 新建独立 aether-admin crate 承载 admin 相关共享契约与纯辅助函数 - 拆分 ai_pipeline 下 kiro/private_envelope/conversion/planner 等大文件为子模块目录 - 重组 admin handlers 各业务域（billing/oauth/provider/system/users 等）为目录结构，移除 shared.rs/builders.rs 等反模式 - 移除 ai_pipeline runtime adapters 旧实现（claude/openai/gemini/kiro/vertex/antigravity 等），改由 provider transport 统一承载 - 移除 control_facade/execution_facade/auth_snapshot_facade 等冗余 facade 层 - 拆分 query/billing 与 query/monitoring 模块、state/runtime/payments 与 security 模块 - 扩展架构测试覆盖 admin_billing/admin_model/admin_users 等新模块 - 删除 docs/architecture/refactor-execution-plan.md 已完成的执行计划文档
94 lines
3.0 KiB
Rust
94 lines
3.0 KiB
Rust
use super::extractors::admin_rpm_key_id;
|
|
use crate::handlers::admin::request::{AdminAppState, AdminRequestContext};
|
|
use crate::GatewayError;
|
|
use axum::{
|
|
body::Body,
|
|
http,
|
|
response::{IntoResponse, Response},
|
|
Json,
|
|
};
|
|
use serde_json::json;
|
|
use std::time::{SystemTime, UNIX_EPOCH};
|
|
|
|
pub(super) async fn maybe_build_local_admin_endpoints_rpm_response(
|
|
state: &AdminAppState<'_>,
|
|
request_context: &AdminRequestContext<'_>,
|
|
) -> Result<Option<Response<Body>>, GatewayError> {
|
|
let Some(decision) = request_context.decision() else {
|
|
return Ok(None);
|
|
};
|
|
|
|
if decision.route_family.as_deref() == Some("endpoints_rpm")
|
|
&& decision.route_kind.as_deref() == Some("key_rpm")
|
|
&& request_context
|
|
.path()
|
|
.starts_with("/api/admin/endpoints/rpm/key/")
|
|
{
|
|
let Some(key_id) = admin_rpm_key_id(request_context.path()) else {
|
|
return Ok(Some(
|
|
(
|
|
http::StatusCode::NOT_FOUND,
|
|
Json(json!({ "detail": "Key 不存在" })),
|
|
)
|
|
.into_response(),
|
|
));
|
|
};
|
|
return Ok(Some(
|
|
match state.build_admin_key_rpm_payload(&key_id).await {
|
|
Some(payload) => Json(payload).into_response(),
|
|
None => (
|
|
http::StatusCode::NOT_FOUND,
|
|
Json(json!({ "detail": format!("Key {key_id} 不存在") })),
|
|
)
|
|
.into_response(),
|
|
},
|
|
));
|
|
}
|
|
|
|
if decision.route_family.as_deref() == Some("endpoints_rpm")
|
|
&& decision.route_kind.as_deref() == Some("reset_key_rpm")
|
|
&& request_context.method() == http::Method::DELETE
|
|
&& request_context
|
|
.path()
|
|
.starts_with("/api/admin/endpoints/rpm/key/")
|
|
{
|
|
let Some(key_id) = admin_rpm_key_id(request_context.path()) else {
|
|
return Ok(Some(
|
|
(
|
|
http::StatusCode::NOT_FOUND,
|
|
Json(json!({ "detail": "Key 不存在" })),
|
|
)
|
|
.into_response(),
|
|
));
|
|
};
|
|
let Some(_key) = state
|
|
.read_provider_catalog_keys_by_ids(std::slice::from_ref(&key_id))
|
|
.await?
|
|
.into_iter()
|
|
.next()
|
|
else {
|
|
return Ok(Some(
|
|
(
|
|
http::StatusCode::NOT_FOUND,
|
|
Json(json!({ "detail": format!("Key {key_id} 不存在") })),
|
|
)
|
|
.into_response(),
|
|
));
|
|
};
|
|
let now_unix_secs = SystemTime::now()
|
|
.duration_since(UNIX_EPOCH)
|
|
.ok()
|
|
.map(|duration| duration.as_secs())
|
|
.unwrap_or(0);
|
|
state.mark_provider_key_rpm_reset(&key_id, now_unix_secs);
|
|
return Ok(Some(
|
|
Json(json!({
|
|
"message": "RPM 计数已重置"
|
|
}))
|
|
.into_response(),
|
|
));
|
|
}
|
|
|
|
Ok(None)
|
|
}
|