snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-23 02:27:40 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix(auth): fix env being closed when dumping

Browse Source
This commit is contained in:
ad hoc 2022-02-09 13:55:36 +01:00
parent cca65499de
commit 23eba82038
No known key found for this signature in database
GPG Key ID: 4F00A782990CC643
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
meilisearch-auth/src/dump.rs
View File

@ -10,7 +10,10 @@ const KEYS_PATH: &str = "keys";
impl AuthController { impl AuthController {
pub fn dump(src: impl AsRef<Path>, dst: impl AsRef<Path>) -> Result<()> { pub fn dump(src: impl AsRef<Path>, dst: impl AsRef<Path>) -> Result<()> {
let store = HeedAuthStore::new(&src)?; let mut store = HeedAuthStore::new(&src)?;
// do not attempt to close the database on drop!
store.set_drop_on_close(false);
let keys_file_path = dst.as_ref().join(KEYS_PATH); let keys_file_path = dst.as_ref().join(KEYS_PATH);

8
meilisearch-auth/src/store.rs
View File

@ -28,11 +28,12 @@ pub struct HeedAuthStore {
env: Arc<Env>, env: Arc<Env>,
keys: Database<ByteSlice, SerdeJson<Key>>, keys: Database<ByteSlice, SerdeJson<Key>>,
action_keyid_index_expiration: Database<KeyIdActionCodec, SerdeJson<Option<DateTime<Utc>>>>, action_keyid_index_expiration: Database<KeyIdActionCodec, SerdeJson<Option<DateTime<Utc>>>>,
should_close_on_drop: bool,
} }
impl Drop for HeedAuthStore { impl Drop for HeedAuthStore {
fn drop(&mut self) { fn drop(&mut self) {
if Arc::strong_count(&self.env) == 1 { if self.should_close_on_drop && Arc::strong_count(&self.env) == 1 {
self.env.as_ref().clone().prepare_for_closing(); self.env.as_ref().clone().prepare_for_closing();
} }
} }
@ -53,9 +54,14 @@ impl HeedAuthStore {
env, env,
keys, keys,
action_keyid_index_expiration, action_keyid_index_expiration,
should_close_on_drop: true,
}) })
} }
pub fn set_drop_on_close(&mut self, v: bool) {
self.should_close_on_drop = v;
}
pub fn is_empty(&self) -> Result<bool> { pub fn is_empty(&self) -> Result<bool> {
let rtxn = self.env.read_txn()?; let rtxn = self.env.read_txn()?;