fix: ensure object cache weight is that of the wrapped item, not the Arc

apache · Aug 19, 2024 · f184c97 · f184c97
1 parent b4b8c01
commit f184c97
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
@@ -60,7 +60,10 @@ impl ObjectCache {
         } else {
             Self {
                 cache: moka::future::Cache::builder()
-                    .weigher(|_, val: &CachedItem| size_of_val(val) as u32)
+                    .weigher(|_, val: &CachedItem| match val {
+                        CachedItem::ManifestList(item) => size_of_val(item.as_ref()),
+                        CachedItem::Manifest(item) => size_of_val(item.as_ref()),
+                    } as u32)
                     .max_capacity(cache_size_bytes)
                     .build(),
                 file_io,