如何保持工作函数的持续性？

一分耕耘，一分收获！既然打开了这篇文章《如何保持工作函数的持续性？》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

假设我有一个包含 10 个工作函数的列表，我希望其中 2 个（或更多）始终并行运行，当其中一个完成时，在列表中前进，然后循环并永远继续。所以不要让服务器超载。

workers := make([]func(), 10)
for i := 0; i < 10; i++ {
  workers[i] = createWorker()
}

func createWorker() func() {
  return func() {
    fmt.Println("I am working")
    time.Sleep(time.Duration(rand.Intn(5)) * time.Second)
  }
}

// My idea, keep sending workers to a buffered channel of size 2, 
// so when one finishes it's no longer filled up and another worker is sent
workerChan := make(chan func(), 2)
go func() {
  worker := <-workerChan
  worker()
}()

for {
  for _, worker := range workers {
    workerChan <- worker
  }
}

这将运行第一个工作函数，仅此而已。也许这个想法是正确的，我需要一些关于如何正确实现它的指导。

解决方案

你已经到达那里了。

func worker(ch chan func()) {
    // worker needs to read from a channel until channel is closed, 
    // then it will stop
    for work := range ch {
        work()
    }
}

func main() {
    workers := 2
    workerChan := make(chan func(), 10)

    for i := 0; i < workers; i++ {
        // start workers
        worker(workerChan)
    }
    
    // add work to channel
    workerChan <- func() {
        // do work
    }
}

如果你将它包装在一个结构中，你可以用它创建一个非常通用的工作池，执行你给它的任何内容。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。