refactor: use StatsEventSender to send events

josecelano · josecelano · commit 720a5841c943 · 2022-10-24T16:39:19.000+01:00
instead of StatsTracker.
diff --git a/src/tracker/tracker.rs b/src/tracker/tracker.rs
@@ -25,15 +25,15 @@ pub struct TorrentTracker {
     whitelist: RwLock<std::collections::HashSet<InfoHash>>,
     torrents: RwLock<std::collections::BTreeMap<InfoHash, TorrentEntry>>,
     stats_tracker: Box<dyn TrackerStatsService>,
-    _stats_event_sender: Option<Box<dyn TrackerStatisticsEventSender>>,
+    stats_event_sender: Option<Box<dyn TrackerStatisticsEventSender>>,
     database: Box<dyn Database>,
 }
 
 impl TorrentTracker {
     pub fn new(
         config: Arc<Configuration>,
         stats_tracker: Box<dyn TrackerStatsService>,
-        _stats_event_sender: Option<Box<dyn TrackerStatisticsEventSender>>,
+        stats_event_sender: Option<Box<dyn TrackerStatisticsEventSender>>,
     ) -> Result<TorrentTracker, r2d2::Error> {
         let database = database::connect_database(&config.db_driver, &config.db_path)?;
 
@@ -44,7 +44,7 @@ impl TorrentTracker {
             whitelist: RwLock::new(std::collections::HashSet::new()),
             torrents: RwLock::new(std::collections::BTreeMap::new()),
             stats_tracker,
-            _stats_event_sender,
+            stats_event_sender,
             database,
         })
     }
@@ -242,7 +242,10 @@ impl TorrentTracker {
     }
 
     pub async fn send_stats_event(&self, event: TrackerStatisticsEvent) -> Option<Result<(), SendError<TrackerStatisticsEvent>>> {
-        self.stats_tracker.send_event(event).await
+        match &self.stats_event_sender {
+            None => None,
+            Some(stats_event_sender) => stats_event_sender.send_event(event).await,
+        }
     }
 
     // Remove inactive peers and (optionally) peerless torrents
