From f82f617fb5d3c36c9ecddbb322240ba2efe756a2 Mon Sep 17 00:00:00 2001
From: Michael Jeffrey <michaeldjeffrey@gmail.com>
Date: Thu, 12 Dec 2024 13:21:38 -0700
Subject: [PATCH] use unique_connections in reward_from_db

---
 mobile_verifier/src/cli/reward_from_db.rs | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/mobile_verifier/src/cli/reward_from_db.rs b/mobile_verifier/src/cli/reward_from_db.rs
index ef94596cf..b669b4409 100644
--- 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,6 +43,8 @@ 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,
@@ -51,7 +52,7 @@ impl Cmd {
             &BoostedHexes::default(),
             &BoostedHexEligibility::default(),
             &BannedRadios::default(),
-            &UniqueConnectionCounts::default(),
+            &unique_connections,
             &epoch,
         )
         .await?;