Method executeInAutoCommitMode

backend/plugin/db/mssql/mssql.go:265–316

executeInAutoCommitMode executes statements sequentially in auto-commit mode

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

// executeInAutoCommitMode executes statements sequentially in auto-commit mode
func (d *Driver) executeInAutoCommitMode(ctx context.Context, statement string, opts db.ExecuteOptions) (int64, error) {
	totalAffectRows := int64(0)

	batch := tsqlbatch.NewBatcher(statement)

	for idx := 0; ; {
		command, err := batch.Next()
		if err != nil {
			if err == io.EOF {
				// Try send the last batch to server.
				v := batch.Batch()
				if v != nil && len(v.Text) > 0 {
					opts.LogCommandExecute(&storepb.Range{Start: int32(v.Start), End: int32(v.End)}, v.Text)
					rowsAffected, err := d.executeAutoCommit(ctx, v.Text)
					if err != nil {
						opts.LogCommandResponse(0, nil, err.Error())
						return totalAffectRows, err
					}
					opts.LogCommandResponse(rowsAffected, []int64{rowsAffected}, "")
					totalAffectRows += rowsAffected
				}
				break
			}
			return 0, errors.Wrapf(err, "failed to get next batch for statement: %s", batch.Batch().Text)
		}
		if command == nil {
			continue
		}
		switch v := command.(type) {
		case *tsqlbatch.GoCommand:
			b := batch.Batch()
			// Execute the batch in auto-commit mode
			idx++
			for i := uint(0); i < v.Count; i++ {
				opts.LogCommandExecute(&storepb.Range{Start: int32(b.Start), End: int32(b.End)}, b.Text)
				rowsAffected, err := d.executeAutoCommit(ctx, b.Text)
				if err != nil {
					opts.LogCommandResponse(0, nil, err.Error())
					// In auto-commit mode, we stop at the first error
					return totalAffectRows, err
				}
				opts.LogCommandResponse(rowsAffected, []int64{rowsAffected}, "")
				totalAffectRows += rowsAffected
			}
		default:
			return 0, errors.Errorf("unsupported command type: %T", v)
		}
		batch.Reset(nil)
	}

	return totalAffectRows, nil
}

// executeAutoCommit executes a single statement in auto-commit mode
func (d *Driver) executeAutoCommit(ctx context.Context, statement string) (int64, error) {