fix initialization of reader

zvonand · zvonand · commit 9777b293309a · 2025-08-24T15:04:22.000+02:00
diff --git a/src/Storages/ObjectStorage/StorageObjectStorageSource.cpp b/src/Storages/ObjectStorage/StorageObjectStorageSource.cpp
@@ -238,8 +238,10 @@ void StorageObjectStorageSource::lazyInitialize()
 
 Chunk StorageObjectStorageSource::generate()
 {
+    lazyInitialize();
+
     // Virtual-only mode: no physical columns requested.
-    if (auto object_info = file_iterator->next(0);
+    if (auto object_info = reader.getObjectInfo();
         read_from_format_info.format_header.columns() == 0 &&
         object_info && object_info->metadata.has_value() &&
         object_info->metadata->attributes.contains("record_count"))
@@ -304,8 +306,6 @@ Chunk StorageObjectStorageSource::generate()
     }
 
 
-    lazyInitialize();
-
     while (true)
     {
         if (isCancelled() || !reader)

Original file line number	Diff line number	Diff line change
`@@ -238,8 +238,10 @@ void StorageObjectStorageSource::lazyInitialize()`
`238`	`238`
`239`	`239`	`Chunk StorageObjectStorageSource::generate()`
`240`	`240`	`{`
	`241`	`+ lazyInitialize();`
	`242`	`+`
`241`	`243`	`// Virtual-only mode: no physical columns requested.`
`242`		`- if (auto object_info = file_iterator->next(0);`
	`244`	`+ if (auto object_info = reader.getObjectInfo();`
`243`	`245`	`read_from_format_info.format_header.columns() == 0 &&`
`244`	`246`	`object_info && object_info->metadata.has_value() &&`
`245`	`247`	`object_info->metadata->attributes.contains("record_count"))`
`@@ -304,8 +306,6 @@ Chunk StorageObjectStorageSource::generate()`
`304`	`306`	`}`
`305`	`307`
`306`	`308`
`307`		`- lazyInitialize();`
`308`		`-`
`309`	`309`	`while (true)`
`310`	`310`	`{`
`311`	`311`	`if (isCancelled() \|\| !reader)`