Golang 工作池实现意外工作

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《Golang 工作池实现意外工作》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

我已经实现了一个 golang 工作池，如下所示，其中 sem 和 work 是通道。 sem 是一个跟踪当前活动的工人（goroutines）数量的渠道。 work 是将功能传递给活跃工作人员执行的通道。超时将使任何工作线程在超时时间内处于空闲状态。

package main

import (
    "time"
)

type pool struct {
    sem chan struct{}
    work chan func()
    timeout time.duration
}

func newpool(max, size, spawn int, timeout time.duration) *pool {
    if spawn <= 0 {
        panic("workpool spawn is <= 0")
    }
    if spawn > max {
        panic("workpool spawn > max workers")
    }
    p := &pool{
        sem: make(chan struct{}, max),
        work: make(chan func(), size),
        timeout: timeout,
    }
    for i := 0; i < spawn; i++ {
        p.sem <- struct{}{}
        go p.worker(func() {})
    }
    return p

}

func (p *pool) addtask(task func()) {
    select {
        case p.work <- task:
            return
        case p.sem <- struct{}{}:
            go p.worker(task)
            return
    }
}

func (p *pool) worker(task func()) {
    t := time.newtimer(p.timeout)
    defer func() {
        t.stop()
        <- p.sem
    }()

    task()

    for {
        select {
            case task := <- p.work:
                t.reset(p.timeout)
                task()
            case <- t.c:
                return
        }
    }
}

我正在通过在 for 循环中打印 i 的值来进行测试，方法是将其传递到封装在匿名函数中的池中，如下所示：

package main

import (
    "fmt"
    "time"
)

func main() {
    fmt.println("hello, world!")

    p := newpool(3, 10, 1, time.duration(5) * time.second)

    for i:=0; i<30; i++ {
        p.addtask(func () {
            fmt.print(i, " ")
        })
    }
    time.sleep(10 * time.second)
    fmt.println("end")

}

预期输出应该是从 0 到 29 的序列号，但实际输出是

Hello, world!
12 12 12 12 12 12 12 12 12 12 12 12 13 25 25 25 25 25 25 25 25 25 25 25 26 25 30 30 30 30 End

我不明白为什么输出是这样的。

正确答案

您的函数闭包都引用相同的 i 值。这会产生竞争条件，当函数被调度时，它们正在读取不断变化的值 - 因此您会看到不可预测的输出。

为了确保闭包获得唯一值，请在循环内声明变量。一个简单的技巧是通过影子声明相同的变量名 i := i

for i:=0; i<30; i++ {
    i:= i                // <- add this
    p.addtask(func () {
        fmt.print(i, " ")
    })
}

https://go.dev/play/p/o0Nyx5A46tp

顺便说一句，Effective Go 文档中介绍了此技术，请参阅本节：

写起来可能看起来很奇怪

req := req

但在 go 中这样做是合法且惯用的。您将获得全新版本 具有相同名称的变量，故意隐藏循环 本地变量，但每个 goroutine 都是唯一的。

终于介绍完啦！小伙伴们，这篇关于《Golang 工作池实现意外工作》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！