add virtual column registration

filimonov · filimonov · commit 05d44f4955d9 · 2025-10-01T23:05:51.000+02:00
diff --git a/src/Storages/StorageDistributed.cpp b/src/Storages/StorageDistributed.cpp
@@ -2202,6 +2202,11 @@ void StorageDistributed::setAdditionalTableFunctions(std::vector<TableFunctionEn
 {
     additional_table_functions = std::move(additional_table_functions_);
     log = getLogger("TieredDistributedMerge (" + getStorageID().table_name + ")");
+
+    auto virtuals = createVirtuals();
+    // or _layer_index?
+    virtuals.addEphemeral("_table_index", std::make_shared<DataTypeUInt32>(), "Index of the table function in TieredDistributedMerge (0 for main table, 1+ for additional table functions)");
+    setVirtuals(virtuals);
 }
 
 void registerStorageDistributed(StorageFactory & factory)

Original file line number	Diff line number	Diff line change
`@@ -2202,6 +2202,11 @@ void StorageDistributed::setAdditionalTableFunctions(std::vector<TableFunctionEn`
`2202`	`2202`	`{`
`2203`	`2203`	`additional_table_functions = std::move(additional_table_functions_);`
`2204`	`2204`	`log = getLogger("TieredDistributedMerge (" + getStorageID().table_name + ")");`
	`2205`	`+`
	`2206`	`+ auto virtuals = createVirtuals();`
	`2207`	`+ // or _layer_index?`
	`2208`	`+ virtuals.addEphemeral("_table_index", std::make_shared<DataTypeUInt32>(), "Index of the table function in TieredDistributedMerge (0 for main table, 1+ for additional table functions)");`
	`2209`	`+ setVirtuals(virtuals);`
`2205`	`2210`	`}`
`2206`	`2211`
`2207`	`2212`	`void registerStorageDistributed(StorageFactory & factory)`