refactor: generic tracker and fee tracker

Signed-off-by: Gustavo Inacio <[email protected]>

Author: gusinacio

tap-agent/src/agent/sender_account.rs

@@ -27,6 +27,7 @@ use tap_core::rav::SignedRAV;
 use tracing::{error, Level};
 
 use super::sender_allocation::{SenderAllocation, SenderAllocationArgs};
+use super::sender_fee_tracker::{BufferedReceiptFee, DurationInfo};
 use crate::agent::sender_allocation::SenderAllocationMessage;
 use crate::agent::sender_fee_tracker::SenderFeeTracker;
 use crate::agent::unaggregated_receipts::UnaggregatedReceipts;
@@ -97,7 +98,7 @@ pub enum SenderAccountMessage {
     UpdateInvalidReceiptFees(Address, UnaggregatedReceipts),
     UpdateRav(SignedRAV),
     #[cfg(test)]
-    GetSenderFeeTracker(ractor::RpcReplyPort<SenderFeeTracker>),
+    GetSenderFeeTracker(ractor::RpcReplyPort<SenderFeeTracker<BufferedReceiptFee, DurationInfo>>),
     #[cfg(test)]
     GetDeny(ractor::RpcReplyPort<bool>),
     #[cfg(test)]
@@ -131,9 +132,9 @@ pub struct SenderAccountArgs {
 }
 pub struct State {
     prefix: Option<String>,
-    sender_fee_tracker: SenderFeeTracker,
-    rav_tracker: SenderFeeTracker,
-    invalid_receipts_tracker: SenderFeeTracker,
+    sender_fee_tracker: SenderFeeTracker<BufferedReceiptFee, DurationInfo>,
+    rav_tracker: SenderFeeTracker<u128>,
+    invalid_receipts_tracker: SenderFeeTracker<u128>,
     allocation_ids: HashSet<Address>,
     _indexer_allocations_handle: JoinHandle<()>,
     _escrow_account_monitor: PipeHandle,
@@ -471,6 +472,7 @@ impl Actor for SenderAccount {
             sender_fee_tracker: SenderFeeTracker::new(Duration::from_millis(
                 config.tap.rav_request_timestamp_buffer_ms,
             )),
+            // sender_fee_tracker: SenderFeeTracker::new(),
             rav_tracker: SenderFeeTracker::default(),
             invalid_receipts_tracker: SenderFeeTracker::default(),
             allocation_ids: allocation_ids.clone(),
@@ -584,7 +586,7 @@ impl Actor for SenderAccount {
                         state.sender_fee_tracker.finish_rav_request(allocation_id);
                         match rav_result {
                             Ok((fees, rav)) => {
-                                state.rav_tracker.ok_rav_request(allocation_id);
+                                state.sender_fee_tracker.ok_rav_request(allocation_id);
 
                                 let rav_value = rav.map_or(0, |rav| rav.message.valueAggregate);
                                 // update rav tracker
@@ -610,7 +612,7 @@ impl Actor for SenderAccount {
                                     .set(fees.value as f64);
                             }
                             Err(err) => {
-                                state.rav_tracker.failed_rav_backoff(allocation_id);
+                                state.sender_fee_tracker.failed_rav_backoff(allocation_id);
                                 error!(
                                     "Error while requesting RAV for sender {} and allocation {}: {}",
                                     state.sender,