Merge pull request #60470 from ClickHouse/ClibMouse-bugfix/kql-distri…

…buted-tables

Merging #59674.

src/Client/HedgedConnections.cpp

@@ -177,6 +177,10 @@ void HedgedConnections::sendQuery(
     {
         Settings modified_settings = settings;
 
+        /// Queries in foreign languages are transformed to ClickHouse-SQL. Ensure the setting before sending.
+        modified_settings.dialect = Dialect::clickhouse;
+        modified_settings.dialect.changed = false;
+
         if (disable_two_level_aggregation)
         {
             /// Disable two-level aggregation due to version incompatibility.

src/Client/MultiplexedConnections.cpp

@@ -126,6 +126,10 @@ void MultiplexedConnections::sendQuery(
 
     Settings modified_settings = settings;
 
+    /// Queries in foreign languages are transformed to ClickHouse-SQL. Ensure the setting before sending.
+    modified_settings.dialect = Dialect::clickhouse;
+    modified_settings.dialect.changed = false;
+
     for (auto & replica : replica_states)
     {
         if (!replica.connection)

tests/queries/0_stateless/02985_dialects_with_distributed_tables.reference

@@ -0,0 +1,9 @@
+123
+234
+315
+123
+234
+315
+123
+234
+315

tests/queries/0_stateless/02985_dialects_with_distributed_tables.sql

@@ -0,0 +1,30 @@
+-- Tags: no-fasttest, distributed
+
+DROP TABLE IF EXISTS shared_test_table;
+DROP TABLE IF EXISTS distributed_test_table;
+
+CREATE TABLE shared_test_table (id UInt64)
+ENGINE = MergeTree
+ORDER BY (id);
+
+CREATE TABLE distributed_test_table
+ENGINE = Distributed(test_cluster_two_shard_three_replicas_localhost, currentDatabase(), shared_test_table);
+
+INSERT INTO shared_test_table VALUES (123), (651), (446), (315), (234), (764);
+
+SELECT id FROM distributed_test_table LIMIT 3;
+
+SET dialect = 'kusto';
+
+distributed_test_table | take 3;
+
+SET dialect = 'prql';
+
+from distributed_test_table
+select {id}
+take 1..3;
+
+SET dialect = 'clickhouse';
+
+DROP TABLE distributed_test_table;
+DROP TABLE shared_test_table;