Question

以下代码有时输出2。为什么等待组没有等待所有goroutine完成？

type Scratch struct {
    //sync.RWMutex
    Itch []int
}

func (s *Scratch) GoScratch(done chan bool, j int) error {

    var ws sync.WaitGroup

    if len(s.Itch) == 0 {
            s.Rash = make([]int, 0)
    }
    for i := 0; i < j; i++ {
            ws.Add(1)
            go func (i int) {
                    defer ws.Done()

                   s.Rash = append(s.Rash, i) 
            }(i)
    }
    ws.Wait()
    done<- true
    return nil
}

func main() {
    done := make(chan bool, 3)
    s := &Scratch{}
    err := s.GoScratch(done, 3)
    if err != nil {
            log.Println("Error:%v",err)
    }
    <-done
    log.Println("Length: ", len(s.Rash)) 
}`

奇怪的是我无法通过主函数输出2但是当我使用测试用例时它有时输出2。

Answer 1

您的代码中存在竞争条件。就在这里：

go func (i int) {
    defer ws.Done()
    // race condition on s.Rash access
    s.Rash = append(s.Rash, i) 
}(i)

由于所有goroutine同时访问s.Rash，这可能会导致切片更新被覆盖。尝试使用sync.Mutex锁定运行相同的代码以防止这种情况：

// create a global mutex
var mutex = &sync.Mutex{}

// use mutex to prevent race condition
go func (i int) {
    defer ws.Done()
    defer mutex.Unlock() // ensure that mutex unlocks

    // Lock the resource before accessing it
    mutex.Lock()
    s.Rash = append(s.Rash, i) 
}(i)

您可以详细了解这个here和here。

Answer 2

如果您使用竞赛检测器运行代码

go test -race .

您将在切片s.Rash上找到竞争条件。

为什么这个WaitGroup有时不等待所有goroutines？

2 个答案: