浅析GO语言中如何优雅地中断定时任务

简介: ### 问题描述 现在我们创建了一个定时器,能定时的去做某件事,并且在执行时间超时的时候,能把这个定时器关掉。例如需要收集一周的日志,创建一个定时任务去收集日志,每5秒钟执行一次,一周的时间过后需要停掉这个定时任务。

问题描述

现在我们创建了一个定时器,能定时的去做某件事,并且在执行时间超时的时候,能把这个定时器关掉。例如需要收集一周的日志,创建一个定时任务去收集日志,每5秒钟执行一次,一周的时间过后需要停掉这个定时任务。

标准库Ticker

标准库提供里的Ticker类,主要功能是定时重复的去做某件事情,如果没有设定超时,它会一直执行下去。常见的写法如下:

t := time.NewTicker(3 * time.Second)
timeout := time.After(10 * time.Second)
go func() {
        for {   
                <-t.C
                 ...
        }       
}()
<-timeout
...

注意到这个Ticker对象是无法关闭的,好的,你可能会发现Ticker类提供了Stop方法。但是我们看看如果你这样去关闭t的话,会出现什么情况。

package main

import (
        "fmt"
        "time"
)

func DoTickerWork(res chan interface{}, timeout <-chan time.Time) {
        t := time.NewTicker(3 * time.Second)
        go func() {
                defer close(res)
                i := 1
                for {
                        <-t.C
                        fmt.Printf("start %d th worker\n", i)
                        res <- i
                        i++
                }
        }()
        <-timeout
        t.Stop()
        return
}

func main() {
        res := make(chan interface{}, 10000)
        timeout := time.After(10 * time.Second)
        DoTickerWork(res, timeout)
        for v := range res {
                fmt.Println(v)
        }
}

直觉上来看,新起的goroutine在等待的过程中,主线程会把定时器关掉,似乎没有什么bug,然而输出是这样:

$go run ticker.go 
start 1 th worker
start 2 th worker
start 3 th worker
1
2
3
fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan receive]:
main.main()
    /home/gepin.zs/go/src/timer/ticker.go:29 +0xad

goroutine 6 [chan receive]:
main.DoTickerWork.func1(0xc42006c060, 0xc4200161c0)
    /home/gepin.zs/go/src/timer/ticker.go:14 +0x8e
created by main.DoTickerWork
    /home/gepin.zs/go/src/timer/ticker.go:19 +0x60
exit status 2

这说明Ticker对象的stop方法并没有关掉这个Ticker的channel,而只是阻止了channel的数据写入,所以goroutine的任务依然在进行中,但是<-t.C一直阻塞,出现了deadlock的情况。可能会有人说调用close(t.C)就可以了,但是编译会报错:cannot close receive-only channel, 因为t.C是一个只读队列,无法调用close方法。

怎么解决

不要以为stop就可以关掉Ticker了,我们可以新建一个名字为done的channel,缓存大小为1,goroutine里面采用select,然后尝试获取timeout,如果能够取到,说明已经触发超时,然后close(done),这个时候任务结束,主线程return。代码如下:

package main

import (
        "fmt"
        "time"
)

func DoTickerWork(res chan interface{}, timeout <-chan time.Time) {
        t := time.NewTicker(3 * time.Second)
        done := make(chan bool, 1)
        go func() {
                defer close(res)
                i := 1
                for {
                        select {
                        case <-t.C:
                                fmt.Printf("start %d th worker\n", i)
                                res <- i
                                i++
                        case <-timeout:
                                close(done)
                                return
                        }
                }
        }()
        <-done
        return
}

func main() {
        res := make(chan interface{}, 10000)
        timeout := time.After(10 * time.Second)
        DoTickerWork(res, timeout)
        for v := range res {
                fmt.Println(v)
        }
}

程序返回结果

$go run ticker.go
start 1 th worker
start 2 th worker
start 3 th worker
1
2
3
目录
相关文章
|
13天前
|
存储 监控 算法
员工上网行为监控中的Go语言算法:布隆过滤器的应用
在信息化高速发展的时代,企业上网行为监管至关重要。布隆过滤器作为一种高效、节省空间的概率性数据结构,适用于大规模URL查询与匹配,是实现精准上网行为管理的理想选择。本文探讨了布隆过滤器的原理及其优缺点,并展示了如何使用Go语言实现该算法,以提升企业网络管理效率和安全性。尽管存在误报等局限性,但合理配置下,布隆过滤器为企业提供了经济有效的解决方案。
54 8
员工上网行为监控中的Go语言算法:布隆过滤器的应用
|
1月前
|
存储 Go 索引
go语言中数组和切片
go语言中数组和切片
41 7
|
1月前
|
Go 开发工具
百炼-千问模型通过openai接口构建assistant 等 go语言
由于阿里百炼平台通义千问大模型没有完善的go语言兼容openapi示例,并且官方答复assistant是不兼容openapi sdk的。 实际使用中发现是能够支持的,所以自己写了一个demo test示例,给大家做一个参考。
|
1月前
|
程序员 Go
go语言中结构体(Struct)
go语言中结构体(Struct)
103 71
|
1月前
|
存储 Go 索引
go语言中的数组(Array)
go语言中的数组(Array)
106 67
|
7天前
|
算法 安全 Go
Go 语言中实现 RSA 加解密、签名验证算法
随着互联网的发展,安全需求日益增长。非对称加密算法RSA成为密码学中的重要代表。本文介绍如何使用Go语言和[forgoer/openssl](https://github.com/forgoer/openssl)库简化RSA加解密操作,包括秘钥生成、加解密及签名验证。该库还支持AES、DES等常用算法,安装简便,代码示例清晰易懂。
39 12
|
1月前
|
存储 Go
go语言中映射
go语言中映射
38 11
|
1月前
|
Go 索引
go语言修改元素
go语言修改元素
34 6
|
11天前
|
监控 算法 安全
解锁企业计算机监控的关键:基于 Go 语言的精准洞察算法
企业计算机监控在数字化浪潮下至关重要,旨在保障信息资产安全与高效运营。利用Go语言的并发编程和系统交互能力,通过进程监控、网络行为分析及应用程序使用记录等手段,实时掌握计算机运行状态。具体实现包括获取进程信息、解析网络数据包、记录应用使用时长等,确保企业信息安全合规,提升工作效率。本文转载自:[VIPShare](https://www.vipshare.com)。
20 0
|
25天前
|
Go 数据安全/隐私保护 UED
优化Go语言中的网络连接:设置代理超时参数
优化Go语言中的网络连接:设置代理超时参数