我有以下代码:
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
player.Exec(command, func(response map[string]interface{}){
if response["statusCode"]==float64(0) { // ok
w.Write([]byte(response["statusMessage"].(string)))
player.SendMessage("<b>"+response["statusMessage"].(string)+"</b>")
fmt.Println("playerExec: "+time.Now().Format("20060102150405"))
} else { // failed to process
w.WriteHeader(http.StatusBadRequest) // 400
w.Write([]byte(response["statusMessage"].(string)))
player.SendMessage(response["statusMessage"].(string))
}
})
// Time.Sleep(Time.Seconds*2)
fmt.Println("cmd: "+time.Now().Format("20060102150405"))
})
player.Exec()
的处理需要一些时间(因为它会启动WebSocket连接),因此,在一段时间后会调用回调函数(请参见下面的证明)。已经晚了,所以我看到以下错误:另外,当我在浏览器中打开“/”页面时,看不到任何内容。
如果将
Time.Sleep()
添加到我的代码中(请参见注释行),那么我将看到其内容。Exec()代码为here。
日志显示以下内容-
cmd: 20200612192659
playerExec: 20200612192659
有什么方法可以等待回调函数返回吗?
最佳答案
是的,您可以按以下方式使用sync.WaitGroup
:
wg := sync.WaitGroup{}
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
wg.Add(1) // add a waitgroup before calling the async function
player.Exec(command, func(response map[string]interface{}){
defer wg.Done() // release when this function returns
if response["statusCode"]==float64(0) { // ok
w.Write([]byte(response["statusMessage"].(string)))
player.SendMessage("<b>"+response["statusMessage"].(string)+"</b>")
fmt.Println("playerExec: "+time.Now().Format("20060102150405"))
} else { // failed to process
w.WriteHeader(http.StatusBadRequest) // 400
w.Write([]byte(response["statusMessage"].(string)))
player.SendMessage(response["statusMessage"].(string))
}
})
wg.Wait() // this will block until all the resources are released
fmt.Println("cmd: "+time.Now().Format("20060102150405"))
})