国产探花免费观看_亚洲丰满少妇自慰呻吟_97日韩有码在线_资源在线日韩欧美_一区二区精品毛片,辰东完美世界有声小说,欢乐颂第一季,yy玄幻小说排行榜完本

首頁 > 網站 > 幫助中心 > 正文

GOLANG使用Context管理關聯goroutine的方法

2024-07-09 22:41:23
字體:
來源:轉載
供稿:網友

一般一個業務很少不用到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())}
發表評論 共有條評論
用戶名: 密碼:
驗證碼: 匿名發表
主站蜘蛛池模板: 武清区| 昆明市| 瑞丽市| 咸阳市| 林州市| 延安市| 高阳县| 连云港市| 肇州县| 宜州市| 临泉县| 江北区| 青神县| 巴楚县| 女性| 古田县| 七台河市| 许昌市| 辉县市| 上虞市| 武川县| 桦甸市| 凤台县| 永清县| 农安县| 德令哈市| 金门县| 丹寨县| 襄垣县| 怀集县| 平定县| 新余市| 义乌市| 辉南县| 敖汉旗| 屏边| 乐安县| 香河县| 光泽县| 海阳市| 韶关市|