Skip to content

Commit f4fb366

Browse files
committed
feat: add sender fee tracker metric
Signed-off-by: Gustavo Inacio <[email protected]>
1 parent 09f645c commit f4fb366

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

tap-agent/src/agent/sender_account.rs

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,12 @@ lazy_static! {
5252
&["sender", "allocation"]
5353
)
5454
.unwrap();
55+
static ref SENDER_FEE_TRACKER: GaugeVec = register_gauge_vec!(
56+
"tap_sender_fee_tracker_grt_total",
57+
"Sender fee tracker metric",
58+
&["sender"]
59+
)
60+
.unwrap();
5561
static ref INVALID_RECEIPT_FEES: GaugeVec = register_gauge_vec!(
5662
"tap_invalid_receipt_fees_grt_total",
5763
"Failed receipt fees",
@@ -325,6 +331,9 @@ impl State {
325331
) {
326332
self.sender_fee_tracker
327333
.update(allocation_id, unaggregated_fees);
334+
SENDER_FEE_TRACKER
335+
.with_label_values(&[&self.sender.to_string()])
336+
.set(self.sender_fee_tracker.get_total_fee() as f64);
328337

329338
UNAGGREGATED_FEES
330339
.with_label_values(&[&self.sender.to_string(), &allocation_id.to_string()])
@@ -727,6 +736,10 @@ impl Actor for SenderAccount {
727736
state
728737
.sender_fee_tracker
729738
.add(allocation_id, value, timestamp_ns);
739+
740+
SENDER_FEE_TRACKER
741+
.with_label_values(&[&state.sender.to_string()])
742+
.set(state.sender_fee_tracker.get_total_fee() as f64);
730743
UNAGGREGATED_FEES
731744
.with_label_values(&[
732745
&state.sender.to_string(),
@@ -972,6 +985,13 @@ impl Actor for SenderAccount {
972985
// remove from sender_fee_tracker
973986
state.sender_fee_tracker.remove(allocation_id);
974987

988+
SENDER_FEE_TRACKER
989+
.with_label_values(&[&state.sender.to_string()])
990+
.set(state.sender_fee_tracker.get_total_fee() as f64);
991+
992+
let _ = UNAGGREGATED_FEES
993+
.remove_label_values(&[&state.sender.to_string(), &allocation_id.to_string()]);
994+
975995
// check for deny conditions
976996
let _ = myself.cast(SenderAccountMessage::UpdateReceiptFees(
977997
allocation_id,

0 commit comments

Comments
 (0)