feat: add log support (#19)

Signed-off-by: MrCroxx <[email protected]>
foyer-rs · Jun 1, 2023 · 79610aa · 79610aa
1 parent 8e80c6c
commit 79610aa
Show file tree

Hide file tree

Showing 4 changed files with 65 additions and 14 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/foyer/Cargo.toml b/foyer/Cargo.toml
@@ -31,6 +31,7 @@ tokio = { version = "1", features = [
     "time",
     "signal",
 ] }
+tracing = "0.1"
 twox-hash = "1"
 
 [dev-dependencies]

diff --git a/foyer/src/container.rs b/foyer/src/container.rs
@@ -95,6 +95,8 @@ where
         config: Config<I, P, H, S>,
         registry: prometheus::Registry,
     ) -> Result<Self> {
+        tracing::info!("open foyer with config: \n{:#?}", config);
+
         let pool_count = 1 << config.pool_count_bits;
         let capacity = config.capacity >> config.pool_count_bits;
 

diff --git a/foyer/src/store/read_only_file_store.rs b/foyer/src/store/read_only_file_store.rs
@@ -286,13 +286,18 @@ where
                     .read(location.offset as u64, location.len as usize)
                     .await?
             } else {
-                files
-                    .frozens
-                    .get(&fid)
-                    .expect("frozen file not found")
-                    .cache_file
-                    .read(location.offset as u64, location.len as usize)
-                    .await?
+                match files.frozens.get(&fid) {
+                    Some(frozen) => {
+                        frozen
+                            .cache_file
+                            .read(location.offset as u64, location.len as usize)
+                            .await?
+                    }
+                    None => {
+                        tracing::error!("frozen file {} not found", fid);
+                        return Ok(None);
+                    }
+                }
             }
         };
 
@@ -325,13 +330,17 @@ where
                     .write(sid as u64 * Self::meta_entry_size() as u64, empty_entry)
                     .await?;
             } else {
-                files
-                    .frozens
-                    .get(&fid)
-                    .expect("frozen file not found")
-                    .meta_file
-                    .write(sid as u64 * Self::meta_entry_size() as u64, empty_entry)
-                    .await?;
+                match files.frozens.get(&fid) {
+                    Some(frozen) => {
+                        frozen
+                            .meta_file
+                            .write(sid as u64 * Self::meta_entry_size() as u64, empty_entry)
+                            .await?;
+                    }
+                    None => {
+                        tracing::error!("frozen file {} not found", fid);
+                    }
+                }
             }
         }
 
@@ -369,10 +378,14 @@ where
         // update frozen map
         files.frozens.insert(id, frozen);
 
+        tracing::info!("active file rotated: {} => {}", id, id + 1);
+
         Ok(())
     }
 
     async fn reclaim_frozen_file(&self, id: FileId) -> Result<()> {
+        tracing::info!("reclaiming frozen file {}", id);
+
         let (fid, meta_file, cache_file) = {
             let mut files = self.files.write().await;
 
@@ -404,6 +417,8 @@ where
         meta_file.reclaim().await?;
         cache_file.reclaim().await?;
 
+        tracing::info!("frozen file {} reclaimed", id);
+
         Ok(())
     }