arch/udpt
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Ability to save compressed database.

Browse source
This commit is contained in:
Naim A 2018-10-28 08:41:52 +02:00
parent 1908056c8b
commit 95b1d5b4b8
No known key found for this signature in database
GPG key ID: FD7948915D9EF8B9
4 changed files with 47 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
Cargo.lock generated
View file

@ -253,6 +253,24 @@ dependencies = [
"iovec 0.1.2 (registry+https://github.com/rust-lang/crates.io-index)",
]
[[package]]
name = "bzip2"
version = "0.3.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
dependencies = [
"bzip2-sys 0.1.6 (registry+https://github.com/rust-lang/crates.io-index)",
"libc 0.2.43 (registry+https://github.com/rust-lang/crates.io-index)",
]
[[package]]
name = "bzip2-sys"
version = "0.1.6"
source = "registry+https://github.com/rust-lang/crates.io-index"
dependencies = [
"cc 1.0.25 (registry+https://github.com/rust-lang/crates.io-index)",
"libc 0.2.43 (registry+https://github.com/rust-lang/crates.io-index)",
]
[[package]]
name = "cc"
version = "1.0.25"
@ -1441,6 +1459,7 @@ dependencies = [
"actix-web 0.7.13 (registry+https://github.com/rust-lang/crates.io-index)",
"binascii 0.1.2 (registry+https://github.com/rust-lang/crates.io-index)",
"bincode 1.0.1 (registry+https://github.com/rust-lang/crates.io-index)",
"bzip2 0.3.3 (registry+https://github.com/rust-lang/crates.io-index)",
"clap 2.32.0 (registry+https://github.com/rust-lang/crates.io-index)",
"fern 0.5.6 (registry+https://github.com/rust-lang/crates.io-index)",
"log 0.4.5 (registry+https://github.com/rust-lang/crates.io-index)",
@ -1623,6 +1642,8 @@ dependencies = [
"checksum build_const 0.2.1 (registry+https://github.com/rust-lang/crates.io-index)" = "39092a32794787acd8525ee150305ff051b0aa6cc2abaf193924f5ab05425f39"
"checksum byteorder 1.2.6 (registry+https://github.com/rust-lang/crates.io-index)" = "90492c5858dd7d2e78691cfb89f90d273a2800fc11d98f60786e5d87e2f83781"
"checksum bytes 0.4.10 (registry+https://github.com/rust-lang/crates.io-index)" = "0ce55bd354b095246fc34caf4e9e242f5297a7fd938b090cadfea6eee614aa62"
"checksum bzip2 0.3.3 (registry+https://github.com/rust-lang/crates.io-index)" = "42b7c3cbf0fa9c1b82308d57191728ca0256cb821220f4e2fd410a72ade26e3b"
"checksum bzip2-sys 0.1.6 (registry+https://github.com/rust-lang/crates.io-index)" = "2c5162604199bbb17690ede847eaa6120a3f33d5ab4dcc8e7c25b16d849ae79b"
"checksum cc 1.0.25 (registry+https://github.com/rust-lang/crates.io-index)" = "f159dfd43363c4d08055a07703eb7a3406b0dac4d0584d96965a3262db3c9d16"
"checksum cfg-if 0.1.6 (registry+https://github.com/rust-lang/crates.io-index)" = "082bb9b28e00d3c9d39cc03e64ce4cea0f1bb9b3fde493f0cbc008472d22bdf4"
"checksum clap 2.32.0 (registry+https://github.com/rust-lang/crates.io-index)" = "b957d88f4b6a63b9d70d5f454ac8011819c6efa7727858f458ab71c756ce2d3e"

1
Cargo.toml
View file

@ -19,3 +19,4 @@ log = "0.4.5"
fern = "0.5.6"
num_cpus = "1.8.0"
serde_json = "1.0.32"
bzip2 = "0.3.3"

1
src/main.rs
View file

@ -11,6 +11,7 @@ extern crate toml;
extern crate fern;
extern crate num_cpus;
extern crate serde_json;
extern crate bzip2;
mod server;
mod tracker;

24
src/tracker.rs
View file

@ -1,6 +1,7 @@
use std;
use serde;
use binascii;
use serde_json;
use server::Events;
@ -298,6 +299,18 @@ impl TorrentTracker {
pub (crate) fn get_database(&self) -> std::sync::RwLockReadGuard<std::collections::BTreeMap<InfoHash, TorrentEntry>>{
self.database.torrent_peers.read().unwrap()
}
pub fn save_database<W: std::io::Write>(&self, writer: &mut W) -> serde_json::Result<()> {
use bzip2;
let compressor = bzip2::write::BzEncoder::new(writer, bzip2::Compression::Best);
let db_lock = self.database.torrent_peers.read().unwrap();
let db = &*db_lock;
serde_json::to_writer(compressor, &db)
}
}
#[cfg(test)]
@ -317,6 +330,17 @@ mod tests {
is_sync::<TorrentTracker>();
}
#[test]
fn test_save_db() {
let tracker = TorrentTracker::new(TrackerMode::DynamicMode);
tracker.add_torrent(&[0u8, 1, 2, 3, 4, 5, 6, 7, 8, 9, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0].into());
let mut out = Vec::new();
tracker.save_database(&mut out).unwrap();
assert!(out.len() > 0);
}
#[test]
fn test_infohash_de() {
use serde_json;