出现 Golang waitGroup.Wait() 死锁的解决方法

怎么入门Golang编程？需要学习哪些知识点？这是新手们刚接触编程时常见的问题；下面golang学习网就来给大家整理分享一些知识点，希望能够给初学者一些帮助。本篇文章就来介绍《出现 Golang waitGroup.Wait() 死锁的解决方法》，涉及到，有需要的可以收藏一下

我正在尝试编写一种代码，该代码可以并发读取文件并将内容发布到一个通道。

这是我的代码的链接，以及代码：

func main() {
    bufferSize := int64(10)
    f, err := os.Open("tags-c.csv")
    if err != nil {
        panic(err)
    }
    fileinfo, err := f.Stat()
    if err != nil {
        fmt.Println(err)
        return
    }
    filesize := int64(fileinfo.Size())
    fmt.Println(filesize)
    routines := filesize / bufferSize
    if remainder := filesize % bufferSize; remainder != 0 {
        routines++
    }
    fmt.Println("Total routines : ", routines)

    channel := make(chan string, 10)
    wg := &sync.WaitGroup{}

    for i := int64(0); i < int64(routines); i++ {
        wg.Add(1)
        go read(i*bufferSize, f, channel, bufferSize, filesize, wg)

    }
    fmt.Println("waiting")
    wg.Wait()
    fmt.Println("wait over")
    close(channel)

    readChannel(channel)
}

func readChannel(channel chan string) {
    for {
        data, more := <-channel
        if more == false {
            break
        }
        fmt.Print(data)
    }
}

func read(seek int64, file *os.File, channel chan string, bufferSize int64, filesize int64, wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Println("read :: ", seek)
    var buf []byte
    if filesize < bufferSize {
        buf = make([]byte, filesize)
    } else if (filesize - seek) < bufferSize {
        buf = make([]byte, filesize-seek)
    } else {
        buf = make([]byte, bufferSize)
    }

    n, err := file.ReadAt(buf, seek)
    if err != nil {
        log.Printf("loc %d err: %v", seek, err)
        return
    }
    if n > 0 {
        channel <- string(buf[:n])
        fmt.Println("ret :: ", seek)
    }
}

我尝试在线查看，但令我惊讶的是我已经解决了提到的解决方案。任何帮助将不胜感激。

解决方案

问题是，您希望所有已启动的 reader goroutine 都完成，然后再继续耗尽它们传递结果的通道。

并且通道是有缓冲的，最多可以容纳10个元素。一旦 10 个 goroutine 向其发送消息，其余的 goroutine 将被阻塞，因此它们永远不会完成（因为从该通道读取数据只能在它们全部返回后才开始：这就是死锁）。

因此，您应该启动另一个 goroutine 来与 reader goroutine 同时接收结果：

done := make(chan struct{})
go readchannel(channel, done)

fmt.println("waiting")
wg.wait()
fmt.println("wait over")
close(channel)

// wait for completion of collecting the results:
<-done

读取通道的地方应该是 for range （当通道关闭并且已从通道接收到在通道关闭之前发送的所有值时终止）：

func readChannel(channel chan string, done chan struct{}) {
    for data := range channel {
        fmt.Print(data)
    }
    close(done)
}

请注意，我使用了 done 通道，因此主 goroutine 也会等待接收结果的 goroutine 完成。

另请注意，由于在大多数情况下磁盘 io 是瓶颈而不是 cpu，并且由于从多个 goroutine 传递和接收结果也有一些开销，因此您很可能不会看到任何改进阅读该文件同时来自多个 goroutine。

终于介绍完啦！小伙伴们，这篇关于《出现 Golang waitGroup.Wait() 死锁的解决方法》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！