fix: ensure that object cache insertions are weighted by size

apache · Aug 19, 2024 · 29812b1 · 29812b1
1 parent 263c929
commit 29812b1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/crates/iceberg/src/io/object_cache.rs b/crates/iceberg/src/io/object_cache.rs
@@ -59,7 +59,10 @@ impl ObjectCache {
             Self::with_disabled_cache(file_io)
         } else {
             Self {
-                cache: moka::future::Cache::new(cache_size_bytes),
+                cache: moka::future::Cache::builder()
+                    .weigher(|_, val: &CachedItem| size_of_val(val) as u32)
+                    .max_capacity(cache_size_bytes)
+                    .build(),
                 file_io,
                 cache_disabled: false,
             }