mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-03 01:40:21 +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:
@@ -1,10 +1,10 @@
|
||||
use async_trait::async_trait;
|
||||
use sqlx::{PgPool, Row};
|
||||
|
||||
use super::types::{
|
||||
use super::{
|
||||
ProviderQuotaReadRepository, ProviderQuotaWriteRepository, StoredProviderQuotaSnapshot,
|
||||
};
|
||||
use crate::DataLayerError;
|
||||
use crate::{error::SqlxResultExt, DataLayerError};
|
||||
|
||||
const FIND_BY_PROVIDER_ID_SQL: &str = r#"
|
||||
SELECT
|
||||
@@ -56,7 +56,8 @@ impl ProviderQuotaReadRepository for SqlxProviderQuotaRepository {
|
||||
let row = sqlx::query(FIND_BY_PROVIDER_ID_SQL)
|
||||
.bind(provider_id)
|
||||
.fetch_optional(&self.pool)
|
||||
.await?;
|
||||
.await
|
||||
.map_postgres_err()?;
|
||||
row.as_ref().map(map_row).transpose()
|
||||
}
|
||||
}
|
||||
@@ -69,21 +70,24 @@ impl ProviderQuotaWriteRepository for SqlxProviderQuotaRepository {
|
||||
DataLayerError::InvalidInput("provider quota reset timestamp overflow".to_string())
|
||||
})?)
|
||||
.execute(&self.pool)
|
||||
.await?;
|
||||
.await
|
||||
.map_postgres_err()?;
|
||||
Ok(result.rows_affected() as usize)
|
||||
}
|
||||
}
|
||||
|
||||
fn map_row(row: &sqlx::postgres::PgRow) -> Result<StoredProviderQuotaSnapshot, DataLayerError> {
|
||||
StoredProviderQuotaSnapshot::new(
|
||||
row.try_get("provider_id")?,
|
||||
row.try_get("billing_type")?,
|
||||
row.try_get("monthly_quota_usd")?,
|
||||
row.try_get("monthly_used_usd")?,
|
||||
row.try_get("quota_reset_day")?,
|
||||
row.try_get("quota_last_reset_at_unix_secs")?,
|
||||
row.try_get("quota_expires_at_unix_secs")?,
|
||||
row.try_get("is_active")?,
|
||||
row.try_get("provider_id").map_postgres_err()?,
|
||||
row.try_get("billing_type").map_postgres_err()?,
|
||||
row.try_get("monthly_quota_usd").map_postgres_err()?,
|
||||
row.try_get("monthly_used_usd").map_postgres_err()?,
|
||||
row.try_get("quota_reset_day").map_postgres_err()?,
|
||||
row.try_get("quota_last_reset_at_unix_secs")
|
||||
.map_postgres_err()?,
|
||||
row.try_get("quota_expires_at_unix_secs")
|
||||
.map_postgres_err()?,
|
||||
row.try_get("is_active").map_postgres_err()?,
|
||||
)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user