Question

我正在开发一个小型实用程序，该实用程序需要在动态范围的项目（可以是100或可以是100000）上进行迭代，并将其放入通道中。从该通道，另一个功能读取项目并分别对每个项目进行一些处理。我正在尝试使用sync.WaitGroup确保在处理通道中的所有项目之前我的实用程序不会退出。由于我对渠道和等待组还很陌生，因此我陷入了错误panic: sync: WaitGroup is reused before previous Wait has returned

https://play.golang.org/p/nMw3END_9qw

package main

import (
    "fmt"
    "github.com/dchest/uniuri"
    "math/rand"
    "sync"
    "time"
)

var wg sync.WaitGroup
var count = 0

func printMe(msg string) {
    time.Sleep(1 * time.Second)
    fmt.Println(count, msg)
}

func makeMePrint(ch chan string) {
    for s := range ch {
        count++
        wg.Add(1)
        printMe(s)
        wg.Done()
    }

}

func writePrint(ch chan<- string) {
    fmt.Println("Starting to insert data in channel")
    for i := 1; i <= rand.Intn(30); i++ {
        s := uniuri.New()
        ch <- s
    }
    fmt.Println("We are done inserting all data in the channel")
    close(ch)
}

func main() {

    var ch = make(chan string)

    go writePrint(ch)
    go makeMePrint(ch)
    time.Sleep(1 * time.Second)
    wg.Wait()
}

这是我正在从事的工作的主要思想（不是确切的代码，而是功能数量相同的完全相同的体系结构）。

如何确保仅当通道中的所有项目都是进程时才退出实用程序。

感谢您的帮助。

Answer 1

我终于让它工作了。

package main

import (
    "fmt"
    "github.com/dchest/uniuri"
    "math/rand"
    "sync"
    "time"
)

var count = 0

func printMe(msg string) {
    time.Sleep(1 * time.Second)
    fmt.Println(count, msg)
}

func makeMePrint(wg *sync.WaitGroup, ch chan string) {
    wg.Add(1)
    defer wg.Done()

    for s := range ch {
        count++
        printMe(s)

    }

}

func writePrint(wg *sync.WaitGroup, ch chan<- string) {
    wg.Add(1)
    defer wg.Done()

    fmt.Println("Starting to insert data in channel")
    for i := 1; i <= rand.Intn(30); i++ {
        s := uniuri.New()
        ch <- s
    }
    fmt.Println("We are done inserting all data in the channel")
    close(ch)
}

func main() {
    wg := &sync.WaitGroup{}
    var ch = make(chan string)

    go writePrint(wg, ch)
    go makeMePrint(wg, ch)
    time.Sleep(1 * time.Second)
    wg.Wait()
}

通过动态循环从通道读取goroutine并获取waitgroup，在返回之前的值之前已被重用

1 个答案: