fix: fix local stake

graphops · Sep 14, 2023 · 4cb090a · 4cb090a
1 parent 9381d92
commit 4cb090a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/subgraph-radio/src/server/model/mod.rs b/subgraph-radio/src/server/model/mod.rs
@@ -154,8 +154,11 @@ pub fn calc_ratios(
     let mut stakes: Vec<String> = Vec::new();
     let mut senders: Vec<String> = Vec::new();
 
+    let mut matched_once = false;
+
     for att in temp_attestations.iter() {
         if local_ppoi.is_some() && att.ppoi.as_str() == local_ppoi.unwrap() {
+            matched_once = true;
             stakes.push(format!("{}*", att.stake_weight + local_stake as i64));
             senders.push(format!("{}*", att.senders.len() + 1));
         } else {
@@ -166,6 +169,10 @@ pub fn calc_ratios(
     }
 
     // Add zeros at the end if there is no local attestation
+    if local_ppoi.is_some() && !matched_once {
+        stakes.push(format!("{}*", local_stake));
+    }
+
     if local_ppoi.is_none() {
         stakes.push("0*".to_string());
         senders.push("0*".to_string());