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

Use core affinity helpers directly

Browse Source
This commit is contained in:
fawney19
2026-04-27 16:13:24 +08:00
parent 51c3beb614
commit 218e73e324
4 changed files with 10 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
apps/aether-gateway/src/scheduler/candidate/affinity.rs
View File

@@ -1,6 +1,5 @@
pub(super) use aether_scheduler_core::{
build_scheduler_affinity_cache_key_for_api_key_id, candidate_affinity_hash, candidate_key,
matches_affinity_target, SchedulerAffinityTarget,
use aether_scheduler_core::{
build_scheduler_affinity_cache_key_for_api_key_id, SchedulerAffinityTarget,
};
use crate::data::auth::GatewayAuthApiKeySnapshot;

5
apps/aether-gateway/src/scheduler/candidate/ranking.rs
View File

@@ -1,13 +1,12 @@
use aether_scheduler_core::{
apply_scheduler_candidate_ranking, effective_provider_key_health_score,
matches_affinity_target, provider_key_health_bucket,
apply_scheduler_candidate_ranking, candidate_affinity_hash,
effective_provider_key_health_score, matches_affinity_target, provider_key_health_bucket,
requested_capability_priority_for_candidate, SchedulerAffinityTarget,
SchedulerRankableCandidate, SchedulerRankingContext, SchedulerRankingMode,
};
use crate::scheduler::config::{SchedulerOrderingConfig, SchedulerSchedulingMode};
use super::affinity::candidate_affinity_hash;
use super::runtime::CandidateRuntimeSelectionSnapshot;
use super::SchedulerMinimalCandidateSelectionCandidate;

3
apps/aether-gateway/src/scheduler/candidate/resolution.rs
View File

@@ -1,8 +1,7 @@
use std::collections::BTreeSet;
use aether_scheduler_core::SchedulerAffinityTarget;
use aether_scheduler_core::{candidate_key, SchedulerAffinityTarget};
use super::affinity::candidate_key;
use super::runtime::{current_candidate_runtime_skip_reason, CandidateRuntimeSelectionSnapshot};
use super::{SchedulerMinimalCandidateSelectionCandidate, SchedulerSkippedCandidate};

10
apps/aether-gateway/src/scheduler/candidate/tests/affinity.rs
View File

@@ -10,10 +10,10 @@ use aether_data_contracts::repository::candidates::{
RequestCandidateStatus, StoredRequestCandidate,
};
use aether_scheduler_core::{
apply_scheduler_candidate_ranking, enumerate_minimal_candidate_selection,
EnumerateMinimalCandidateSelectionInput, SchedulerMinimalCandidateSelectionCandidate,
SchedulerPriorityMode, SchedulerRankableCandidate, SchedulerRankingContext,
SchedulerRankingMode,
apply_scheduler_candidate_ranking, candidate_affinity_hash,
enumerate_minimal_candidate_selection, EnumerateMinimalCandidateSelectionInput,
SchedulerMinimalCandidateSelectionCandidate, SchedulerPriorityMode, SchedulerRankableCandidate,
SchedulerRankingContext, SchedulerRankingMode,
};
use crate::cache::SchedulerAffinityTarget;
@@ -24,7 +24,7 @@ use crate::data::candidate_selection::{
use crate::data::GatewayDataState;
use crate::{AppState, GatewayError};
use super::super::affinity::{build_scheduler_affinity_cache_key, candidate_affinity_hash};
use super::super::affinity::build_scheduler_affinity_cache_key;
use super::super::selection::select_minimal_candidate as select_candidate_impl;
use super::support::{sample_auth_snapshot, sample_key, sample_provider, sample_row};