Browse Source

Reduce data races (#14549)

* Add race conditions into test

* Fix Race in GetManager()

* DataAsync() use error chan

* just log no chan

* finish
tags/v1.15.0-dev
6543 3 years ago
parent
commit
87009ab40a
No account linked to committer's email address
3 changed files with 18 additions and 6 deletions
  1. 6
    4
      modules/git/blob_nogogit.go
  2. 3
    2
      modules/process/manager.go
  3. 9
    0
      modules/process/manager_test.go

+ 6
- 4
modules/git/blob_nogogit.go View File

@@ -11,6 +11,8 @@ import (
"io"
"strconv"
"strings"

gitea_log "code.gitea.io/gitea/modules/log"
)

// Blob represents a Git object.
@@ -27,13 +29,13 @@ type Blob struct {
// Calling the Close function on the result will discard all unread output.
func (b *Blob) DataAsync() (io.ReadCloser, error) {
stdoutReader, stdoutWriter := io.Pipe()
var err error

go func() {
stderr := &strings.Builder{}
err = NewCommand("cat-file", "--batch").RunInDirFullPipeline(b.repoPath, stdoutWriter, stderr, strings.NewReader(b.ID.String()+"\n"))
err := NewCommand("cat-file", "--batch").RunInDirFullPipeline(b.repoPath, stdoutWriter, stderr, strings.NewReader(b.ID.String()+"\n"))
if err != nil {
err = ConcatenateError(err, stderr.String())
gitea_log.Error("Blob.DataAsync Error: %v", err)
_ = stdoutWriter.CloseWithError(err)
} else {
_ = stdoutWriter.Close()
@@ -50,8 +52,8 @@ func (b *Blob) DataAsync() (io.ReadCloser, error) {
return &LimitedReaderCloser{
R: bufReader,
C: stdoutReader,
N: int64(size),
}, err
N: size,
}, nil
}

// Size returns the uncompressed size of the blob

+ 3
- 2
modules/process/manager.go View File

@@ -25,6 +25,7 @@ var (
// ErrExecTimeout represent a timeout error
ErrExecTimeout = errors.New("Process execution timeout")
manager *Manager
managerInit sync.Once

// DefaultContext is the default context to run processing commands in
DefaultContext = context.Background()
@@ -48,11 +49,11 @@ type Manager struct {

// GetManager returns a Manager and initializes one as singleton if there's none yet
func GetManager() *Manager {
if manager == nil {
managerInit.Do(func() {
manager = &Manager{
processes: make(map[int64]*Process),
}
}
})
return manager
}


+ 9
- 0
modules/process/manager_test.go View File

@@ -12,6 +12,15 @@ import (
"github.com/stretchr/testify/assert"
)

func TestGetManager(t *testing.T) {
go func() {
// test race protection
_ = GetManager()
}()
pm := GetManager()
assert.NotNil(t, pm)
}

func TestManager_Add(t *testing.T) {
pm := Manager{processes: make(map[int64]*Process)}


Loading…
Cancel
Save