当使用 Channel 和 WaitGroup 时如何避免死锁问题

本篇文章向大家介绍《当使用 Channel 和 WaitGroup 时如何避免死锁问题》，主要包括，具有一定的参考价值，需要的朋友可以参考一下。

我是 go 新手，目前正在尝试运行一个创建文件并返回其文件名的函数，并使其同时运行。

我决定尝试使用 goroutine 和 waitgroup 来完成此任务。当我使用这种方法时，我最终得到的列表大小比输入大小小几百个文件。例如。对于 5,000 个文件，我创建了大约 4,700 个文件。

我相信这是由于一些竞争条件造成的：

wg := sync.waitgroup{}

filenames := make([]string, 0)

for i := 0; i < totalfiles; i++ {
    wg.add(1)
    go func() {
        defer wg.done()
        filenames = append(filenames, createfile())
    }()
}

wg.wait()

return filenames, nil

不要通过共享内存进行通信；通过通信共享内存。

我尝试使用通道“通过通信共享内存”。每当我这样做时，似乎就会陷入僵局，我似乎无法理解为什么。有人能指出我正确的方向，正确地一起使用通道和等待组，以便将所有创建的文件保存到共享数据结构吗？

这是为我产生死锁的代码（致命错误：所有 goroutine 都在睡觉 - 死锁！）：

wg := sync.WaitGroup{}

filenames := make([]string, 0)
ch := make(chan string)

for i := 0; i < totalFiles; i++ {
    wg.Add(1)
    go func() {
        defer wg.Done()
        ch <- createFile()
    }()
}

wg.Wait()

for i := range ch {
    filenames = append(filenames, i)
}

return filenames, nil

谢谢！

正确答案

第一个有一场比赛。您必须保护对 filenames 的访问：

mu:=sync.mutex{}
for i := 0; i < totalfiles; i++ {
    wg.add(1)
    go func() {
        defer wg.done()
        mu.lock()
        defer mu.unlock()
        filenames = append(filenames, createfile())
    }()
}

对于第二种情况，您正在等待 goroutine 完成，但 goroutine 只能在您从通道中读取数据后才能完成，因此会出现死锁。您可以通过在单独的 goroutine 中读取通道来修复它。

go func() {
  for i := range ch {
      filenames = append(filenames, i)
  }
}()

wg.wait()
close(ch) // required, so the goroutine can terminate

return filenames, nil

有无锁版本，如果文件数量固定的话：

filenames := make([]string, totalFiles)
for i := 0; i < totalFiles; i++ {
    wg.Add(1)
    go func(index int) {
        defer wg.Done()
        filenames[index]=createFile()
    }(i)
}
wg.Wait()

终于介绍完啦！小伙伴们，这篇关于《当使用 Channel 和 WaitGroup 时如何避免死锁问题》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！