Golang使用协程实现批量获取数据.docx

第

Golang使用协程实现批量获取数据

目录使用channel使用WaitGroup应用到实践服务端经常需要返回一个列表，里面包含很多用户数据，常规做法当然是遍历然后读缓存。

使用Go语言后，可以并发获取，极大提升效率。

使用channel

packagemain

import(

fmt

time

funcadd2(a,bint,chchanint){

c:=a+b

fmt.Printf(%d+%d=%d\n,a,b,c)

ch-1//执行完了就写一条表示自己完成了

funcmain(){

start:=time.Now()

chs:=make([]chanint,10)

fori:=0;ii++{

chs[i]=make(chanint)

goadd2(1,i,chs[i])//分配了10个协程出去了

for_,ch:=rangechs{

-ch//循环等待，要每个完成才能继续，不然就等待

end:=time.Now()

consume:=end.Sub(start).Seconds()

fmt.Println(程序执行耗时(s)：,consume)

在每个协程的add()函数业务逻辑完成后，我们通过ch-1语句向对应的通道中发送一个数据。

在所有的协程启动完成后，我们再通过-ch语句从通道切片chs中依次接收数据（不对结果做任何处理，相当于写入通道的数据只是个标识而已，表示这个通道所属的协程逻辑执行完毕）.

直到所有通道数据接收完毕，然后打印主程序耗时并退出。

使用WaitGroup

Add：WaitGroup类型有一个计数器，默认值是0，我们可以通过Add方法来增加这个计数器的值，通常我们可以通过个方法来标记需要等待的子协程数量；Done：当某个子协程执行完毕后，可以通过Done方法标记已完成，该方法会将所属WaitGroup类型实例计数器值减一，通常可以通过defer语句来调用它；Wait：Wait方法的作用是阻塞当前协程，直到对应WaitGroup类型实例的计数器值归零，如果在该方法被调用的时候，对应计数器的值已经是0，那么它将不会做任何事情

packagemain

import(

fmt

sync

funcaddNum(a,bint,deferFuncfunc()){

deferfunc(){

deferFunc()

c:=a+b

fmt.Printf(%d+%d=%d\n,a,b,c)

funcmain(){

varwgsync.WaitGroup

wg.Add(10)//等于发了10个令牌

fori:=0;ii++{

goaddNum(i,1,wg.Done)//每次执行都消耗令牌

wg.Wait()//等待令牌消耗完

需要注意的是，该类型计数器不能小于0，否则会抛出如下panic：

panic:sync:negativeWaitGroupcounter

应用到实践

funcGetManyBase(userIds[]int64)[]UserBase{

userCaches:=make([]UserBase,len(userIds))

varwgsync.WaitGroup

forindex,userId:=rangeuserIds{

wg.Add(1)

gofunc(indexint,userIdint64,userCaches[]UserBase){

userCaches[index]=NewUserCache(userId).GetBase()

wg.Done()

}(index,userId,userCaches)

wg.Wait()

returnuserCaches

}

这种写法有两个问题：

1.并发肯定带来乱序，所以要考虑需要排序的业务场景。

2.map是线程不安全的，并发读写会panic。

优化一下：

funcGetManyBase(userIds[]int64)[]UserBase{

userCaches:=make([]UserBase,len(use