feat: add routing profile scheduling policies

2026-10-09 18:59:50 +08:00 · 2026-05-18 11:03:49 +08:00
parent a2f91b4108
commit 92813e6122
124 changed files with 11681 additions and 578 deletions
@@ -0,0 +1,131 @@
+use aether_data_contracts::repository::routing_profiles::{
+    CreateRoutingGroupBindingRecord, CreateRoutingGroupRecord, CreateRoutingGroupVersionRecord,
+    RoutingGroupBindingQuery, RoutingGroupLookupKey, RoutingGroupReadRepository,
+    StoredRoutingGroup, StoredRoutingGroupBinding, StoredRoutingGroupVersion,
+    UpdateRoutingGroupBindingRecord, UpdateRoutingGroupRecord,
+};
+use std::sync::Arc;
+
+use super::{DataLayerError, GatewayDataState};
+
+impl GatewayDataState {
+    pub(crate) fn routing_group_read_repository(
+        &self,
+    ) -> Option<Arc<dyn RoutingGroupReadRepository>> {
+        self.routing_group_reader.clone()
+    }
+
+    pub(crate) async fn list_routing_groups(
+        &self,
+    ) -> Result<Vec<StoredRoutingGroup>, DataLayerError> {
+        match &self.routing_group_reader {
+            Some(repository) => repository.list_routing_groups().await,
+            None => Ok(Vec::new()),
+        }
+    }
+
+    pub(crate) async fn find_routing_group(
+        &self,
+        lookup: RoutingGroupLookupKey<'_>,
+    ) -> Result<Option<StoredRoutingGroup>, DataLayerError> {
+        match &self.routing_group_reader {
+            Some(repository) => repository.find_routing_group(lookup).await,
+            None => Ok(None),
+        }
+    }
+
+    pub(crate) async fn list_routing_group_bindings(
+        &self,
+        query: &RoutingGroupBindingQuery,
+    ) -> Result<Vec<StoredRoutingGroupBinding>, DataLayerError> {
+        match &self.routing_group_reader {
+            Some(repository) => repository.list_routing_group_bindings(query).await,
+            None => Ok(Vec::new()),
+        }
+    }
+
+    pub(crate) async fn list_routing_group_versions(
+        &self,
+        group_id: &str,
+    ) -> Result<Vec<StoredRoutingGroupVersion>, DataLayerError> {
+        match &self.routing_group_reader {
+            Some(repository) => repository.list_routing_group_versions(group_id).await,
+            None => Ok(Vec::new()),
+        }
+    }
+
+    pub(crate) async fn create_routing_group(
+        &self,
+        record: CreateRoutingGroupRecord,
+    ) -> Result<Option<StoredRoutingGroup>, DataLayerError> {
+        match &self.routing_group_writer {
+            Some(repository) => repository.create_routing_group(record).await.map(Some),
+            None => Ok(None),
+        }
+    }
+
+    pub(crate) async fn update_routing_group(
+        &self,
+        id: &str,
+        patch: UpdateRoutingGroupRecord,
+    ) -> Result<Option<StoredRoutingGroup>, DataLayerError> {
+        match &self.routing_group_writer {
+            Some(repository) => repository.update_routing_group(id, patch).await,
+            None => Ok(None),
+        }
+    }
+
+    pub(crate) async fn delete_routing_group(&self, id: &str) -> Result<bool, DataLayerError> {
+        match &self.routing_group_writer {
+            Some(repository) => repository.delete_routing_group(id).await,
+            None => Ok(false),
+        }
+    }
+
+    pub(crate) async fn create_routing_group_binding(
+        &self,
+        record: CreateRoutingGroupBindingRecord,
+    ) -> Result<Option<StoredRoutingGroupBinding>, DataLayerError> {
+        match &self.routing_group_writer {
+            Some(repository) => repository
+                .create_routing_group_binding(record)
+                .await
+                .map(Some),
+            None => Ok(None),
+        }
+    }
+
+    pub(crate) async fn update_routing_group_binding(
+        &self,
+        id: &str,
+        patch: UpdateRoutingGroupBindingRecord,
+    ) -> Result<Option<StoredRoutingGroupBinding>, DataLayerError> {
+        match &self.routing_group_writer {
+            Some(repository) => repository.update_routing_group_binding(id, patch).await,
+            None => Ok(None),
+        }
+    }
+
+    pub(crate) async fn delete_routing_group_binding(
+        &self,
+        id: &str,
+    ) -> Result<bool, DataLayerError> {
+        match &self.routing_group_writer {
+            Some(repository) => repository.delete_routing_group_binding(id).await,
+            None => Ok(false),
+        }
+    }
+
+    pub(crate) async fn create_routing_group_version(
+        &self,
+        record: CreateRoutingGroupVersionRecord,
+    ) -> Result<Option<StoredRoutingGroupVersion>, DataLayerError> {
+        match &self.routing_group_writer {
+            Some(repository) => repository
+                .create_routing_group_version(record)
+                .await
+                .map(Some),
+            None => Ok(None),
+        }
+    }
+}