首页 >后端开发 >Golang >如何正确配置goroutine限制?

如何正确配置goroutine限制?

WBOY
WBOY转载
2024-02-13 15:39:08632浏览

如何正确配置goroutine限制?

php小编鱼仔将为您介绍如何正确配置goroutine限制。在Go语言中,goroutine是一种轻量级的线程,可以并发执行任务。然而,如果不进行限制,过多的goroutine可能会导致资源耗尽或程序崩溃。因此,正确配置goroutine限制是非常重要的。接下来,我们将探讨一些常见的配置方法,以确保您的程序能够稳定运行并充分利用系统资源。

问题内容

我有一个任务,如果选择三个功能来工作,则执行三个功能。我想限制每个函数的 horoutines 数量。例如,每个 goroutine 最多只能运行 10 个。

func main() {
    checkMethod1 := true
    checkMethod2 := false
    checkMethod3 := true

    list := []string{"info1", "info2", "info3", "info5"}

    for _, value := range list {
        value := value
        if checkMethod1 {
            go func() {
                //use value
                fmt.Println(value)
            }()
        }
        if checkMethod2 {
            go func() {
                //use value
                fmt.Println(value)
            }()
        }
        if checkMethod3 {
            go func() {
                //use value
                fmt.Println(value)
            }()
        }
    }
    //finish
    fmt.Println("All done")
}

我知道你可以将 goroutine 的数量限制为工作池。但是,如果我创建一个限制为 10 个 goroutine 的工作池,则该数字除以 3 个任务,并且我需要每个函数有 10 个 goroutine。

我可以创建 3 个池,但这对我来说似乎不是一个可行的方法。

我想使用这个库来创建工作池:https://github.com/sourcegraph/conc

解决方法

这里有一个方法:为每个选项使用一个缓冲通道,这样你就可以限制活动的 goroutine:

m1:=make(chan struct{},10)
m2:=make(chan struct{},10)
m3:=make(chan struct{},10)
wg:=sync.WaitGroup{}

for _, value := range list {
   value := value
   if checkMethod1 {
       m1<-struct{}{}
       wg.Add(1)
       go func() {
           defer func() { 
              <-m1 
              wg.Done()
           }()
           // do work
       }()
   }
   if checkMethod2 {
       m2<-struct{}{}
       wg.Add(1)
       go func() {
           defer func() { 
               <-m2 
               wg.Done()
           }()
           // do work
       }()
   }
   ...

  wg.Wait()
}

以上是如何正确配置goroutine限制?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文转载于:stackoverflow.com。如有侵权,请联系admin@php.cn删除