63 lines
1.2 KiB
Rust
63 lines
1.2 KiB
Rust
|
use std::io::{Read, Write};
|
||
|
|
use std::path::PathBuf;
|
||
|
|
|
||
|
|
use tokio::io::AsyncWriteExt;
|
||
|
|
|
||
|
|
use serde::{Deserialize, Serialize};
|
||
|
|
|
||
|
|
use crate::data::*;
|
||
|
|
use crate::error::Error;
|
||
|
|
|
||
|
|
pub struct Persister<T: Serialize + for<'de> Deserialize<'de>> {
|
||
|
|
path: PathBuf,
|
||
|
|
|
||
|
|
_marker: std::marker::PhantomData<T>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<T> Persister<T>
|
||
|
|
where
|
||
|
|
T: Serialize + for<'de> Deserialize<'de>,
|
||
|
|
{
|
||
|
|
pub fn new(base_dir: &PathBuf, file_name: &str) -> Self {
|
||
|
|
let mut path = base_dir.clone();
|
||
|
|
path.push(file_name);
|
||
|
|
Self {
|
||
|
|
path,
|
||
|
|
_marker: Default::default(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn load(&self) -> Result<T, Error> {
|
||
|
|
let mut file = std::fs::OpenOptions::new().read(true).open(&self.path)?;
|
||
|
|
|
||
|
|
let mut bytes = vec![];
|
||
|
|
file.read_to_end(&mut bytes)?;
|
||
|
|
|
||
|
|
let value = rmp_serde::decode::from_read_ref(&bytes[..])?;
|
||
|
|
Ok(value)
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn save(&self, t: &T) -> Result<(), Error> {
|
||
|
|
let bytes = rmp_to_vec_all_named(t)?;
|
||
|
|
|
||
|
|
let mut file = std::fs::OpenOptions::new()
|
||
|
|
.write(true)
|
||
|
|
.create(true)
|
||
|
|
.truncate(true)
|
||
|
|
.open(&self.path)?;
|
||
|
|
|
||
|
|
file.write_all(&bytes[..])?;
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn save_async(&self, t: &T) -> Result<(), Error> {
|
||
|
|
let bytes = rmp_to_vec_all_named(t)?;
|
||
|
|
|
||
|
|
let mut file = tokio::fs::File::create(&self.path).await?;
|
||
|
|
file.write_all(&bytes[..]).await?;
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
}
|