i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-02 17:30:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: 抽离 AI pipeline 与调度共享能力逻辑

Browse Source
This commit is contained in:
fawney19
2026-04-10 01:46:14 +08:00
parent b901a6ffc7
commit 5014e2f5fd
255 changed files with 15057 additions and 3115 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
apps/aether-gateway/src/maintenance/tests.rs
View File

@@ -34,17 +34,17 @@ async fn start_server(app: Router) -> (String, tokio::task::JoinHandle<()>) {
#[tokio::test]
async fn gateway_background_request_candidate_cleanup_deletes_expired_entries_in_batches() {
fn now_unix_secs() -> u64 {
fn now_unix_ms() -> u64 {
std::time::SystemTime::now()
.duration_since(std::time::UNIX_EPOCH)
.unwrap_or_default()
.as_secs()
.as_millis() as u64
}
async fn seed_candidate(
repository: &InMemoryRequestCandidateRepository,
id: &str,
created_at_unix_secs: u64,
created_at_unix_ms: u64,
) {
repository
.upsert(UpsertRequestCandidateRecord {
@@ -69,18 +69,18 @@ async fn gateway_background_request_candidate_cleanup_deletes_expired_entries_in
concurrent_requests: Some(1),
extra_data: None,
required_capabilities: None,
created_at_unix_secs: Some(created_at_unix_secs),
started_at_unix_secs: Some(created_at_unix_secs),
finished_at_unix_secs: Some(created_at_unix_secs.saturating_add(1)),
created_at_unix_ms: Some(created_at_unix_ms),
started_at_unix_ms: Some(created_at_unix_ms),
finished_at_unix_ms: Some(created_at_unix_ms.saturating_add(1)),
})
.await
.expect("candidate should seed");
}
let repository = Arc::new(InMemoryRequestCandidateRepository::default());
seed_candidate(&repository, "cand-expired-1", 1).await;
seed_candidate(&repository, "cand-expired-2", 2).await;
seed_candidate(&repository, "cand-active", now_unix_secs()).await;
seed_candidate(&repository, "cand-expired-1", 1_000).await;
seed_candidate(&repository, "cand-expired-2", 2_000).await;
seed_candidate(&repository, "cand-active", now_unix_ms()).await;
let data_state = crate::data::GatewayDataState::with_request_candidate_repository_for_tests(
Arc::clone(&repository),