使用空结构在通道中进行信号传输以停止 for 循环

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《使用空结构在通道中进行信号传输以停止 for 循环》，聊聊，我们一起来看看吧！

我正在尝试在 go 中创建一个轮询器，它会旋转并每 24 小时执行一个函数。

我还希望能够停止轮询，我尝试通过一个完成通道并传递一个空结构来停止 for 循环来实现此目的。

在我的测试中，for 只是无限循环，我似乎无法停止它，我是否错误地使用了完成通道？股票代码大小写按预期工作。

poller struct {
    handlerfunc handlerfunc
    interval    *time.ticker
    done        chan struct{}
}

func (p *poller) start() error {
    for {
        select {
        case <-p.interval.c:
            err := p.handlerfunc()
            if err != nil {
                return err
            }
        case <-p.done:
            return nil
        }
    }
}

func (p *poller) stop() {
    p.done <- struct{}{}
}

这是执行代码并导致无限循环的测试。

poller := poller.NewPoller(
    testHandlerFunc,
    time.NewTicker(1*time.Millisecond),
)

err := poller.Start()
assert.Error(t, err)
poller.Stop()

解决方案

似乎问题出在您的用例中，您以阻塞方式调用 poller.start() ，因此永远不会调用 poller.stop() 。在 go 项目中，在 start/run 方法内部调用 goroutine 是很常见的，因此，在 poller.start() 中，我会执行类似的操作：

func (p *poller) start() <-chan error {
    errc := make(chan error, 1 )

    go func() {
        defer close(errc)

        for {
            select {
            case <-p.interval.c:
                err := p.handlerfunc()
                if err != nil {
                    errc <- err
                    return
                }
            case <-p.done:
                return
            }
        }
    }

    return errc
}

此外，无需将空的 struct 发送到完成通道。像 close(p.done) 这样的关闭通道更多的是 idiomatic for go。

go 中没有明确的方法来广播事件以执行诸如取消之类的例程。相反，它的惯用做法是创建一个通道，当关闭时表示一条消息，例如取消它必须执行的任何工作。像这样的东西是一个可行的模式：

var done = make(chan struct{})

func cancelled() bool {
    select {
    case <-done:
        return true
    default:
        return false
    }
}     

go 例程可以调用取消来轮询取消。

然后您的主循环可以响应此类事件，但请确保耗尽任何可能导致 go 例程阻塞的通道。

for {
    select {
    case <-done:
    // Drain whatever channels you need to.
        for range someChannel { }
        return
    //.. Other cases
   }
}

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~