Skip to content

Commit d2bc422

Browse files
committed
Fix cache key
Signed-off-by: Adam Gutglick <adam@spiraldb.com>
1 parent 272f664 commit d2bc422

File tree

1 file changed

+13
-15
lines changed

1 file changed

+13
-15
lines changed

vortex-datafusion/src/persistent/format.rs

Lines changed: 13 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,7 @@ impl FileFormat for VortexFormat {
246246

247247
SpawnedTask::spawn(async move {
248248
// Check if we have cached metadata for this file
249-
if let Some(cached) = cache.get_with_extra(&object, &object)
249+
if let Some(cached) = cache.get(&object)
250250
&& let Some(cached_vortex) =
251251
cached.as_any().downcast_ref::<CachedVortexMetadata>()
252252
{
@@ -305,20 +305,18 @@ impl FileFormat for VortexFormat {
305305

306306
SpawnedTask::spawn(async move {
307307
// Try to get cached metadata first
308-
let cached_metadata = file_metadata_cache
309-
.get_with_extra(&object, &object)
310-
.and_then(|cached| {
311-
cached
312-
.as_any()
313-
.downcast_ref::<CachedVortexMetadata>()
314-
.map(|m| {
315-
(
316-
m.footer().dtype().clone(),
317-
m.footer().statistics().cloned(),
318-
m.footer().row_count(),
319-
)
320-
})
321-
});
308+
let cached_metadata = file_metadata_cache.get(&object).and_then(|cached| {
309+
cached
310+
.as_any()
311+
.downcast_ref::<CachedVortexMetadata>()
312+
.map(|m| {
313+
(
314+
m.footer().dtype().clone(),
315+
m.footer().statistics().cloned(),
316+
m.footer().row_count(),
317+
)
318+
})
319+
});
322320

323321
let (dtype, file_stats, row_count) = match cached_metadata {
324322
Some(metadata) => metadata,

0 commit comments

Comments
 (0)