Method executeBoundPlan

pgserver/duck_handler.go:492–593

executeBoundPlan is a QueryExecutor that calls QueryWithBindings on the given engine using the given query and parsed statement, which may be nil.

(ctx *sql.Context, query string, _ tree.Statement, stmt *duckdb.Stmt, vars []any)

	// return h.e.QueryWithBindings(ctx, query, parsed, nil, nil)

	sql.IncrementStatusVariable(ctx, "Questions", 1)
	if _, ok := parsed.(*tree.Select); ok {
		sql.IncrementStatusVariable(ctx, "Com_select", 1)
	}

	var (
		schema sql.Schema
		iter   sql.RowIter
		rows   *stdsql.Rows
		result stdsql.Result
		err    error
	)

	// NOTE: The query is parsed using Postgres parser, which does not support all DuckDB syntax.
	//   Consequently, the following classification is not perfect.
	switch parsed := parsed.(type) {
	case *tree.BeginTransaction, *tree.CommitTransaction, *tree.RollbackTransaction,
		*tree.CreateTable, *tree.DropTable, *tree.AlterTable, *tree.CreateIndex, *tree.DropIndex,
		*tree.Update, *tree.Delete, *tree.Truncate, *tree.CopyFrom, *tree.CopyTo, *tree.SetVar:
		result, err = adapter.Exec(ctx, query)
		if err != nil {
			break
		}
		affected, _ := result.RowsAffected()
		insertId, _ := result.LastInsertId()
		schema = types.OkResultSchema
		iter = sql.RowsToRowIter(sql.NewRow(types.OkResult{
			RowsAffected: uint64(affected),
			InsertID:     uint64(insertId),
		}))
	case *tree.Insert:
		if _, ok := postgresInsertReturningRows(parsed); ok {
			rows, err = adapter.QueryCatalog(ctx, query)
			if err != nil {
				break
			}
			schema, err = pgtypes.InferSchema(rows)
			if err != nil {
				rows.Close()
				break
			}
			iter, err = backend.NewSQLRowIter(rows, schema)
			if err != nil {
				rows.Close()
			}
			break
		}
		result, err = adapter.Exec(ctx, query)
		if err != nil {
			break
		}
		affected, _ := result.RowsAffected()
		insertID, _ := result.LastInsertId()
		schema = types.OkResultSchema
		iter = sql.RowsToRowIter(sql.NewRow(types.OkResult{
			RowsAffected: uint64(affected),
			InsertID:     uint64(insertID),
		}))