Method executeQuery

pgserver/duck_handler.go:406–488

executeQuery 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, parsed tree.Statement, _ *duckdb.Stmt, _ []any)

// executeQuery is a QueryExecutor that calls QueryWithBindings on the given engine using the given query and parsed
// statement, which may be nil.
func (h *DuckHandler) executeQuery(ctx *sql.Context, query string, parsed tree.Statement, _ *duckdb.Stmt, _ []any) (sql.Schema, sql.RowIter, *sql.QueryFlags, error) {
	// 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.(type) {
	case *tree.BeginTransaction, *tree.CommitTransaction, *tree.RollbackTransaction,
		*tree.CreateTable, *tree.DropTable, *tree.AlterTable, *tree.CreateIndex, *tree.DropIndex,
		*tree.Insert, *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.CreateDatabase:
		provider := h.GetCatalogProvider()
		if provider == nil {
			err = fmt.Errorf("database provider not found")
			break
		}
		p := parsed.(*tree.CreateDatabase)
		dbName := p.Name.String()
		err = provider.CreateCatalog(dbName, p.IfNotExists)
		if err != nil {
			break
		}
		schema = types.OkResultSchema
		iter = sql.RowsToRowIter(sql.NewRow(types.OkResult{}))
	case *tree.DropDatabase:
		provider := h.GetCatalogProvider()
		if provider == nil {
			err = fmt.Errorf("database provider not found")
			break
		}
		p := parsed.(*tree.DropDatabase)
		dbName := parsed.(*tree.DropDatabase).Name.String()
		err = provider.DropCatalog(dbName, p.IfExists)
		if err != nil {
			break