Skip to content

Commit 0508a6a

Browse files
committed
refactor: [#1541] rename methods
1 parent 7e27d31 commit 0508a6a

File tree

9 files changed

+68
-68
lines changed

9 files changed

+68
-68
lines changed

packages/tracker-core/src/announce_handler.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ impl AnnounceHandler {
169169
// downloads across all torrents. The in-memory metric will count only
170170
// the number of downloads during the current tracker uptime.
171171
let opt_persistent_torrent = if self.config.tracker_policy.persistent_torrent_completed_stat {
172-
self.db_downloads_metric_repository.load(info_hash)?
172+
self.db_downloads_metric_repository.load_torrent_downloads(info_hash)?
173173
} else {
174174
None
175175
};

packages/tracker-core/src/databases/driver/mod.rs

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -169,9 +169,9 @@ pub(crate) mod tests {
169169

170170
let number_of_downloads = 1;
171171

172-
driver.save_persistent_torrent(&infohash, number_of_downloads).unwrap();
172+
driver.save_torrent_downloads(&infohash, number_of_downloads).unwrap();
173173

174-
let number_of_downloads = driver.load_persistent_torrent(&infohash).unwrap().unwrap();
174+
let number_of_downloads = driver.load_torrent_downloads(&infohash).unwrap().unwrap();
175175

176176
assert_eq!(number_of_downloads, 1);
177177
}
@@ -181,9 +181,9 @@ pub(crate) mod tests {
181181

182182
let number_of_downloads = 1;
183183

184-
driver.save_persistent_torrent(&infohash, number_of_downloads).unwrap();
184+
driver.save_torrent_downloads(&infohash, number_of_downloads).unwrap();
185185

186-
let torrents = driver.load_persistent_torrents().unwrap();
186+
let torrents = driver.load_all_torrents_downloads().unwrap();
187187

188188
assert_eq!(torrents.len(), 1);
189189
assert_eq!(torrents.get(&infohash), Some(number_of_downloads).as_ref());
@@ -194,11 +194,11 @@ pub(crate) mod tests {
194194

195195
let number_of_downloads = 1;
196196

197-
driver.save_persistent_torrent(&infohash, number_of_downloads).unwrap();
197+
driver.save_torrent_downloads(&infohash, number_of_downloads).unwrap();
198198

199-
driver.increase_number_of_downloads(&infohash).unwrap();
199+
driver.increase_downloads_for_torrent(&infohash).unwrap();
200200

201-
let number_of_downloads = driver.load_persistent_torrent(&infohash).unwrap().unwrap();
201+
let number_of_downloads = driver.load_torrent_downloads(&infohash).unwrap().unwrap();
202202

203203
assert_eq!(number_of_downloads, 2);
204204
}
@@ -208,31 +208,31 @@ pub(crate) mod tests {
208208
pub fn it_should_save_and_load_the_global_number_of_downloads(driver: &Arc<Box<dyn Database>>) {
209209
let number_of_downloads = 1;
210210

211-
driver.save_global_number_of_downloads(number_of_downloads).unwrap();
211+
driver.save_global_downloads(number_of_downloads).unwrap();
212212

213-
let number_of_downloads = driver.load_global_number_of_downloads().unwrap().unwrap();
213+
let number_of_downloads = driver.load_global_downloads().unwrap().unwrap();
214214

215215
assert_eq!(number_of_downloads, 1);
216216
}
217217

218218
pub fn it_should_load_the_global_number_of_downloads(driver: &Arc<Box<dyn Database>>) {
219219
let number_of_downloads = 1;
220220

221-
driver.save_global_number_of_downloads(number_of_downloads).unwrap();
221+
driver.save_global_downloads(number_of_downloads).unwrap();
222222

223-
let number_of_downloads = driver.load_global_number_of_downloads().unwrap().unwrap();
223+
let number_of_downloads = driver.load_global_downloads().unwrap().unwrap();
224224

225225
assert_eq!(number_of_downloads, 1);
226226
}
227227

228228
pub fn it_should_increase_the_global_number_of_downloads(driver: &Arc<Box<dyn Database>>) {
229229
let number_of_downloads = 1;
230230

231-
driver.save_global_number_of_downloads(number_of_downloads).unwrap();
231+
driver.save_global_downloads(number_of_downloads).unwrap();
232232

233-
driver.increase_global_number_of_downloads().unwrap();
233+
driver.increase_global_downloads().unwrap();
234234

235-
let number_of_downloads = driver.load_global_number_of_downloads().unwrap().unwrap();
235+
let number_of_downloads = driver.load_global_downloads().unwrap().unwrap();
236236

237237
assert_eq!(number_of_downloads, 2);
238238
}

packages/tracker-core/src/databases/driver/mysql.rs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ impl Database for Mysql {
146146
}
147147

148148
/// Refer to [`databases::Database::load_persistent_torrents`](crate::core::databases::Database::load_persistent_torrents).
149-
fn load_persistent_torrents(&self) -> Result<PersistentTorrents, Error> {
149+
fn load_all_torrents_downloads(&self) -> Result<PersistentTorrents, Error> {
150150
let mut conn = self.pool.get().map_err(|e| (e, DRIVER))?;
151151

152152
let torrents = conn.query_map(
@@ -161,7 +161,7 @@ impl Database for Mysql {
161161
}
162162

163163
/// Refer to [`databases::Database::load_persistent_torrent`](crate::core::databases::Database::load_persistent_torrent).
164-
fn load_persistent_torrent(&self, info_hash: &InfoHash) -> Result<Option<PersistentTorrent>, Error> {
164+
fn load_torrent_downloads(&self, info_hash: &InfoHash) -> Result<Option<PersistentTorrent>, Error> {
165165
let mut conn = self.pool.get().map_err(|e| (e, DRIVER))?;
166166

167167
let query = conn.exec_first::<u32, _, _>(
@@ -175,7 +175,7 @@ impl Database for Mysql {
175175
}
176176

177177
/// Refer to [`databases::Database::save_persistent_torrent`](crate::core::databases::Database::save_persistent_torrent).
178-
fn save_persistent_torrent(&self, info_hash: &InfoHash, completed: u32) -> Result<(), Error> {
178+
fn save_torrent_downloads(&self, info_hash: &InfoHash, completed: u32) -> Result<(), Error> {
179179
const COMMAND : &str = "INSERT INTO torrents (info_hash, completed) VALUES (:info_hash_str, :completed) ON DUPLICATE KEY UPDATE completed = VALUES(completed)";
180180

181181
let mut conn = self.pool.get().map_err(|e| (e, DRIVER))?;
@@ -186,7 +186,7 @@ impl Database for Mysql {
186186
}
187187

188188
/// Refer to [`databases::Database::increase_number_of_downloads`](crate::core::databases::Database::increase_number_of_downloads).
189-
fn increase_number_of_downloads(&self, info_hash: &InfoHash) -> Result<(), Error> {
189+
fn increase_downloads_for_torrent(&self, info_hash: &InfoHash) -> Result<(), Error> {
190190
let mut conn = self.pool.get().map_err(|e| (e, DRIVER))?;
191191

192192
let info_hash_str = info_hash.to_string();
@@ -200,17 +200,17 @@ impl Database for Mysql {
200200
}
201201

202202
/// Refer to [`databases::Database::load_global_number_of_downloads`](crate::core::databases::Database::load_global_number_of_downloads).
203-
fn load_global_number_of_downloads(&self) -> Result<Option<PersistentTorrent>, Error> {
203+
fn load_global_downloads(&self) -> Result<Option<PersistentTorrent>, Error> {
204204
self.load_torrent_aggregate_metric(TORRENTS_DOWNLOADS_TOTAL)
205205
}
206206

207207
/// Refer to [`databases::Database::save_global_number_of_downloads`](crate::core::databases::Database::save_global_number_of_downloads).
208-
fn save_global_number_of_downloads(&self, downloaded: PersistentTorrent) -> Result<(), Error> {
208+
fn save_global_downloads(&self, downloaded: PersistentTorrent) -> Result<(), Error> {
209209
self.save_torrent_aggregate_metric(TORRENTS_DOWNLOADS_TOTAL, downloaded)
210210
}
211211

212212
/// Refer to [`databases::Database::increase_global_number_of_downloads`](crate::core::databases::Database::increase_global_number_of_downloads).
213-
fn increase_global_number_of_downloads(&self) -> Result<(), Error> {
213+
fn increase_global_downloads(&self) -> Result<(), Error> {
214214
let mut conn = self.pool.get().map_err(|e| (e, DRIVER))?;
215215

216216
let metric_name = TORRENTS_DOWNLOADS_TOTAL;

packages/tracker-core/src/databases/driver/sqlite.rs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ impl Database for Sqlite {
152152
}
153153

154154
/// Refer to [`databases::Database::load_persistent_torrents`](crate::core::databases::Database::load_persistent_torrents).
155-
fn load_persistent_torrents(&self) -> Result<PersistentTorrents, Error> {
155+
fn load_all_torrents_downloads(&self) -> Result<PersistentTorrents, Error> {
156156
let conn = self.pool.get().map_err(|e| (e, DRIVER))?;
157157

158158
let mut stmt = conn.prepare("SELECT info_hash, completed FROM torrents")?;
@@ -168,7 +168,7 @@ impl Database for Sqlite {
168168
}
169169

170170
/// Refer to [`databases::Database::load_persistent_torrent`](crate::core::databases::Database::load_persistent_torrent).
171-
fn load_persistent_torrent(&self, info_hash: &InfoHash) -> Result<Option<PersistentTorrent>, Error> {
171+
fn load_torrent_downloads(&self, info_hash: &InfoHash) -> Result<Option<PersistentTorrent>, Error> {
172172
let conn = self.pool.get().map_err(|e| (e, DRIVER))?;
173173

174174
let mut stmt = conn.prepare("SELECT completed FROM torrents WHERE info_hash = ?")?;
@@ -184,7 +184,7 @@ impl Database for Sqlite {
184184
}
185185

186186
/// Refer to [`databases::Database::save_persistent_torrent`](crate::core::databases::Database::save_persistent_torrent).
187-
fn save_persistent_torrent(&self, info_hash: &InfoHash, completed: u32) -> Result<(), Error> {
187+
fn save_torrent_downloads(&self, info_hash: &InfoHash, completed: u32) -> Result<(), Error> {
188188
let conn = self.pool.get().map_err(|e| (e, DRIVER))?;
189189

190190
let insert = conn.execute(
@@ -203,7 +203,7 @@ impl Database for Sqlite {
203203
}
204204

205205
/// Refer to [`databases::Database::increase_number_of_downloads`](crate::core::databases::Database::increase_number_of_downloads).
206-
fn increase_number_of_downloads(&self, info_hash: &InfoHash) -> Result<(), Error> {
206+
fn increase_downloads_for_torrent(&self, info_hash: &InfoHash) -> Result<(), Error> {
207207
let conn = self.pool.get().map_err(|e| (e, DRIVER))?;
208208

209209
let _ = conn.execute(
@@ -215,17 +215,17 @@ impl Database for Sqlite {
215215
}
216216

217217
/// Refer to [`databases::Database::load_global_number_of_downloads`](crate::core::databases::Database::load_global_number_of_downloads).
218-
fn load_global_number_of_downloads(&self) -> Result<Option<PersistentTorrent>, Error> {
218+
fn load_global_downloads(&self) -> Result<Option<PersistentTorrent>, Error> {
219219
self.load_torrent_aggregate_metric(TORRENTS_DOWNLOADS_TOTAL)
220220
}
221221

222222
/// Refer to [`databases::Database::save_global_number_of_downloads`](crate::core::databases::Database::save_global_number_of_downloads).
223-
fn save_global_number_of_downloads(&self, downloaded: PersistentTorrent) -> Result<(), Error> {
223+
fn save_global_downloads(&self, downloaded: PersistentTorrent) -> Result<(), Error> {
224224
self.save_torrent_aggregate_metric(TORRENTS_DOWNLOADS_TOTAL, downloaded)
225225
}
226226

227227
/// Refer to [`databases::Database::increase_global_number_of_downloads`](crate::core::databases::Database::increase_global_number_of_downloads).
228-
fn increase_global_number_of_downloads(&self) -> Result<(), Error> {
228+
fn increase_global_downloads(&self) -> Result<(), Error> {
229229
let conn = self.pool.get().map_err(|e| (e, DRIVER))?;
230230

231231
let metric_name = TORRENTS_DOWNLOADS_TOTAL;

packages/tracker-core/src/databases/mod.rs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ pub trait Database: Sync + Send {
101101
/// # Errors
102102
///
103103
/// Returns an [`Error`] if the metrics cannot be loaded.
104-
fn load_persistent_torrents(&self) -> Result<PersistentTorrents, Error>;
104+
fn load_all_torrents_downloads(&self) -> Result<PersistentTorrents, Error>;
105105

106106
/// Loads torrent metrics data from the database for one torrent.
107107
///
@@ -110,7 +110,7 @@ pub trait Database: Sync + Send {
110110
/// # Errors
111111
///
112112
/// Returns an [`Error`] if the metrics cannot be loaded.
113-
fn load_persistent_torrent(&self, info_hash: &InfoHash) -> Result<Option<PersistentTorrent>, Error>;
113+
fn load_torrent_downloads(&self, info_hash: &InfoHash) -> Result<Option<PersistentTorrent>, Error>;
114114

115115
/// Saves torrent metrics data into the database.
116116
///
@@ -124,7 +124,7 @@ pub trait Database: Sync + Send {
124124
/// # Errors
125125
///
126126
/// Returns an [`Error`] if the metrics cannot be saved.
127-
fn save_persistent_torrent(&self, info_hash: &InfoHash, downloaded: u32) -> Result<(), Error>;
127+
fn save_torrent_downloads(&self, info_hash: &InfoHash, downloaded: u32) -> Result<(), Error>;
128128

129129
/// Increases the number of downloads for a given torrent.
130130
///
@@ -140,7 +140,7 @@ pub trait Database: Sync + Send {
140140
/// # Errors
141141
///
142142
/// Returns an [`Error`] if the query failed.
143-
fn increase_number_of_downloads(&self, info_hash: &InfoHash) -> Result<(), Error>;
143+
fn increase_downloads_for_torrent(&self, info_hash: &InfoHash) -> Result<(), Error>;
144144

145145
/// Loads the total number of downloads for all torrents from the database.
146146
///
@@ -149,7 +149,7 @@ pub trait Database: Sync + Send {
149149
/// # Errors
150150
///
151151
/// Returns an [`Error`] if the total downloads cannot be loaded.
152-
fn load_global_number_of_downloads(&self) -> Result<Option<PersistentTorrent>, Error>;
152+
fn load_global_downloads(&self) -> Result<Option<PersistentTorrent>, Error>;
153153

154154
/// Saves the total number of downloads for all torrents into the database.
155155
///
@@ -163,7 +163,7 @@ pub trait Database: Sync + Send {
163163
/// # Errors
164164
///
165165
/// Returns an [`Error`] if the total downloads cannot be saved.
166-
fn save_global_number_of_downloads(&self, downloaded: PersistentTorrent) -> Result<(), Error>;
166+
fn save_global_downloads(&self, downloaded: PersistentTorrent) -> Result<(), Error>;
167167

168168
/// Increases the total number of downloads for all torrents.
169169
///
@@ -172,7 +172,7 @@ pub trait Database: Sync + Send {
172172
/// # Errors
173173
///
174174
/// Returns an [`Error`] if the query failed.
175-
fn increase_global_number_of_downloads(&self) -> Result<(), Error>;
175+
fn increase_global_downloads(&self) -> Result<(), Error>;
176176

177177
// Whitelist
178178

packages/tracker-core/src/statistics/event/handler.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ pub async fn handle_event(
5353

5454
if persistent_torrent_completed_stat {
5555
// Increment the number of downloads for the torrent in the database
56-
match db_downloads_metric_repository.increase_number_of_downloads(&info_hash) {
56+
match db_downloads_metric_repository.increase_downloads_for_torrent(&info_hash) {
5757
Ok(()) => {
5858
tracing::debug!(info_hash = ?info_hash, "Number of torrent downloads increased");
5959
}
@@ -63,7 +63,7 @@ pub async fn handle_event(
6363
}
6464

6565
// Increment the global number of downloads (for all torrents) in the database
66-
match db_downloads_metric_repository.increase_global_number_of_downloads() {
66+
match db_downloads_metric_repository.increase_global_downloads() {
6767
Ok(()) => {
6868
tracing::debug!("Global number of downloads increased");
6969
}

packages/tracker-core/src/statistics/persisted/downloads.rs

Lines changed: 23 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -60,12 +60,12 @@ impl DatabaseDownloadsMetricRepository {
6060
/// # Errors
6161
///
6262
/// Returns an [`Error`] if the database operation fails.
63-
pub(crate) fn increase_number_of_downloads(&self, info_hash: &InfoHash) -> Result<(), Error> {
64-
let torrent = self.load(info_hash)?;
63+
pub(crate) fn increase_downloads_for_torrent(&self, info_hash: &InfoHash) -> Result<(), Error> {
64+
let torrent = self.load_torrent_downloads(info_hash)?;
6565

6666
match torrent {
67-
Some(_number_of_downloads) => self.database.increase_number_of_downloads(info_hash),
68-
None => self.save(info_hash, 1),
67+
Some(_number_of_downloads) => self.database.increase_downloads_for_torrent(info_hash),
68+
None => self.save_torrent_downloads(info_hash, 1),
6969
}
7070
}
7171

@@ -77,8 +77,8 @@ impl DatabaseDownloadsMetricRepository {
7777
/// # Errors
7878
///
7979
/// Returns an [`Error`] if the underlying database query fails.
80-
pub(crate) fn load_all(&self) -> Result<PersistentTorrents, Error> {
81-
self.database.load_persistent_torrents()
80+
pub(crate) fn load_all_torrents_downloads(&self) -> Result<PersistentTorrents, Error> {
81+
self.database.load_all_torrents_downloads()
8282
}
8383

8484
/// Loads one persistent torrent metrics from the database.
@@ -89,8 +89,8 @@ impl DatabaseDownloadsMetricRepository {
8989
/// # Errors
9090
///
9191
/// Returns an [`Error`] if the underlying database query fails.
92-
pub(crate) fn load(&self, info_hash: &InfoHash) -> Result<Option<PersistentTorrent>, Error> {
93-
self.database.load_persistent_torrent(info_hash)
92+
pub(crate) fn load_torrent_downloads(&self, info_hash: &InfoHash) -> Result<Option<PersistentTorrent>, Error> {
93+
self.database.load_torrent_downloads(info_hash)
9494
}
9595

9696
/// Saves the persistent torrent metric into the database.
@@ -106,8 +106,8 @@ impl DatabaseDownloadsMetricRepository {
106106
/// # Errors
107107
///
108108
/// Returns an [`Error`] if the database operation fails.
109-
pub(crate) fn save(&self, info_hash: &InfoHash, downloaded: u32) -> Result<(), Error> {
110-
self.database.save_persistent_torrent(info_hash, downloaded)
109+
pub(crate) fn save_torrent_downloads(&self, info_hash: &InfoHash, downloaded: u32) -> Result<(), Error> {
110+
self.database.save_torrent_downloads(info_hash, downloaded)
111111
}
112112

113113
// Aggregate Metrics
@@ -119,12 +119,12 @@ impl DatabaseDownloadsMetricRepository {
119119
/// # Errors
120120
///
121121
/// Returns an [`Error`] if the database operation fails.
122-
pub(crate) fn increase_global_number_of_downloads(&self) -> Result<(), Error> {
123-
let torrent = self.database.load_global_number_of_downloads()?;
122+
pub(crate) fn increase_global_downloads(&self) -> Result<(), Error> {
123+
let torrent = self.database.load_global_downloads()?;
124124

125125
match torrent {
126-
Some(_number_of_downloads) => self.database.increase_global_number_of_downloads(),
127-
None => self.database.save_global_number_of_downloads(1),
126+
Some(_number_of_downloads) => self.database.increase_global_downloads(),
127+
None => self.database.save_global_downloads(1),
128128
}
129129
}
130130

@@ -133,8 +133,8 @@ impl DatabaseDownloadsMetricRepository {
133133
/// # Errors
134134
///
135135
/// Returns an [`Error`] if the underlying database query fails.
136-
pub(crate) fn load_global_number_of_downloads(&self) -> Result<Option<PersistentTorrent>, Error> {
137-
self.database.load_global_number_of_downloads()
136+
pub(crate) fn load_global_downloads(&self) -> Result<Option<PersistentTorrent>, Error> {
137+
self.database.load_global_downloads()
138138
}
139139
}
140140

@@ -159,9 +159,9 @@ mod tests {
159159

160160
let infohash = sample_info_hash();
161161

162-
repository.save(&infohash, 1).unwrap();
162+
repository.save_torrent_downloads(&infohash, 1).unwrap();
163163

164-
let torrents = repository.load_all().unwrap();
164+
let torrents = repository.load_all_torrents_downloads().unwrap();
165165

166166
assert_eq!(torrents.get(&infohash), Some(1).as_ref());
167167
}
@@ -172,9 +172,9 @@ mod tests {
172172

173173
let infohash = sample_info_hash();
174174

175-
repository.increase_number_of_downloads(&infohash).unwrap();
175+
repository.increase_downloads_for_torrent(&infohash).unwrap();
176176

177-
let torrents = repository.load_all().unwrap();
177+
let torrents = repository.load_all_torrents_downloads().unwrap();
178178

179179
assert_eq!(torrents.get(&infohash), Some(1).as_ref());
180180
}
@@ -186,10 +186,10 @@ mod tests {
186186
let infohash_one = sample_info_hash_one();
187187
let infohash_two = sample_info_hash_two();
188188

189-
repository.save(&infohash_one, 1).unwrap();
190-
repository.save(&infohash_two, 2).unwrap();
189+
repository.save_torrent_downloads(&infohash_one, 1).unwrap();
190+
repository.save_torrent_downloads(&infohash_two, 2).unwrap();
191191

192-
let torrents = repository.load_all().unwrap();
192+
let torrents = repository.load_all_torrents_downloads().unwrap();
193193

194194
let mut expected_torrents = PersistentTorrents::new();
195195
expected_torrents.insert(infohash_one, 1);

0 commit comments

Comments
 (0)