use unique_connections in reward_from_db

helium · Dec 12, 2024 · f82f617 · f82f617
1 parent 10aef6b
commit f82f617
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/mobile_verifier/src/cli/reward_from_db.rs b/mobile_verifier/src/cli/reward_from_db.rs
@@ -6,8 +6,7 @@ use crate::{
     rewarder::boosted_hex_eligibility::BoostedHexEligibility,
     sp_boosted_rewards_bans::BannedRadios,
     speedtests_average::SpeedtestAverages,
-    unique_connections::UniqueConnectionCounts,
-    Settings,
+    unique_connections, Settings,
 };
 use anyhow::Result;
 use chrono::NaiveDateTime;
@@ -44,14 +43,16 @@ impl Cmd {
         let speedtest_averages =
             SpeedtestAverages::aggregate_epoch_averages(epoch.end, &pool).await?;
 
+        let unique_connections = unique_connections::db::get(&pool, &epoch).await?;
+
         let reward_shares = CoverageShares::new(
             &pool,
             heartbeats,
             &speedtest_averages,
             &BoostedHexes::default(),
             &BoostedHexEligibility::default(),
             &BannedRadios::default(),
-            &UniqueConnectionCounts::default(),
+            &unique_connections,
             &epoch,
         )
         .await?;