浅析GO语言中如何优雅地中断定时任务-阿里云开发者社区

开发者社区> 葛平> 正文

浅析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

版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。

相关文章
阿里云服务器怎么设置密码?怎么停机?怎么重启服务器?
如果在创建实例时没有设置密码,或者密码丢失,您可以在控制台上重新设置实例的登录密码。本文仅描述如何在 ECS 管理控制台上修改实例登录密码。
10076 0
职播间 | 自然语言处理中的多任务学习 & 复旦大学NLP实验室介绍
在本次公开课中,讲者将分享其所在研究组关于多任务学习在自然语言处理领域的最新工作。
304 0
手把手教你创建定时任务 | 带你学《Java语言高级特性》之三十三
在开发过程中经常会遇到自定义时钟的需求,这时我们可以创建一个定时任务。本节将为你讲解如何使用Timer类与TimerTask类实现定时任务的管理工作。
537 0
ACL 2019 | 基于知识增强的语言表示模型,多项NLP任务表现超越BERT(附论文解读)
本文提出了一种新方法,将知识图谱的信息加入到模型的训练中。
746 0
阿里云服务器如何登录?阿里云服务器的三种登录方法
购买阿里云ECS云服务器后如何登录?场景不同,阿里云优惠总结大概有三种登录方式: 登录到ECS云服务器控制台 在ECS云服务器控制台用户可以更改密码、更换系.
13883 0
阿里云ECS云服务器初始化设置教程方法
阿里云ECS云服务器初始化是指将云服务器系统恢复到最初状态的过程,阿里云的服务器初始化是通过更换系统盘来实现的,是免费的,阿里云百科网分享服务器初始化教程: 服务器初始化教程方法 本文的服务器初始化是指将ECS云服务器系统恢复到最初状态,服务器中的数据也会被清空,所以初始化之前一定要先备份好。
11888 0
阿里云ECS云服务器初始化设置教程方法
阿里云ECS云服务器初始化是指将云服务器系统恢复到最初状态的过程,阿里云的服务器初始化是通过更换系统盘来实现的,是免费的,阿里云百科网分享服务器初始化教程: 服务器初始化教程方法 本文的服务器初始化是指将ECS云服务器系统恢复到最初状态,服务器中的数据也会被清空,所以初始化之前一定要先备份好。
7365 0
+关注
葛平
阿里云研发工程师
5
文章
4
问答
文章排行榜
最热
最新
相关电子书
更多
《2021云上架构与运维峰会演讲合集》
立即下载
《零基础CSS入门教程》
立即下载
《零基础HTML入门教程》
立即下载