Skip to content

Commit 32052e4

Browse files
committed
Fix CTE registration: wrap cte.Select in Subquery struct
1 parent 9c3c79c commit 32052e4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

internal/stackql/astanalysis/earlyanalysis/ast_expand.go

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -257,8 +257,10 @@ func (v *indirectExpandAstVisitor) Visit(node sqlparser.SQLNode) error {
257257
logging.GetLogger().Infof("Registering %d CTEs from WITH clause", len(node.With.CTEs))
258258
for _, cte := range node.With.CTEs {
259259
cteName := cte.Name.GetRawVal()
260-
v.cteRegistry[cteName] = cte.Subquery
261-
logging.GetLogger().Debugf("Registered CTE '%s' with subquery: %s", cteName, sqlparser.String(cte.Subquery))
260+
// Wrap the CTE's SELECT statement in a Subquery struct
261+
cteSubquery := &sqlparser.Subquery{Select: cte.Select}
262+
v.cteRegistry[cteName] = cteSubquery
263+
logging.GetLogger().Debugf("Registered CTE '%s' with subquery: %s", cteName, sqlparser.String(cteSubquery))
262264
}
263265
}
264266

0 commit comments

Comments
 (0)