快捷導(dǎo)航

golang實現(xiàn)并發(fā)數(shù)控制的方法

更新時間：2017年12月15日 10:24:46 作者：hzwy23

下面小編就為大家分享一篇golang實現(xiàn)并發(fā)數(shù)控制的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

golang并發(fā)

談到golang這門語言，很自然的想起了他的的并發(fā)goroutine。這也是這門語言引以為豪的功能點。并發(fā)處理，在某種程度上，可以提高我們對機器的使用率，提升系統(tǒng)業(yè)務(wù)處理能力。但是并不是并發(fā)量越大越好，太大了，硬件環(huán)境就會吃不消，反而會影響到系統(tǒng)整體性能，甚至奔潰。所以，在使用golang提供便捷的goroutine時，既要能夠?qū)崿F(xiàn)開啟并發(fā)，也要學(xué)會如果控制并發(fā)量。

開啟golang并發(fā)

golang開啟并發(fā)處理非常簡單，只需要在調(diào)用函數(shù)時，在函數(shù)前邊添加上go關(guān)鍵字即可。如下邊例子所示：

package main
import (
  "fmt"
  "time"
)
type Demo struct {
  input     chan string
  output    chan string
  max_goroutine chan int
}
func NewDemo() *Demo {
  d := new(Demo)
  d.input = make(chan string, 24)
  d.output = make(chan string, 24)
  d.max_goroutine = make(chan int, 20)
  return d
}
func (this *Demo) Goroutine() {
  var i = 1000
  for {
    this.input <- time.Now().Format("2006-01-02 15:04:05")
    time.Sleep(time.Second * 1)
    if i < 0 {
      break
    }
    i--
  }
  close(this.input)
}
func (this *Demo) Handle() {
  for t := range this.input {
    fmt.Println("datatime is :", t)
    this.output <- t
  }
}
func main() {
  demo := NewDemo()
  go demo.Goroutine()
  demo.Handle()
}

上邊代碼，在調(diào)用Demo的Goroutine方法時，在前邊加上了go關(guān)鍵字，則函數(shù)Goroutine并發(fā)執(zhí)行開啟成功。

可見，在golang中開啟并發(fā)非常的方便。

下邊再來看看，在golang中，怎么實現(xiàn)并發(fā)量的控制。

當(dāng)goroutine并發(fā)執(zhí)行的任務(wù)達到一定值時，主程序等待goroutine執(zhí)行完成退出，一旦發(fā)現(xiàn)并發(fā)數(shù)量低于某一個設(shè)定的值，就從新開始執(zhí)行主程序邏輯。

實現(xiàn)代碼如下：

package main
import (
  "fmt"
  "time"
)
type Demo struct {
  input     chan string
  output    chan string
  goroutine_cnt chan int
}
func NewDemo() *Demo {
  d := new(Demo)
  d.input = make(chan string, 8192)
  d.output = make(chan string, 8192)
  d.goroutine_cnt = make(chan int, 10)
  return d
}
func (this *Demo) Goroutine() {
  this.input <- time.Now().Format("2006-01-02 15:04:05")
  time.Sleep(time.Millisecond * 500)
  <-this.goroutine_cnt
}
func (this *Demo) Handle() {
  for t := range this.input {
    fmt.Println("datatime is :", t, "goroutine count is :", len(this.goroutine_cnt))
    this.output <- t + "handle"
  }
}
func main() {
  demo := NewDemo()
  go demo.Handle()
  for i := 0; i < 10000; i++ {
    demo.goroutine_cnt <- 1
    go demo.Goroutine()
  }
  close(demo.input)
}

如上邊示例，Goroutine()函數(shù)，每隔500毫秒寫入一個時間戳到管道中，不考慮管道的讀取時間，也就是說，每個Goroutine會存在大概500毫秒時間，如果不做控制的話，一瞬間可以開啟上萬個甚至更多的goroutine出來，這樣系統(tǒng)就會奔潰。

在上述代碼中，我們引入了帶10個buffer的chan int字段，每創(chuàng)建一個goroutine時，就會向這個chan中寫入一個1，每完成一個goroutine時，就會從chan中彈出一個1。當(dāng)chan中裝滿10個1時，就會自動阻塞，等待goroutine執(zhí)行完，彈出chan中的值時，才能繼續(xù)開啟goroutine。通過chan阻塞特點，實現(xiàn)了goroutine的最大并發(fā)量控制。

以上這篇golang實現(xiàn)并發(fā)數(shù)控制的方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: