Method executeInTransactionMode

backend/plugin/db/snowflake/snowflake.go:253–301

executeInTransactionMode executes statements within a single transaction

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

// executeInTransactionMode executes statements within a single transaction
func (d *Driver) executeInTransactionMode(ctx context.Context, statement string, opts db.ExecuteOptions) (int64, error) {
	tx, err := d.db.BeginTx(ctx, nil)
	if err != nil {
		opts.LogTransactionControl(storepb.TaskRunLog_TransactionControl_BEGIN, err.Error())
		return 0, err
	}
	opts.LogTransactionControl(storepb.TaskRunLog_TransactionControl_BEGIN, "")

	committed := false
	defer func() {
		if !committed {
			if err := tx.Rollback(); err != nil {
				opts.LogTransactionControl(storepb.TaskRunLog_TransactionControl_ROLLBACK, err.Error())
			} else {
				opts.LogTransactionControl(storepb.TaskRunLog_TransactionControl_ROLLBACK, "")
			}
		}
	}()

	// To submit a variable number of SQL statements in the statement field, set MULTI_STATEMENT_COUNT to 0."
	// https://docs.snowflake.com/en/developer-guide/sql-api/submitting-multiple-statements
	mctx := snow.WithMultiStatement(ctx, 0 /* MULTI_STATEMENT_COUNT */)

	// Log the entire multi-statement execution
	opts.LogCommandExecute(&storepb.Range{Start: 0, End: int32(len(statement))}, statement)

	result, err := tx.ExecContext(mctx, statement)
	if err != nil {
		opts.LogCommandResponse(0, nil, err.Error())
		return 0, err
	}

	if err := tx.Commit(); err != nil {
		opts.LogTransactionControl(storepb.TaskRunLog_TransactionControl_COMMIT, err.Error())
		return 0, err
	}
	opts.LogTransactionControl(storepb.TaskRunLog_TransactionControl_COMMIT, "")
	committed = true

	rowsAffected, err := result.RowsAffected()
	// Since we cannot differentiate DDL and DML yet, we have to ignore the error.
	if err != nil {
		slog.Debug("rowsAffected returns error", log.BBError(err))
		opts.LogCommandResponse(0, nil, "")
		return 0, nil
	}
	opts.LogCommandResponse(rowsAffected, nil, "")
	return rowsAffected, nil
}

// executeInAutoCommitMode executes statements with autocommit enabled (no explicit transaction)
func (d *Driver) executeInAutoCommitMode(ctx context.Context, statement string, opts db.ExecuteOptions) (int64, error) {