leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Buffered read/writes in dht persistence

Browse source
This commit is contained in:
Igor Katson 2023-07-14 00:15:38 +01:00
parent 3428de8f3d
commit 2a2a4da196
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
crates/dht/src/persistence.rs
View file

@ -2,6 +2,7 @@
use serde::{Deserialize, Serialize};
use std::fs::OpenOptions;
use std::io::{BufReader, BufWriter};
use std::net::SocketAddr;
use std::path::{Path, PathBuf};
use std::time::Duration;
@ -30,12 +31,13 @@ pub struct PersistentDht {
}
fn dump_dht(dht: &Dht, filename: &Path, tempfile_name: &Path) -> anyhow::Result<()> {
let mut file = OpenOptions::new()
let file = OpenOptions::new()
.truncate(true)
.create(true)
.write(true)
.open(tempfile_name)
.with_context(|| format!("error opening {tempfile_name:?}"))?;
let mut file = BufWriter::new(file);
let addr = dht.listen_addr();
match dht
@ -76,7 +78,8 @@ impl PersistentDht {
let de = match OpenOptions::new().read(true).open(&config_filename) {
Ok(dht_json) => {
match serde_json::from_reader::<_, DhtSerialize<RoutingTable>>(&dht_json) {
let reader = BufReader::new(dht_json);
match serde_json::from_reader::<_, DhtSerialize<RoutingTable>>(reader) {
Ok(r) => {
info!("loaded DHT routing table from {:?}", &config_filename);
Some(r)