feat(gateway): unify background task runtime and storage

2026-09-03 01:40:21 +08:00 · 2026-05-09 21:19:29 +08:00
parent 4a64d078f3
commit 209322b499
64 changed files with 4602 additions and 280 deletions
--- a/apps/aether-gateway/src/data/state/core.rs
+++ b/apps/aether-gateway/src/data/state/core.rs
@@ -34,6 +34,8 @@ impl GatewayDataState {
                proxy_node_reader: None,
                proxy_node_writer: None,
                billing_reader: None,
+                background_task_reader: None,
+                background_task_writer: None,
                gemini_file_mapping_reader: None,
                gemini_file_mapping_writer: None,
                global_model_reader: None,
@@ -73,6 +75,8 @@ impl GatewayDataState {
        let proxy_node_reader = backends.read().proxy_nodes();
        let proxy_node_writer = backends.write().proxy_nodes();
        let billing_reader = backends.read().billing();
+        let background_task_reader = backends.read().background_tasks();
+        let background_task_writer = backends.write().background_tasks();
        let gemini_file_mapping_reader = backends.read().gemini_file_mappings();
        let global_model_reader = backends.read().global_models();
        let global_model_writer = backends.write().global_models();
@@ -110,6 +114,8 @@ impl GatewayDataState {
            proxy_node_reader,
            proxy_node_writer,
            billing_reader,
+            background_task_reader,
+            background_task_writer,
            gemini_file_mapping_reader,
            gemini_file_mapping_writer,
            global_model_reader,
@@ -197,6 +203,14 @@ impl GatewayDataState {
        self.announcement_writer.is_some()
    }

+    pub(crate) fn has_background_task_reader(&self) -> bool {
+        self.background_task_reader.is_some()
+    }
+
+    pub(crate) fn has_background_task_writer(&self) -> bool {
+        self.background_task_writer.is_some()
+    }
+
    pub(crate) fn has_audit_log_reader(&self) -> bool {
        self.backends
            .as_ref()
--- a/apps/aether-gateway/src/data/state/mod.rs
+++ b/apps/aether-gateway/src/data/state/mod.rs
@@ -81,6 +81,11 @@ use aether_data::{
    DataBackends, DataLayerError, DatabaseMaintenanceSummary, WalletDailyUsageAggregationInput,
    WalletDailyUsageAggregationResult,
 };
+use aether_data_contracts::repository::background_tasks::{
+    BackgroundTaskListQuery, BackgroundTaskReadRepository, BackgroundTaskSummary,
+    BackgroundTaskWriteRepository, StoredBackgroundTaskEvent, StoredBackgroundTaskRun,
+    StoredBackgroundTaskRunPage, UpsertBackgroundTaskEvent, UpsertBackgroundTaskRun,
+};
 use aether_data_contracts::repository::billing::{
    AdminBillingCollectorRecord, AdminBillingCollectorWriteInput, AdminBillingMutationOutcome,
    AdminBillingPresetApplyResult, AdminBillingRuleRecord, AdminBillingRuleWriteInput,
@@ -140,6 +145,8 @@ pub(crate) struct GatewayDataState {
    proxy_node_reader: Option<Arc<dyn ProxyNodeReadRepository>>,
    proxy_node_writer: Option<Arc<dyn ProxyNodeWriteRepository>>,
    billing_reader: Option<Arc<dyn BillingReadRepository>>,
+    background_task_reader: Option<Arc<dyn BackgroundTaskReadRepository>>,
+    background_task_writer: Option<Arc<dyn BackgroundTaskWriteRepository>>,
    gemini_file_mapping_reader: Option<Arc<dyn GeminiFileMappingReadRepository>>,
    gemini_file_mapping_writer: Option<Arc<dyn GeminiFileMappingWriteRepository>>,
    global_model_reader: Option<Arc<dyn GlobalModelReadRepository>>,
@@ -206,6 +213,14 @@ impl fmt::Debug for GatewayDataState {
            .field("has_proxy_node_reader", &self.proxy_node_reader.is_some())
            .field("has_proxy_node_writer", &self.proxy_node_writer.is_some())
            .field("has_billing_reader", &self.billing_reader.is_some())
+            .field(
+                "has_background_task_reader",
+                &self.background_task_reader.is_some(),
+            )
+            .field(
+                "has_background_task_writer",
+                &self.background_task_writer.is_some(),
+            )
            .field(
                "has_gemini_file_mapping_reader",
                &self.gemini_file_mapping_reader.is_some(),
--- a/apps/aether-gateway/src/data/state/runtime.rs
+++ b/apps/aether-gateway/src/data/state/runtime.rs
@@ -5,11 +5,12 @@ use super::{
    AdminBillingRuleWriteInput, AdminPaymentOrderListQuery, AdminRedeemCodeBatchListQuery,
    AdminRedeemCodeListQuery, AdminWalletLedgerQuery, AdminWalletListQuery,
    AdminWalletRefundRequestListQuery, AnnouncementListQuery, AuditLogListQuery,
-    CompleteAdminWalletRefundInput, CreateAdminRedeemCodeBatchInput,
-    CreateAdminRedeemCodeBatchResult, CreateAnnouncementRecord, CreateManualWalletRechargeInput,
-    CreateWalletRechargeOrderInput, CreateWalletRechargeOrderOutcome,
-    CreateWalletRefundRequestInput, CreateWalletRefundRequestOutcome, CreditAdminPaymentOrderInput,
-    DataLayerError, DatabaseMaintenanceSummary, DecisionTrace, DeleteAdminRedeemCodeBatchInput,
+    BackgroundTaskListQuery, BackgroundTaskSummary, CompleteAdminWalletRefundInput,
+    CreateAdminRedeemCodeBatchInput, CreateAdminRedeemCodeBatchResult, CreateAnnouncementRecord,
+    CreateManualWalletRechargeInput, CreateWalletRechargeOrderInput,
+    CreateWalletRechargeOrderOutcome, CreateWalletRefundRequestInput,
+    CreateWalletRefundRequestOutcome, CreditAdminPaymentOrderInput, DataLayerError,
+    DatabaseMaintenanceSummary, DecisionTrace, DeleteAdminRedeemCodeBatchInput,
    DisableAdminRedeemCodeBatchInput, DisableAdminRedeemCodeInput, FailAdminWalletRefundInput,
    GatewayDataState, GatewayProviderTransportSnapshot, LocalVideoTaskReadResponse,
    ProcessAdminWalletRefundInput, ProcessPaymentCallbackInput, ProcessPaymentCallbackOutcome,
@@ -19,15 +20,16 @@ use super::{
    StoredAdminRedeemCodePage, StoredAdminWalletLedgerPage, StoredAdminWalletListPage,
    StoredAdminWalletRefund, StoredAdminWalletRefundPage, StoredAdminWalletRefundRequestPage,
    StoredAdminWalletTransaction, StoredAdminWalletTransactionPage, StoredAnnouncement,
-    StoredAnnouncementPage, StoredBillingModelContext, StoredProviderQuotaSnapshot,
+    StoredAnnouncementPage, StoredBackgroundTaskEvent, StoredBackgroundTaskRun,
+    StoredBackgroundTaskRunPage, StoredBillingModelContext, StoredProviderQuotaSnapshot,
    StoredProviderUsageSummary, StoredRequestUsageAudit, StoredSuspiciousActivity,
    StoredUsageSettlement, StoredUserAuditLogPage, StoredUserAuthRecord, StoredUserExportRow,
    StoredUserSummary, StoredVideoTask, StoredWalletDailyUsageLedger,
    StoredWalletDailyUsageLedgerPage, StoredWalletSnapshot, UpdateAnnouncementRecord,
-    UpsertUsageRecord, UpsertVideoTask, UsageSettlementInput, VideoTaskLookupKey,
-    VideoTaskModelCount, VideoTaskQueryFilter, VideoTaskStatusCount,
-    WalletDailyUsageAggregationInput, WalletDailyUsageAggregationResult, WalletLookupKey,
-    WalletMutationOutcome,
+    UpsertBackgroundTaskEvent, UpsertBackgroundTaskRun, UpsertUsageRecord, UpsertVideoTask,
+    UsageSettlementInput, VideoTaskLookupKey, VideoTaskModelCount, VideoTaskQueryFilter,
+    VideoTaskStatusCount, WalletDailyUsageAggregationInput, WalletDailyUsageAggregationResult,
+    WalletLookupKey, WalletMutationOutcome,
 };
 use aether_data_contracts::repository::usage::{
    PendingUsageCleanupSummary, ProviderApiKeyWindowUsageRequest,
@@ -1716,6 +1718,82 @@ impl GatewayDataState {
    ) -> Result<Option<LocalVideoTaskReadResponse>, DataLayerError> {
        read_data_backed_video_task_response(self, route_family, request_path).await
    }
+
+    pub(crate) async fn find_background_task_run(
+        &self,
+        run_id: &str,
+    ) -> Result<Option<StoredBackgroundTaskRun>, DataLayerError> {
+        match &self.background_task_reader {
+            Some(repository) => repository.find_run(run_id).await,
+            None => Ok(None),
+        }
+    }
+
+    pub(crate) async fn list_background_task_runs(
+        &self,
+        query: &BackgroundTaskListQuery,
+    ) -> Result<StoredBackgroundTaskRunPage, DataLayerError> {
+        match &self.background_task_reader {
+            Some(repository) => repository.list_runs(query).await,
+            None => Ok(StoredBackgroundTaskRunPage::default()),
+        }
+    }
+
+    pub(crate) async fn list_background_task_events(
+        &self,
+        run_id: &str,
+        offset: usize,
+        limit: usize,
+    ) -> Result<Vec<StoredBackgroundTaskEvent>, DataLayerError> {
+        match &self.background_task_reader {
+            Some(repository) => repository.list_events(run_id, offset, limit).await,
+            None => Ok(Vec::new()),
+        }
+    }
+
+    pub(crate) async fn summarize_background_task_runs(
+        &self,
+    ) -> Result<BackgroundTaskSummary, DataLayerError> {
+        match &self.background_task_reader {
+            Some(repository) => repository.summarize_runs().await,
+            None => Ok(BackgroundTaskSummary::default()),
+        }
+    }
+
+    pub(crate) async fn upsert_background_task_run(
+        &self,
+        run: UpsertBackgroundTaskRun,
+    ) -> Result<Option<StoredBackgroundTaskRun>, DataLayerError> {
+        match &self.background_task_writer {
+            Some(repository) => repository.upsert_run(run).await.map(Some),
+            None => Ok(None),
+        }
+    }
+
+    pub(crate) async fn request_cancel_background_task_run(
+        &self,
+        run_id: &str,
+        updated_at_unix_secs: u64,
+    ) -> Result<bool, DataLayerError> {
+        match &self.background_task_writer {
+            Some(repository) => {
+                repository
+                    .request_cancel(run_id, updated_at_unix_secs)
+                    .await
+            }
+            None => Ok(false),
+        }
+    }
+
+    pub(crate) async fn upsert_background_task_event(
+        &self,
+        event: UpsertBackgroundTaskEvent,
+    ) -> Result<Option<StoredBackgroundTaskEvent>, DataLayerError> {
+        match &self.background_task_writer {
+            Some(repository) => repository.upsert_event(event).await.map(Some),
+            None => Ok(None),
+        }
+    }
 }

 #[cfg(test)]
--- a/apps/aether-gateway/src/data/state/testing/announcements.rs
+++ b/apps/aether-gateway/src/data/state/testing/announcements.rs
@@ -43,6 +43,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -93,6 +95,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
--- a/apps/aether-gateway/src/data/state/testing/mod.rs
+++ b/apps/aether-gateway/src/data/state/testing/mod.rs
@@ -76,6 +76,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -125,6 +127,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -170,6 +174,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -303,6 +309,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -364,6 +372,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -418,6 +428,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -481,6 +493,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -526,6 +540,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -572,6 +588,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -629,6 +647,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -688,6 +708,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -731,6 +753,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -789,6 +813,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -840,6 +866,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: Some(wallet_reader),
            wallet_writer: Some(wallet_writer),
            settlement_writer: None,
@@ -896,6 +924,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: Some(wallet_reader),
            wallet_writer: Some(wallet_writer),
            settlement_writer: None,
@@ -953,6 +983,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: Some(wallet_reader),
            wallet_writer: Some(wallet_writer),
            settlement_writer: None,
@@ -1009,6 +1041,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1054,6 +1088,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1099,6 +1135,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1156,6 +1194,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1218,6 +1258,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1263,6 +1305,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1313,6 +1357,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1380,6 +1426,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1442,6 +1490,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1488,6 +1538,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1534,6 +1586,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1582,6 +1636,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1628,6 +1684,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1674,6 +1732,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1728,6 +1788,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1783,6 +1845,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1841,6 +1905,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1905,6 +1971,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -1970,6 +2038,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -2039,6 +2109,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -2115,6 +2187,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: Some(wallet_reader),
            wallet_writer: Some(wallet_writer),
            settlement_writer: None,
@@ -2173,6 +2247,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -2222,6 +2298,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -2267,6 +2345,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -2318,6 +2398,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: Some(wallet_reader),
            wallet_writer: Some(wallet_writer),
            settlement_writer: None,
@@ -2373,6 +2455,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: Some(wallet_reader),
            wallet_writer: Some(wallet_writer),
            settlement_writer: None,
@@ -2429,6 +2513,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: Some(wallet_reader),
            wallet_writer: Some(wallet_writer),
            settlement_writer: None,
@@ -2478,6 +2564,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: None,
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
--- a/apps/aether-gateway/src/data/state/testing/video_tasks.rs
+++ b/apps/aether-gateway/src/data/state/testing/video_tasks.rs
@@ -47,6 +47,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: Some(repository),
            video_task_writer: None,
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -99,6 +101,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: Some(video_task_reader),
            video_task_writer: Some(video_task_writer),
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -148,6 +152,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: Some(video_task_reader),
            video_task_writer: Some(video_task_writer),
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -201,6 +207,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: Some(video_task_reader),
            video_task_writer: Some(video_task_writer),
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -258,6 +266,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: Some(video_task_reader),
            video_task_writer: Some(video_task_writer),
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,
@@ -324,6 +334,8 @@ impl GatewayDataState {
            usage_worker_queue: None,
            video_task_reader: Some(video_task_reader),
            video_task_writer: Some(video_task_writer),
+            background_task_reader: None,
+            background_task_writer: None,
            wallet_reader: None,
            wallet_writer: None,
            settlement_writer: None,