sync.WaitGroup使用遇到的坑

一、问题

最近在写的一个小工具，需要使用到 goroutine 之间的同步等待问题。之前也写过 golang channel与sync.WaitGroup同步 ，当时测试的时候使用了下官方示例也可以正常执行，不过最近在该工具使用时调用该示例的方法进行使用时，发现出现了无法正常退出的问题。见下图：

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

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))
}

二、解决方法

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

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

三、演示代码

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的错误，如下：

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

解决方法也比较简单，上面第二步已经分析过了，具体你可以修改试试让程序正常运行。