v0.3.11-pre.009

crates/ksp-worker-raw-transaction-ingest-lib/src/snapshot.rs

@@ -1,5 +1,5 @@
 // file: crates/ksp-worker-raw-transaction-ingest-lib/src/snapshot.rs
-// version: 1
+// version: 2
 
 /// Runtime-neutral boxed future resolving to one newer concrete RAW transaction ingest Worker snapshot.
 pub type RawTransactionIngestSnapshotFuture<'a> =
@@ -124,7 +124,7 @@ impl crate::RawTransactionIngestSnapshot {
         return self.source_failure_total;
     }
 
-    /// Returns the number of source-side admission waits explicitly classified as backpressure waits.
+    /// Returns the number of supervisor dequeues that observed the bounded admission queue at full capacity.
     #[must_use]
     pub const fn backpressure_wait_total(&self) -> u64 {
         return self.backpressure_wait_total;
@@ -303,12 +303,19 @@ impl crate::RawTransactionIngestSnapshotPublisher {
         state: ksp_worker_api::WorkerState,
         admission_queue_depth: usize,
         in_flight_persistence: usize,
+        backpressure_wait_observed: bool,
     ) -> ksp_core_lib::Result<()> {
         let admitted_total = match checked_counter(self.snapshot.admitted_total, "admitted_total") {
             std::result::Result::Ok(value) => value,
             std::result::Result::Err(error) => return std::result::Result::Err(error),
         };
+        let backpressure_wait_total =
+            match checked_optional_counter(self.snapshot.backpressure_wait_total, backpressure_wait_observed, "backpressure_wait_total") {
+                std::result::Result::Ok(value) => value,
+                std::result::Result::Err(error) => return std::result::Result::Err(error),
+            };
         self.snapshot.admitted_total = admitted_total;
+        self.snapshot.backpressure_wait_total = backpressure_wait_total;
         return self.publish(state, admission_queue_depth, in_flight_persistence);
     }
 
@@ -318,20 +325,42 @@ impl crate::RawTransactionIngestSnapshotPublisher {
         state: ksp_worker_api::WorkerState,
         admission_queue_depth: usize,
         in_flight_persistence: usize,
+        backpressure_wait_observed: bool,
     ) -> ksp_core_lib::Result<()> {
         let admitted_total = match checked_counter(self.snapshot.admitted_total, "admitted_total") {
             std::result::Result::Ok(value) => value,
             std::result::Result::Err(error) => return std::result::Result::Err(error),
         };
+        let backpressure_wait_total =
+            match checked_optional_counter(self.snapshot.backpressure_wait_total, backpressure_wait_observed, "backpressure_wait_total") {
+                std::result::Result::Ok(value) => value,
+                std::result::Result::Err(error) => return std::result::Result::Err(error),
+            };
         let canonicalized_total = match checked_counter(self.snapshot.canonicalized_total, "canonicalized_total") {
             std::result::Result::Ok(value) => value,
             std::result::Result::Err(error) => return std::result::Result::Err(error),
         };
         self.snapshot.admitted_total = admitted_total;
+        self.snapshot.backpressure_wait_total = backpressure_wait_total;
         self.snapshot.canonicalized_total = canonicalized_total;
         return self.publish(state, admission_queue_depth, in_flight_persistence);
     }
 
+    /// Records one failed private source task without retaining provider-specific error material.
+    pub(crate) fn record_source_failure(
+        &mut self,
+        state: ksp_worker_api::WorkerState,
+        admission_queue_depth: usize,
+        in_flight_persistence: usize,
+    ) -> ksp_core_lib::Result<()> {
+        let source_failure_total = match checked_counter(self.snapshot.source_failure_total, "source_failure_total") {
+            std::result::Result::Ok(value) => value,
+            std::result::Result::Err(error) => return std::result::Result::Err(error),
+        };
+        self.snapshot.source_failure_total = source_failure_total;
+        return self.publish(state, admission_queue_depth, in_flight_persistence);
+    }
+
     /// Records one successful Store persistence outcome and publishes the resulting latest value.
     pub(crate) fn record_persistence_success(
         &mut self,
@@ -458,6 +487,13 @@ fn checked_counter(current: u64, field: &'static str) -> ksp_core_lib::Result<u6
     };
 }
 
+fn checked_optional_counter(current: u64, increment: bool, field: &'static str) -> ksp_core_lib::Result<u64> {
+    if !increment {
+        return std::result::Result::Ok(current);
+    }
+    return checked_counter(current, field);
+}
+
 fn health_for_state(state: ksp_worker_api::WorkerState, previous: ksp_worker_api::WorkerHealth) -> ksp_worker_api::WorkerHealth {
     return match state {
         ksp_worker_api::WorkerState::Running => ksp_worker_api::WorkerHealth::Healthy,