wip

Author: carlosvdr

tap-agent/src/backoff.rs

@@ -0,0 +1,35 @@
+use std::time::{Duration, Instant};
+
+#[derive(Debug, Clone)]
+pub struct BackoffInfo {
+    failed_count: u32,
+    failed_backoff_time: Instant,
+}
+
+impl BackoffInfo {
+    pub fn ok(&mut self) {
+        self.failed_count = 0;
+    }
+
+    pub fn fail(&mut self) {
+        // backoff = max(100ms * 2 ^ retries, 60s)
+        self.failed_backoff_time = Instant::now()
+            + (Duration::from_millis(100) * 2u32.pow(self.failed_count))
+                .min(Duration::from_secs(60));
+        self.failed_count += 1;
+    }
+
+    pub fn in_backoff(&self) -> bool {
+        let now = Instant::now();
+        now < self.failed_backoff_time
+    }
+}
+
+impl Default for BackoffInfo {
+    fn default() -> Self {
+        Self {
+            failed_count: 0,
+            failed_backoff_time: Instant::now(),
+        }
+    }
+}

tap-agent/src/lib.rs

@@ -16,6 +16,7 @@ lazy_static! {
 }
 
 pub mod agent;
+pub mod backoff;
 pub mod config;
 pub mod database;
 pub mod metrics;

tap-agent/src/tracker/sender_fee_stats.rs

@@ -3,10 +3,10 @@
 
 use std::{
     collections::VecDeque,
-    time::{Duration, Instant, SystemTime, UNIX_EPOCH},
+    time::{Duration, SystemTime, UNIX_EPOCH},
 };
 
-use crate::agent::unaggregated_receipts::UnaggregatedReceipts;
+use crate::{agent::unaggregated_receipts::UnaggregatedReceipts, backoff::BackoffInfo};
 
 use super::{AllocationStats, DefaultFromExtra, DurationInfo};
 
@@ -84,31 +84,6 @@ impl BufferInfo {
     }
 }
 
-#[derive(Debug, Clone)]
-pub struct BackoffInfo {
-    failed_ravs_count: u32,
-    failed_rav_backoff_time: Instant,
-}
-
-impl BackoffInfo {
-    pub fn ok(&mut self) {
-        self.failed_ravs_count = 0;
-    }
-
-    pub fn fail(&mut self) {
-        // backoff = max(100ms * 2 ^ retries, 60s)
-        self.failed_rav_backoff_time = Instant::now()
-            + (Duration::from_millis(100) * 2u32.pow(self.failed_ravs_count))
-                .min(Duration::from_secs(60));
-        self.failed_ravs_count += 1;
-    }
-
-    pub fn in_backoff(&self) -> bool {
-        let now = Instant::now();
-        now < self.failed_rav_backoff_time
-    }
-}
-
 impl DefaultFromExtra<DurationInfo> for SenderFeeStats {
     fn default_from_extra(extra: &DurationInfo) -> Self {
         SenderFeeStats {
@@ -121,15 +96,6 @@ impl DefaultFromExtra<DurationInfo> for SenderFeeStats {
     }
 }
 
-impl Default for BackoffInfo {
-    fn default() -> Self {
-        Self {
-            failed_ravs_count: 0,
-            failed_rav_backoff_time: Instant::now(),
-        }
-    }
-}
-
 impl AllocationStats<UnaggregatedReceipts> for SenderFeeStats {
     fn update(&mut self, v: UnaggregatedReceipts) {
         self.total_fee = v.value;