一般一個業務很少不用到goroutine的，因為很多方法是需要等待的，例如http.Server.ListenAndServe這個就是等待的，除非關閉了Server或Listener，否則是不會返回的。除非是一個API服務器，否則肯定需要另外起goroutine發起其他的服務，而且對于API服務器來說，在http.Handler的處理函數中一般也需要起goroutine，如何管理這些goroutine，在GOLANG1.7提供context.Context。
先看一個簡單的，如果啟動兩個goroutine，一個是HTTP，還有個信號處理的收到退出信號做清理：
wg := sync.WaitGroup{}
defer wg.Wait()
wg.Add(1)
go func() {
defer wg.Done()
ss := make(os.Signal, 0)
signal.Notify(ss, syscall.SIGINT, syscall.SIGTERM)
for s := ss {
fmt.Println("Got signal", s)
break
}
}()
wg.Add(1)
go func() {
defer wg.Done()
svr := http.Server{ Addr:":8080", Handler:nil, }
fmt.Println(svr.ListenAndServe())
}
很清楚，起了兩個goroutine，然后用WaitGroup等待它們退出。如果它們之間沒有交互，不互相影響，那真的是蠻簡單的，可惜這樣是不行的，因為信號的goroutine收到退出信號后，應該通知server退出。暴力一點的是直接調用svr.Close()，但是如果有些請求還需要取消怎么辦呢？最好用Context了：
wg := sync.WaitGroup{}
defer wg.Wait()
ctx,cancel := context.WithCancel(context.Background())
wg.Add(1)
go func() {
defer wg.Done()
ss := make(chan os.Signal, 0)
signal.Notify(ss, syscall.SIGINT, syscall.SIGTERM)
select {
case - ctx.Done():
return
case s := - ss:
fmt.Println("Got signal", s)
cancel() // 取消請求，通知用到ctx的所有goroutine
return
}
}()
wg.Add(1)
go func() {
defer wg.Done()
defer cancel()
svr := http.Server{ Addr:":8080", Handler:nil, }
go func(){
select {
case - ctx.Done():
svr.Close()
}
}
fmt.Println(svr.ListenAndServe())
}
這個方式可以在新開goroutine時繼續使用，譬如新加一個goroutine，里面讀寫了UDPConn：
wg.Add(1)
go func() {
defer wg.Done()
defer cancel()
var conn *net.UDPConn
if conn,err = net.Dial("udp", "127.0.0.1:1935"); err != nil {
fmt.Println("Dial UDP server failed, err is", err)
return
}
fmt.Println(UDPRead(ctx, conn))
}()
UDPRead = func(ctx context.Context, conn *net.UDPConn) (err error) {
wg := sync.WaitGroup{}
defer wg.Wait()
ctx, cancel := context.WithCancel(ctx)
wg.Add(1)
go func() {
defer wg.Done()
defer cancel()
for {
b := make([]byte, core.MTUSize)
size, _, err := conn.ReadFromUDP(b)
// 處理UDP包 b[:size]
}
}()
select {
case -ctx.Done():
conn.Close()
}
return
}
如果只是用到HTTP Server，可以這么寫：
func run(ctx contex.Context) {
server := http.Server{Addr: addr, Handler: nil}
go func() {
select {
case -ctx.Done():
server.Close()
}
}()
http.HandleFunc("/api", func(w http.ResponseWriter, r *http.Request) {
})
fmt.Println(server.ListenAndServe())
}
如果需要提供一個API來讓服務器退出，可以這么寫：
func run(ctx contex.Context) {
server := http.Server{Addr: addr, Handler: nil}
ctx, cancel := context.WithCancel(ctx)
http.HandleFunc("/quit", func(w http.ResponseWriter, r *http.Request) {
cancel() // 使用局部的ctx和cancel
})
go func() {
select {
case -ctx.Done():
server.Close()
}
}()
fmt.Println(server.ListenAndServe())
}
使用局部的ctx和cancel，可以避免cancel傳入的ctx，只是影響當前的ctx。
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。
您可能感興趣的文章:- golang中for循環遍歷channel時需要注意的問題詳解
- golang實現基于channel的通用連接池詳解
- Golang優雅關閉channel的方法示例
- golang中單向channel的語法介紹
- 解決Golang中goroutine執行速度的問題
- golang gin 框架 異步同步 goroutine 并發操作
- 關于golang利用channel和goroutine完成統計素數的思路
