rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-01-14 17:23:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: wrap tql cte in a subquery alias (#6910)

Browse Source 
Signed-off-by: Ruihang Xia <waynestxia@gmail.com>
This commit is contained in:
Ruihang Xia
2025-09-04 23:07:20 -07:00
committed by Weny Xu
parent f1b7581dc3
commit aa7e7942f8
3 changed files with 32 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/query/src/planner.rs
View File

@@ -282,6 +282,16 @@ impl DfLogicalPlanner {
.build()
.context(PlanSqlSnafu)?;
}
// Wrap in SubqueryAlias to ensure proper table qualification for CTE
logical_plan = LogicalPlan::SubqueryAlias(
datafusion_expr::SubqueryAlias::try_new(
Arc::new(logical_plan),
cte.name.value.clone(),
)
.context(PlanSqlSnafu)?,
);
planner_context.insert_cte(&cte.name.value, logical_plan);
}
CteContent::Sql(_) => {

16
tests/cases/standalone/common/tql/tql-cte.result
View File

@@ -109,6 +109,22 @@ SELECT count(*) as host1_points FROM host_metrics;
| 5 |
+--------------+
-- TQL CTE with column reference
WITH host_metrics AS (
TQL EVAL (0, 40, '10s') labels{host="host1"}
)
SELECT host_metrics.ts, host_metrics.host FROM host_metrics;
+---------------------+-------+
| ts | host |
+---------------------+-------+
| 1970-01-01T00:00:00 | host1 |
| 1970-01-01T00:00:10 | host1 |
| 1970-01-01T00:00:20 | host1 |
| 1970-01-01T00:00:30 | host1 |
| 1970-01-01T00:00:40 | host1 |
+---------------------+-------+
-- Multiple TQL CTEs referencing different tables
WITH
metric_data(ts, val) AS (TQL EVAL (0, 40, '10s') metric),

6
tests/cases/standalone/common/tql/tql-cte.sql
View File

@@ -56,6 +56,12 @@ WITH host_metrics AS (
)
SELECT count(*) as host1_points FROM host_metrics;
-- TQL CTE with column reference
WITH host_metrics AS (
TQL EVAL (0, 40, '10s') labels{host="host1"}
)
SELECT host_metrics.ts, host_metrics.host FROM host_metrics;
-- Multiple TQL CTEs referencing different tables
WITH
metric_data(ts, val) AS (TQL EVAL (0, 40, '10s') metric),