Skip to content

Commit db049c4

Browse files
committed
fix(dawgrun): bind materialized parameters to the output query
1 parent 7708d57 commit db049c4

1 file changed

Lines changed: 16 additions & 2 deletions

File tree

tools/dawgrun/pkg/commands/cypher.go

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,14 @@ func translateToPsqlCmd() CommandDesc {
7777
fmt.Fprintf(ctx.output, "\n")
7878
}
7979

80-
sqlQuery, err := format.SyntaxNode(result.Statement)
80+
// Certain queries will materialize parameters into the output when translated, so we need to build
81+
// an OutputBuilder so we can carry forward those params.
82+
queryBuilder := format.NewOutputBuilder()
83+
if result.Parameters != nil {
84+
queryBuilder.WithMaterializedParameters(result.Parameters)
85+
}
86+
87+
sqlQuery, err := format.Statement(result.Statement, queryBuilder)
8188
if err != nil {
8289
return fmt.Errorf("could not format translated statement into a string query: %w", err)
8390
}
@@ -131,7 +138,14 @@ func explainAsPsqlCmd() CommandDesc {
131138
return fmt.Errorf("could not translate cypher query to pgsql: %w", err)
132139
}
133140

134-
sqlQuery, err := format.SyntaxNode(result.Statement)
141+
// Certain queries will materialize parameters into the output when translated, so we need to build
142+
// an OutputBuilder so we can carry forward those params.
143+
queryBuilder := format.NewOutputBuilder()
144+
if result.Parameters != nil {
145+
queryBuilder.WithMaterializedParameters(result.Parameters)
146+
}
147+
148+
sqlQuery, err := format.Statement(result.Statement, queryBuilder)
135149
if err != nil {
136150
return fmt.Errorf("could not format translated statement into a string query: %w", err)
137151
}

0 commit comments

Comments
 (0)