fix(core): Make TRIGGER backup work while stores are in use
`TRIGGER backup` answered `ERR internal_error` in two common cases: - a KV store was open in the pool (e.g. right after a `PUSH`): the backup opened the database a second time, which RocksDB refuses as its `LOCK` file is held. The pooled store is now backed up instead, after a flush. - the KV or FST store directory did not exist yet (nothing ingested, or nothing consolidated yet): listing it failed. There is nothing to back up in that case. Also enable the `trigger_backup` test, which covers both cases. Fixes #274
This commit is contained in:
1 parent
fdad02e264
commit
51a00dd099
3 files changed
+75
-9
No files matched your search
@@ -20,6 +20,12 @@ impl FstStorePool {
|
||||
// Create backup directory (full path)
|
||||
fs::create_dir_all(path)?;
|
||||
|
||||
// NOTE: The FST store directory gets created when the first FST is
|
||||
// consolidated, so there is nothing to back up until then.
|
||||
if !self.fst_store_config.path.exists() {
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
// Proceed dump action (backup)
|
||||
self.dump_action(
|
||||
"backup",
|
||||
|
||||
@@ -6,6 +6,7 @@
|
||||
// License: Mozilla Public License v2.0 (MPL v2.0)
|
||||
|
||||
use std::path::Path;
|
||||
use std::sync::Arc;
|
||||
use std::{fs, io};
|
||||
|
||||
use rocksdb::backup::{
|
||||
@@ -22,6 +23,12 @@ impl KvStorePool {
|
||||
// Create backup directory (full path)
|
||||
fs::create_dir_all(path)?;
|
||||
|
||||
// NOTE: The KV store directory gets created when the first KV store
|
||||
// is opened, so there is nothing to back up until then.
|
||||
if !self.kv_store_config.path.exists() {
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
// Proceed dump action (backup)
|
||||
self.dump_action(
|
||||
"backup",
|
||||
@@ -97,10 +104,6 @@ impl KvStorePool {
|
||||
// Create backup folder for collection
|
||||
fs::create_dir_all(backup_path.join(collection_hash))?;
|
||||
|
||||
let origin_kv = self
|
||||
.open(store_id, |_| {})
|
||||
.map_err(|_| io::Error::other("database open failure"))?;
|
||||
|
||||
// Initialize KV database backup engine
|
||||
let kv_backup_options = DBBackupEngineOptions::new(&kv_backup_path)
|
||||
.map_err(|_| io::Error::other("backup engine options acquire failure"))?;
|
||||
@@ -111,9 +114,23 @@ impl KvStorePool {
|
||||
.map_err(|_| io::Error::other("backup engine failure"))?;
|
||||
|
||||
// Proceed actual KV database backup
|
||||
kv_backup_engine
|
||||
.create_new_backup(&origin_kv)
|
||||
.map_err(|_| io::Error::other("database backup failure"))?;
|
||||
// NOTE: RocksDB refuses to open a database which is already open in
|
||||
// this process (its `LOCK` file is held), so back up the pooled
|
||||
// store if there is one. Flush it first, as its memtables may hold
|
||||
// writes not persisted yet (e.g. if the write-ahead log is disabled).
|
||||
let pooled_store = self.read().unwrap().get(&store_id).map(Arc::clone);
|
||||
|
||||
match pooled_store {
|
||||
Some(store) => kv_backup_engine.create_new_backup_flush(&store.database, true),
|
||||
None => {
|
||||
let origin_kv = self
|
||||
.open(store_id, |_| {})
|
||||
.map_err(|_| io::Error::other("database open failure"))?;
|
||||
|
||||
kv_backup_engine.create_new_backup(&origin_kv)
|
||||
}
|
||||
}
|
||||
.map_err(|_| io::Error::other("database backup failure"))?;
|
||||
|
||||
tracing::info!("kv store {store_id} backed up to path: {kv_backup_path:?}");
|
||||
|
||||
|
||||
Reference in new issue
Block a user