Method BuildRowIter

server/node/trigger_execution.go:77–126

BuildRowIter implements the interface sql.ExecBuilderNode.

(ctx *sql.Context, b sql.NodeExecBuilder, r sql.Row)

// BuildRowIter implements the interface sql.ExecBuilderNode.
func (te *TriggerExecution) BuildRowIter(ctx *sql.Context, b sql.NodeExecBuilder, r sql.Row) (sql.RowIter, error) {
	sourceIter, err := b.Build(ctx, te.Source, r)
	if err != nil {
		return nil, err
	}
	// If there are no triggers, then we'll just return the source iter
	if len(te.Triggers) == 0 {
		return sourceIter, nil
	}
	trigFuncs := make([]framework.InterpretedFunction, len(te.Triggers))
	whens := make([]framework.InterpretedFunction, len(te.Triggers))
	for i, trig := range te.Triggers {
		trigFuncs[i], err = te.loadTriggerFunction(ctx, trig)
		if err != nil {
			return nil, err
		}
		// If we have a WHEN expression, then we need to build a "function" to execute the expression
		if len(trig.When) > 0 {
			whens[i] = framework.InterpretedFunction{
				ID:         trigFuncs[i].ID, // Assign the same ID just so we have a valid one for later
				ReturnType: pgtypes.Bool,
				Statements: trig.When,
			}
		}
	}

	tgOp := ""
	switch te.Source.(type) {
	case *plan.InsertInto:
		tgOp = "INSERT"
	case *plan.Update:
		tgOp = "UPDATE"
	case *plan.DeleteFrom:
		tgOp = "DELETE"
	case *plan.Truncate:
		tgOp = "TRUNCATE"
	}

	return &triggerExecutionIter{
		functions: trigFuncs,
		whens:     whens,
		split:     te.Split,
		treturn:   te.Return,
		runner:    te.Runner.Runner,
		sch:       te.Sch,
		source:    sourceIter,
		tgOp:      tgOp,
		timing:    te.Timing,
	}, nil
}

// Schema implements the interface sql.ExecBuilderNode.
func (te *TriggerExecution) Schema(ctx *sql.Context) sql.Schema {