feat: [#1358] inject Swarms into InMemoryTorrentRepository in production code

todo: do the same for testing code.

Author: josecelano

packages/torrent-repository/src/swarms.rs

@@ -9,7 +9,7 @@ use torrust_tracker_primitives::swarm_metadata::{AggregateSwarmMetadata, SwarmMe
 use torrust_tracker_primitives::{peer, DurationSinceUnixEpoch, PersistentTorrent, PersistentTorrents};
 
 use crate::swarm::Swarm;
-use crate::SwarmHandle;
+use crate::{SwarmHandle, TORRENT_REPOSITORY_LOG_TARGET};
 
 #[derive(Default, Debug)]
 pub struct Swarms {
@@ -43,6 +43,8 @@ impl Swarms {
         peer: &peer::Peer,
         opt_persistent_torrent: Option<PersistentTorrent>,
     ) -> Result<bool, Error> {
+        tracing::trace!(target: TORRENT_REPOSITORY_LOG_TARGET, "Handling announcement for torrent: {info_hash}");
+
         let swarm_handle = if let Some(number_of_downloads) = opt_persistent_torrent {
             SwarmHandle::new(Swarm::new(number_of_downloads).into())
         } else {

packages/tracker-core/src/container.rs

@@ -1,6 +1,8 @@
 use std::sync::Arc;
 
 use torrust_tracker_configuration::Core;
+use torrust_tracker_torrent_repository::container::TorrentRepositoryContainer;
+use torrust_tracker_torrent_repository::Swarms;
 
 use crate::announce_handler::AnnounceHandler;
 use crate::authentication::handler::KeysHandler;
@@ -35,8 +37,19 @@ pub struct TrackerCoreContainer {
 }
 
 impl TrackerCoreContainer {
+    #[must_use]
+    pub fn initialize_from(core_config: &Arc<Core>, torrent_repository_container: &Arc<TorrentRepositoryContainer>) -> Self {
+        Self::inner_initialize(core_config, &torrent_repository_container.swarms)
+    }
+
     #[must_use]
     pub fn initialize(core_config: &Arc<Core>) -> Self {
+        let swarms = Arc::new(Swarms::default());
+        Self::inner_initialize(core_config, &swarms)
+    }
+
+    #[must_use]
+    fn inner_initialize(core_config: &Arc<Core>, swarms: &Arc<Swarms>) -> Self {
         let database = initialize_database(core_config);
         let in_memory_whitelist = Arc::new(InMemoryWhitelist::default());
         let whitelist_authorization = Arc::new(WhitelistAuthorization::new(core_config, &in_memory_whitelist.clone()));
@@ -48,7 +61,7 @@ impl TrackerCoreContainer {
             &db_key_repository.clone(),
             &in_memory_key_repository.clone(),
         ));
-        let in_memory_torrent_repository = Arc::new(InMemoryTorrentRepository::default());
+        let in_memory_torrent_repository = Arc::new(InMemoryTorrentRepository::new(swarms.clone()));
         let db_torrent_repository = Arc::new(DatabasePersistentTorrentRepository::new(&database));
 
         let torrents_manager = Arc::new(TorrentsManager::new(

packages/tracker-core/src/torrent/repository/in_memory.rs

@@ -25,6 +25,11 @@ pub struct InMemoryTorrentRepository {
 }
 
 impl InMemoryTorrentRepository {
+    #[must_use]
+    pub fn new(swarms: Arc<Swarms>) -> Self {
+        Self { swarms }
+    }
+
     /// Inserts or updates a peer in the torrent entry corresponding to the
     /// given infohash.
     ///

src/container.rs

@@ -64,7 +64,10 @@ impl AppContainer {
 
         // Core
 
-        let tracker_core_container = Arc::new(TrackerCoreContainer::initialize(&core_config));
+        let tracker_core_container = Arc::new(TrackerCoreContainer::initialize_from(
+            &core_config,
+            &torrent_repository_container,
+        ));
 
         // HTTP