主頁 > 知識庫 > GOLANG使用Context管理關(guān)聯(lián)goroutine的方法

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

熱門標(biāo)簽:廣州呼叫中心外呼系統(tǒng) 南通如皋申請開通400電話 地圖標(biāo)注的汽車標(biāo) 浙江高速公路地圖標(biāo)注 西部云谷一期地圖標(biāo)注 江西轉(zhuǎn)化率高的羿智云外呼系統(tǒng) 學(xué)海導(dǎo)航地圖標(biāo)注 中國地圖標(biāo)注省會高清 高德地圖標(biāo)注口訣

一般一個業(yè)務(wù)很少不用到goroutine的,因為很多方法是需要等待的,例如http.Server.ListenAndServe這個就是等待的,除非關(guān)閉了Server或Listener,否則是不會返回的。除非是一個API服務(wù)器,否則肯定需要另外起goroutine發(fā)起其他的服務(wù),而且對于API服務(wù)器來說,在http.Handler的處理函數(shù)中一般也需要起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收到退出信號后,應(yīng)該通知server退出。暴力一點的是直接調(diào)用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的所有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())
}

這個方式可以在新開goroutine時繼續(xù)使用,譬如新加一個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來讓服務(wù)器退出,可以這么寫:

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。

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

您可能感興趣的文章:
  • golang中for循環(huán)遍歷channel時需要注意的問題詳解
  • golang實現(xiàn)基于channel的通用連接池詳解
  • Golang優(yōu)雅關(guān)閉channel的方法示例
  • golang中單向channel的語法介紹
  • 解決Golang中g(shù)oroutine執(zhí)行速度的問題
  • golang gin 框架 異步同步 goroutine 并發(fā)操作
  • 關(guān)于golang利用channel和goroutine完成統(tǒng)計素數(shù)的思路

標(biāo)簽:吐魯番 德宏 曲靖 常州 貴州 許昌 保定 東營

巨人網(wǎng)絡(luò)通訊聲明:本文標(biāo)題《GOLANG使用Context管理關(guān)聯(lián)goroutine的方法》,本文關(guān)鍵詞  GOLANG,使用,Context,管理,關(guān)聯(lián),;如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題,煩請?zhí)峁┫嚓P(guān)信息告之我們,我們將及時溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò),涉及言論、版權(quán)與本站無關(guān)。
  • 相關(guān)文章
  • 下面列出與本文章《GOLANG使用Context管理關(guān)聯(lián)goroutine的方法》相關(guān)的同類信息!
  • 本頁收集關(guān)于GOLANG使用Context管理關(guān)聯(lián)goroutine的方法的相關(guān)信息資訊供網(wǎng)民參考!
  • 推薦文章