如何在执行工作人员池之间正确延迟

最近发现不少小伙伴都对Golang很感兴趣，所以今天继续给大家介绍Golang相关的知识，本文《如何在执行工作人员池之间正确延迟》主要内容涉及到等等知识点，希望能帮到你！当然如果阅读本文时存在不同想法，可以在评论中表达，但是请勿使用过激的措辞~

美好的一天，

我正在尝试在工作人员执行之间实现正确的延迟，例如，工作人员需要完成 30 个任务并进入睡眠 5 秒，我如何在代码中准确跟踪 30 个任务已完成，然后才进入睡眠状态5 秒？

下面的代码创建了一个30个工人的池，这些工人依次以无序的方式一次执行30个任务，代码如下：

import (
    "fmt"
    "math/rand"
    "sync"
    "time"
)

type Job struct {
    id       int
    randomno int
}
type Result struct {
    job         Job
    sumofdigits int
}

var jobs = make(chan Job, 10)
var results = make(chan Result, 10)

func digits(number int) int {
    sum := 0
    no := number
    for no != 0 {
        digit := no % 10
        sum += digit
        no /= 10
    }
    time.Sleep(2 * time.Second)
    return sum
}
func worker(wg *sync.WaitGroup) {
    for job := range jobs {
        output := Result{job, digits(job.randomno)}
        results <- output
    }
    wg.Done()
}
func createWorkerPool(noOfWorkers int) {
    var wg sync.WaitGroup
    for i := 0; i < noOfWorkers; i++ {
        wg.Add(1)
        go worker(&wg)
    }

    wg.Wait()
    close(results)
}
func allocate(noOfJobs int) {
    for i := 0; i < noOfJobs; i++ {
        if i != 0 && i%30 == 0 {
            fmt.Printf("SLEEPAGE 5 sec...")
            time.Sleep(10 * time.Second)

        }
        randomno := rand.Intn(999)
        job := Job{i, randomno}
        jobs <- job
    }
    close(jobs)
}
func result(done chan bool) {
    for result := range results {
        fmt.Printf("Job id %d, input random no %d , sum of digits %d\n", result.job.id, result.job.randomno, result.sumofdigits)
    }
    done <- true
}
func main() {
    startTime := time.Now()
    noOfJobs := 100
    go allocate(noOfJobs)
    done := make(chan bool)
    go result(done)
    noOfWorkers := 30
    createWorkerPool(noOfWorkers)
    <-done
    endTime := time.Now()
    diff := endTime.Sub(startTime)
    fmt.Println("total time taken ", diff.Seconds(), "seconds")
}

播放：https://go.dev/play/p/lehl7hoo-kp

目前尚不清楚如何确保完成 30 个任务以及在哪里插入延迟，我将不胜感激任何帮助

正确答案

好的，让我们从这个工作示例开始：

func Test_t(t *testing.T) {

    // just a published, this publishes result on a chan
    publish := func(s int, ch chan int, wg *sync.WaitGroup) {
        ch <- s // this is blocking!!!
        wg.Done()
    }

    wg := &sync.WaitGroup{}
    wg.Add(100)

    // we'll use done channel to notify the work is done
    res := make(chan int)
    done := make(chan struct{})
    // create worker that will notify that all results were published
    go func() {
        wg.Wait()
        done <- struct{}{}
    }()
    
    // let's create a jobs that publish on our res chan
    // please note all goroutines are created immediately
    for i := 0; i < 100; i++ {
        go publish(i, res, wg)
    }

    // lets get 30 args and then wait
    var resCounter int
forloop:
    for {
        select {
        case ss := <-res:
            println(ss)
            resCounter += 1
            // break the loop
            if resCounter%30 == 0 {
                // after receiving 30 results we are blocking this thread
                // no more results will be taken from the channel for 5 seconds
                println("received 30 results, waiting...")
                time.Sleep(5 * time.Second)
            }
        case <-done:
            // we are done here, let's break this infinite loop
            break forloop
        }
    }
}

我希望这进一步展示了如何做到这一点。

那么，你的代码有什么问题吗？ 说实话，看起来不错（我的意思是发布了 30 个结果，然后代码等待，然后是另外 30 个结果，等等），但问题是您想在哪里等待？

我猜测有几种可能性：

• 创建工作线程（这就是你的代码现在的工作方式，正如我所见，它以 30 个包的形式发布作业；请注意，digit 函数中的 2 秒延迟仅适用于 goroutine代码被执行）

• 触发工作程序（因此“等待”代码应该位于工作程序函数中，不允许运行更多工作程序 - 因此它必须监视发布了多少结果）

• 处理结果（这就是我的代码的工作方式，正确的同步位于 forloop 中）

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《如何在执行工作人员池之间正确延迟》文章吧，也可关注golang学习网公众号了解相关技术文章。