Method Execute

backend/plugin/db/pg/pg.go:382–486

Execute will execute the statement. For CREATE DATABASE statement, some types of databases such as Postgres will not use transactions to execute the statement but will still use transactions to execute the rest of statements.

(ctx context.Context, statement string, opts db.ExecuteOptions)

// Execute will execute the statement. For CREATE DATABASE statement, some types of databases such as Postgres
// will not use transactions to execute the statement but will still use transactions to execute the rest of statements.
func (d *Driver) Execute(ctx context.Context, statement string, opts db.ExecuteOptions) (int64, error) {
	if opts.CreateDatabase {
		if err := d.createDatabaseExecute(ctx, statement); err != nil {
			return 0, err
		}
		return 0, nil
	}

	// Parse transaction mode from the script
	config, cleanedStatement := base.ParseTransactionConfig(statement)
	statement = cleanedStatement
	transactionMode := config.Mode

	// Apply default when transaction mode is not specified
	if transactionMode == common.TransactionModeUnspecified {
		transactionMode = common.GetDefaultTransactionMode()
	}

	owner, err := d.GetCurrentDatabaseOwner(ctx)
	if err != nil {
		return 0, err
	}

	var commands []base.Statement
	var nonTransactionAndSetRoleStmts []base.Statement
	var isPlsql bool
	// HACK(p0ny): always split for pg
	//nolint
	if true || len(statement) <= common.MaxSheetCheckSize {
		singleSQLs, err := pgparser.SplitSQL(statement)
		if err != nil {
			return 0, err
		}
		commands = base.FilterEmptyStatements(singleSQLs)

		// If the statement is a single statement and is a PL/pgSQL block,
		// we should execute it as a single statement without transaction.
		// If the statement is a PL/pgSQL block, we should execute it as a single statement.
		// https://www.postgresql.org/docs/current/plpgsql-control-structures.html
		if len(singleSQLs) == 1 && pgparser.IsPlSQLBlock(singleSQLs[0].Text) {
			isPlsql = true
		}

		var tmpCommands []base.Statement
		for _, command := range commands {
			switch {
			case isSetRoleStatement(command.Text):
				nonTransactionAndSetRoleStmts = append(nonTransactionAndSetRoleStmts, command)
			case IsNonTransactionStatement(command.Text):
				nonTransactionAndSetRoleStmts = append(nonTransactionAndSetRoleStmts, command)
				continue
			case isSuperuserStatement(command.Text) && d.connectionCtx.TenantMode:
				// Use superuser privilege to run privileged statements.
				slog.Info("Use superuser privilege to run privileged statements", slog.String("statement", command.Text))
				ct := command.Text
				if !strings.HasSuffix(strings.TrimRightFunc(ct, unicode.IsSpace), ";") {
					ct += ";"
				}