Method Execute

backend/plugin/db/redshift/redshift.go:158–204

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 d.datashare {
		return 0, errors.Errorf("datashare database cannot be updated")
	}
	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()
	}

	var commands []base.Statement
	oneshot := true
	if len(statement) <= common.MaxSheetCheckSize {
		singleSQLs, err := redshiftparser.SplitSQL(statement)
		if err != nil {
			return 0, err
		}
		commands = base.FilterEmptyStatements(singleSQLs)
		if len(commands) <= common.MaximumCommands {
			oneshot = false
		}
	}
	if oneshot {
		commands = []base.Statement{
			{
				Text: statement,
			},
		}
	}

	// Execute based on transaction mode
	if transactionMode == common.TransactionModeOff {
		return d.executeInAutoCommitMode(ctx, commands, opts)
	}
	return d.executeInTransactionMode(ctx, commands, opts)
}

func (d *Driver) createDatabaseExecute(ctx context.Context, statement string) error {
	databaseName, err := getDatabaseInCreateDatabaseStatement(statement)