快捷導(dǎo)航

GOLANG使用Context管理關(guān)聯(lián)goroutine的方法

更新時(shí)間：2019年01月14日 09:32:48 作者：winlin

這篇文章主要介紹了GOLANG使用Context管理關(guān)聯(lián)goroutine的方法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

一般一個(gè)業(yè)務(wù)很少不用到goroutine的，因?yàn)楹芏喾椒ㄊ切枰却?，例?code>http.Server.ListenAndServe這個(gè)就是等待的，除非關(guān)閉了Server或Listener，否則是不會(huì)返回的。除非是一個(gè)API服務(wù)器，否則肯定需要另外起goroutine發(fā)起其他的服務(wù)，而且對(duì)于API服務(wù)器來(lái)說(shuō)，在http.Handler的處理函數(shù)中一般也需要起goroutine，如何管理這些goroutine，在GOLANG1.7提供context.Context。

先看一個(gè)簡(jiǎn)單的，如果啟動(dòng)兩個(gè)goroutine，一個(gè)是HTTP，還有個(gè)信號(hào)處理的收到退出信號(hào)做清理：

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())
}

很清楚，起了兩個(gè)goroutine，然后用WaitGroup等待它們退出。如果它們之間沒(méi)有交互，不互相影響，那真的是蠻簡(jiǎn)單的，可惜這樣是不行的，因?yàn)樾盘?hào)的goroutine收到退出信號(hào)后，應(yīng)該通知server退出。暴力一點(diǎn)的是直接調(diào)用svr.Close()，但是如果有些請(qǐng)求還需要取消怎么辦呢？最好用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() // 取消請(qǐng)求，通知用到ctx的所有g(shù)oroutine
    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())
}

這個(gè)方式可以在新開(kāi)goroutine時(shí)繼續(xù)使用，譬如新加一個(gè)goroutine，里面讀寫(xiě)了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，可以這么寫(xiě)：

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())
}

如果需要提供一個(gè)API來(lái)讓服務(wù)器退出，可以這么寫(xiě)：

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，只是影響當(dāng)前的ctx。

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: