fix: [#84] restore behavior. Update torrent stats after upload

josecelano · josecelano · commit c29155752601 · 2022-11-30T08:39:26.000Z
When a new torrent is uploaded we have to update the tracker torrent
stats in the `torrust_torrent_tracker_stats` table.

It was from the UI but not for the DB migration script becuase the
frontend makes a request to get the torrent info after the upload and
that endpint updates the torrent tracket stats.

It must update the stats even if that endpoint is not called after
uploading a new torrent.
diff --git a/src/routes/torrent.rs b/src/routes/torrent.rs
@@ -96,6 +96,12 @@ pub async fn upload_torrent(req: HttpRequest, payload: Multipart, app_data: WebA
         )
         .await?;
 
+    // update torrent tracker stats
+    let _ = app_data
+        .tracker
+        .update_torrent_tracker_stats(torrent_id, &torrent_request.torrent.info_hash())
+        .await;
+
     // whitelist info hash on tracker
     if let Err(e) = app_data
         .tracker
diff --git a/src/tracker.rs b/src/tracker.rs
@@ -179,13 +179,19 @@ impl TrackerService {
     }
 
     pub async fn update_torrents(&self) -> Result<(), ServiceError> {
-        println!("Updating torrents..");
+        println!("Updating torrents ...");
         let torrents = self.database.get_all_torrents_compact().await?;
 
         for torrent in torrents {
-            let _ = self.get_torrent_info(torrent.torrent_id, &torrent.info_hash).await;
+            let _ = self
+                .update_torrent_tracker_stats(torrent.torrent_id, &torrent.info_hash)
+                .await;
         }
 
         Ok(())
     }
+
+    pub async fn update_torrent_tracker_stats(&self, torrent_id: i64, info_hash: &str) -> Result<TorrentInfo, ServiceError> {
+        self.get_torrent_info(torrent_id, info_hash).await
+    }
 }

Original file line number	Diff line number	Diff line change
`@@ -179,13 +179,19 @@ impl TrackerService {`
`179`	`179`	`}`
`180`	`180`
`181`	`181`	`pub async fn update_torrents(&self) -> Result<(), ServiceError> {`
`182`		`- println!("Updating torrents..");`
	`182`	`+ println!("Updating torrents ...");`
`183`	`183`	`let torrents = self.database.get_all_torrents_compact().await?;`
`184`	`184`
`185`	`185`	`for torrent in torrents {`
`186`		`- let _ = self.get_torrent_info(torrent.torrent_id, &torrent.info_hash).await;`
	`186`	`+ let _ = self`
	`187`	`+ .update_torrent_tracker_stats(torrent.torrent_id, &torrent.info_hash)`
	`188`	`+ .await;`
`187`	`189`	`}`
`188`	`190`
`189`	`191`	`Ok(())`
`190`	`192`	`}`
	`193`	`+`
	`194`	`+ pub async fn update_torrent_tracker_stats(&self, torrent_id: i64, info_hash: &str) -> Result<TorrentInfo, ServiceError> {`
	`195`	`+ self.get_torrent_info(torrent_id, info_hash).await`
	`196`	`+ }`
`191`	`197`	`}`