Skip to content

Commit ef3867d

Browse files
author
James Cor
committed
properly prepend nodes to procedues within triggers
1 parent 6e5d21a commit ef3867d

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

sql/rowexec/dml_iters.go

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -200,6 +200,15 @@ func prependRowInPlanForTriggerExecution(row sql.Row) func(c transform.Context)
200200
}
201201
case *plan.ResolvedTable, *plan.IndexedTableAccess:
202202
return plan.NewPrependNode(n, row), transform.NewTree, nil
203+
case *plan.Call:
204+
newNode, same, err := transform.NodeWithCtx(n.Procedure, prependRowForTriggerExecutionSelector, prependRowInPlanForTriggerExecution(row))
205+
if err != nil {
206+
return nil, transform.SameTree, err
207+
}
208+
if same {
209+
return n, transform.SameTree, nil
210+
}
211+
return n.WithProcedure(newNode.(*plan.Procedure)), transform.NewTree, nil
203212
default:
204213
return n, transform.SameTree, nil
205214
}

0 commit comments

Comments
 (0)