force refill whole unit

Signed-off-by: MrCroxx <[email protected]>

src/storage/Cargo.toml

@@ -26,7 +26,7 @@ dyn-clone = "1.0.14"
 either = "1"
 enum-as-inner = "0.6"
 fail = "0.5"
-foyer = { git = "https://github.com/mrcroxx/foyer", rev = "f097abe" }
+foyer = { git = "https://github.com/mrcroxx/foyer", rev = "438eec8" }
 futures = { version = "0.3", default-features = false, features = ["alloc"] }
 futures-async-stream = { workspace = true }
 hex = "0.4"

src/storage/src/hummock/event_handler/refiller.rs

@@ -416,7 +416,7 @@ impl CacheRefillTask {
                         .read(&sstable_store.get_sst_data_path(object_id), range.clone())
                         .await?;
                     let mut futures = vec![];
-                    for (writer, r) in writers.into_iter().zip_eq_fast(ranges) {
+                    for (mut writer, r) in writers.into_iter().zip_eq_fast(ranges) {
                         let offset = r.start - range.start;
                         let len = r.end - r.start;
                         let bytes = data.slice(offset..offset + len);
@@ -427,6 +427,7 @@ impl CacheRefillTask {
                                 writer.weight() - writer.key().serialized_len(),
                             )?;
                             let block = Box::new(block);
+                            writer.force();
                             let res = writer.finish(block).await.map_err(HummockError::file_cache);
                             if matches!(res, Ok(true)) {
                                 GLOBAL_CACHE_REFILL_METRICS

src/storage/src/hummock/file_cache/store.rs

@@ -169,6 +169,13 @@ where
         }
     }
 
+    fn force(&mut self) {
+        match self {
+            FileCacheWriter::Foyer { writer } => writer.force(),
+            FileCacheWriter::None { writer } => writer.force(),
+        }
+    }
+
     async fn finish(self, value: Self::Value) -> FileCacheResult<bool> {
         match self {
             FileCacheWriter::Foyer { writer } => writer.finish(value).await,