sync.WaitGroup使用遇到的坑

一、问题

最近在写的一个小工具,需要使用到 goroutine 之间的同步等待问题。之前也写过&nbsp;<a href="https://www.361way.com/golang-sync-channel/5979.html" target="_blank" rel="noopener noreferrer">golang channel与sync.WaitGroup同步</a>&nbsp;,当时测试的时候使用了下<a href="https://golang.org/pkg/sync/#example_WaitGroup" target="_blank" rel="noopener noreferrer">官方示例</a>也可以正常执行,不过最近在该工具使用时调用该示例的方法进行使用时,发现出现了无法正常退出的问题。见下图:



<img src="https://www.361way.com/wp-content/uploads/2019/04/sync-waitgroup.png" width="445" height="441" title="sync-waitgroup" alt="sync-waitgroup" align="" />



上面是使用跟踪工具跟踪时的结果。发现正常调用也返回了想要的结果,但是最后看情形,应该是卡在main进程还在一直wait等待上。具体调用的主要代码类似如下:



<br />
var wg sync.WaitGroup
            lines := strings.Split(Ioutil(*ipfile), "\n")
            fmt.Println(lines)
            fmt.Println(len(lines))
            //wg.Add(len(lines))
            for _, r := range lines {
                if r != "" {
                    wg.Add(1)
                    fmt.Println("value is" + r)
                    go Multrun(r, data, wg)
                }
            }
            wg.Wait()
        } else {
            fmt.Printf("didn't hava ip address,use \t%s -h\n check", os.Args[0])
            os.Exit(0)
        }
    }
}
func Multrun(ip string, data []byte, wg sync.WaitGroup) {
    defer wg.Done()
    url := "http://" + ip + ":10000"
    x, _ := DoBytesPost(url, data)
    fmt.Println(ip + "\n========================\n" + string(x))
}
<br />

二、解决方法

后来在查找了很多资料以后,发现如下一段话:golang里如果方法传递的不是地址,那么就会做一个拷贝,所以这里调用的wg根本就不是一个对象。这句话啥意思呢?就是如果在方法里不传递对象的内存地址,传过去的内容就会变成一个新的对象,这就导致上面我们给给Multrun的wg sync.WaitGroup是一个新的对象,而老的在main里的对象是持续等待不到结束调用的指令的。如何解决呢?传内存地址给函数。



<br />
go doSomething(i, &wg, ch)
func doSomething(index int, wg *sync.WaitGroup, ch chan int) {

三、演示代码

<br />
package main
import (
    "fmt"
    "strconv"
    "sync"
)
func main() {
    var wg sync.WaitGroup
    ch := make(chan int, 1000)
    for i := 0; i < 1000; i++ {
        wg.Add(1)
        go doSomething(i, wg, ch)
    }
    wg.Wait()
    fmt.Println("all done")
    for i := 0; i < 1000; i++ {
        dd := <-ch
        fmt.Println("from ch:" + strconv.Itoa(dd))
    }
}
func doSomething(index int, wg sync.WaitGroup, ch chan int) {
    defer wg.Done()
    fmt.Println("start done:" + strconv.Itoa(index))
    //time.Sleep(20 * time.Millisecond)
    ch <- index
}
上面是一段演示代码,使用上面的代码直接运行时会报deadlock的错误,如下:



<br />
fatal error: all goroutines are asleep - deadlock!
goroutine 1 [semacquire]:
sync.runtime_Semacquire(0xc4200b401c)
        /usr/lib/golang/src/runtime/sema.go:56 +0x39
sync.(*WaitGroup).Wait(0xc4200b4010)
        /usr/lib/golang/src/sync/waitgroup.go:131 +0x72
main.main()
        /tmp/test/test.go:16 +0xe7
解决方法也比较简单,上面第二步已经分析过了,具体你可以修改试试让程序正常运行。



<br />

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注