运维之路

一、问题

最近在写的一个小工具，需要使用到 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

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