better sqlite store drop

cmu-db · Apr 28, 2024 · b6d96d9 · b6d96d9
1 parent 9c6c861
commit b6d96d9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/storage-node/src/cache/data_store_cache/sqlite/mod.rs b/storage-node/src/cache/data_store_cache/sqlite/mod.rs
@@ -5,6 +5,7 @@ use std::fs;
 use async_trait::async_trait;
 use bytes::Bytes;
 use futures::StreamExt;
+use log::warn;
 use rusqlite::{Connection, DatabaseName, OpenFlags};
 use tokio::sync::{
     mpsc::{channel, Receiver},
@@ -89,7 +90,11 @@ impl<R: DataStoreReplacer<SqliteStoreReplacerKey, SqliteStoreReplacerValue>> Dro
     for SqliteStoreCache<R>
 {
     fn drop(&mut self) {
-        fs::remove_file(self.sqlite_base_path.clone()).expect("remove sqlite db files failed");
+        if fs::metadata(&self.sqlite_base_path).is_ok() {
+            fs::remove_file(self.sqlite_base_path.clone()).expect("remove sqlite db files failed");
+        } else {
+            warn!("sqlite db file not found: {}", self.sqlite_base_path);
+        }
     }
 }