Start9/patch-db
2
0
Fork 0
mirror of https://github.com/Start9Labs/patch-db.git synced 2026-03-26 02:11:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

prefer compression as a failsafe (#48)

Browse Source
This commit is contained in:
Aiden McClelland
2022-09-22 08:40:06 -06:00
committed by GitHub
parent 4d987b1921
commit e74f36f073
1 changed files with 30 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
patch-db/src/store.rs
View File

@@ -1,4 +1,4 @@
use std::collections::{HashMap, VecDeque};
use std::collections::{BTreeSet, HashMap, VecDeque};
use std::fs::OpenOptions;
use std::io::SeekFrom;
use std::path::{Path, PathBuf};
@@ -11,15 +11,14 @@ use json_ptr::{JsonPointer, SegList};
use lazy_static::lazy_static;
use serde::{Deserialize, Serialize};
use serde_json::Value;
use std::collections::BTreeSet;
use tokio::fs::File;
use tokio::io::AsyncSeekExt;
use tokio::sync::{Mutex, OwnedMutexGuard, RwLock, RwLockWriteGuard};
use crate::handle::HandleId;
use crate::locker::Locker;
use crate::patch::{diff, DiffPatch, Dump, Revision};
use crate::Error;
use crate::{locker::Locker, PatchDbHandle};
use crate::{Error, PatchDbHandle};
lazy_static! {
static ref OPEN_STORES: Mutex<HashMap<PathBuf, Arc<Mutex<()>>>> = Mutex::new(HashMap::new());
@@ -216,6 +215,9 @@ impl Store {
let patch_bin = serde_cbor::to_vec(&*patch)?;
let persistent_undo = json_patch::patch(&mut self.persistent, &*patch)?;
self.revision += 1;
if let Err(_e) = if self.revision % 4096 == 0 {
self.compress().await
} else {
async {
let file_len = self.file.metadata().await?.len();
if file_len != self.file_cursor {
@@ -229,11 +231,18 @@ impl Store {
Ok::<_, Error>(())
}
.await
.map_err(|e| {
} {
#[cfg(feature = "tracing")]
tracing::error!("Error saving patch to disk: {}, attempting to compress", _e);
if let Err(e) = self.compress().await {
#[cfg(feature = "tracing")]
tracing::error!("Compression failed: {}", e);
persistent_undo.apply(&mut self.persistent);
self.revision -= 1;
e
})?;
return Err(e);
}
};
drop(persistent_undo);
let id = self.revision;
let res = Arc::new(Revision { id, patch });