golang 如何判断一个 goroutine 函数执行完成？

Question

用 golang 的一个定时任务的第三方包，每隔 5s 执行 foo 函数，如下：func foo() {    onlineRooms := getOnlineRooms()     for _, room := range onlineRooms {         go doSomeThing(room)     } }这里取出所有在线的房间ID，然后启用 goroutine 执行 doSomeThing 函数，这个函数里有计算，从 Redis，MySQL 读写数据等操作，如果第一轮调用没执行完，就开始第二轮调用，就会有计算错误。那么我想每次定时任务来的时候，判断这个函数有没有执行完，如果没有执行完就不重新调用，否则才调用，在 golang 中如何实现这个判断呢？

Answer 1

提供一个简单便于理解方案：

func foo() {
    onlineRooms := getOnlineRooms()    for _, room := range onlineRooms {        go doSomeThing(room)
    }
}func doSomeThing(room Room) {    //把redis操作换成全局变量亦可
    if redis.Get(room){        return
    }
    redis.Set(room,true)    defer redis.Set(room,false)
}

Answer 2

package main

import (

    "fmt"

    "sync"

)

var wg *sync.WaitGroup

func main() {

    wg = &sync.WaitGroup{}

    for {

        wg.Wait()

        wg.Add(1)

        go test()

    }

    fmt.Println("vim-go")

}

func test() {

    //do something

    wg.Done()

}

Answer 3

可以在room里加一个字段记录操作是否完成

func foo() {    onlineRooms := getOnlineRooms()
    for _, room := range onlineRooms {
        if room.Done {
            go doSomeThing(room)
        }
    }
}