Go 编写定时器和定时任务

Go 编写定时器和定时任务

在 项目开发当中，可能会遇到这样的场景:

1 A 任务需要在多久之后执行一次（定时器）

2.B 任务需要每隔多长时间执行一次（定时任务）

谈到定时任务，离不开 linux 的 crontab. 先简单了解下 crontab

// 每隔 1 秒执行一次
*/1 * * * * ?

// 每隔 1 分钟执行一次
0 */1 * * * ?

// 每天 0 点执行一次
0 0 0 * * ?

// 每月 1 号凌晨 1 点执行一次
0 0 1 1 * ?

// 在 1 分、2 分、3 分执行一次
0 1,2,3 * * * ?

// 每天的 0 点、1 点、2 点执行一次
0 0 0,1,2 * * ?

使用 crontab 实现定时器和定时任务也可以，但是 crontab 默认只能精确到分钟, 如果想要精确到秒可以使用 for 循环处理秒级人物

* * * * * for i in $(seq 1 11); do echo test >> /etc/demo/demo.txt; sleep 1; done

本文简单介绍下 go 实现定时器和定时任务的用法以及简单介绍一个第三方库 gocron 的用法

Timer

1 timer 实现定时器 (延迟执行), 这个定时器只会触发一次. 使用于场景 1（当然如果把下面的代码写在一个 for 循环中也可以实现定时任务）

Demo:

package main

import (
    "fmt"
    "time"
)

func main() {t := time.NewTimer(1 * time.Second)
    <-t.C
    fmt.Println("1 s 时间到 ")

}

结果:

1 s 时间到 

2 停止定时器

使用 time.Stop() 停止定时器

​ 通过向通道发送一个信号，通知定时器是否关闭

Demo:

package main

import (
    "fmt"
    "time"
)

func test12(ch chan int) bool {timer := time.NewTimer(1 * time.Second)

    select {
    case <-ch:
        if timer.Stop() {fmt.Println(" 关闭定时器 ")
        }
        return true
    default:
        fmt.Println(" 继续执行定时器 ")
        return true
    }
}

func main() {ch := make(chan int, 1)
    // ch <- 1

    go test12(ch)

    for {}}

结果：

 继续执行定时器

重置定时器

Demo:

func main() {
    // 创建一个一个定时器
    fmt.Println(" 定时器开始 ")
    myT := time.NewTimer(7 * time.Second) // 设置 7 秒后执行一次

    // 重置定时器为 1s 后执行
    myT.Reset(1 * time.Second)
    <-myT.C
    fmt.Println(" 1 s 时间到 ", time.Now().Unix())

    for {}}

​ 结果:

 定时器开始
 1 s 时间到  1648268475

Ticker

Ticker 也是定时器，它是一个周期性的定时器 使用场景 2

设置 3 秒执行一次 Demo

package main

import (
    "fmt"
    "time"
)

func main() {ticker := time.NewTicker(3 * time.Second)
    defer ticker.Stop()

    for range ticker.C {fmt.Println(" 执行的业务逻辑 ")
    }
}

结果:

ticker ticker ticker ...
ticker ticker ticker ...
ticker ticker ticker ...

gocron

一个第三方库实现定时任务 场景 2

安装

go get -u github.com/go-co-op/gocron

s := gocron.NewScheduler(time.UTC)

s.Every(5).Seconds().Do(func(){...}) 

s.Every("5m").Do(func(){...})

s.Every(5).Days().Do(fu

s.Every(1).Month(1, 2, 3).Do(func(){...})

s.Every(1).Day().At("10:30").Do(func(){...})

s.Every(1).Day().At("10:30;08:00").Do(func(){...})

s.Every(1).Day().At("10:30").At("08:00").Do(func(){...})

s.Every(1).MonthLastDay().Do(func(){...})

s.Every(2).MonthLastDay().Do(func(){...})

s.Cron("*/1 * * * *").Do(task) 

s.StartAsync()

s.StartBlocking()

Demo

package main

import (
    "fmt"
    "time"

    "github.com/go-co-op/gocron"
)

func cron1() {fmt.Println("cron1")
}

func cron2() {fmt.Println("cron2")
}

func main() {timezone, _ := time.LoadLocation("Asia/Shanghai")
    s := gocron.NewScheduler(timezone)

    // 每秒执行一次
    s.Every(1).Seconds().Do(func() {go cron1()
    })

    // 每秒执行一次
    s.Every(1).Second().Do(func() {go cron2()
    })

    s.StartBlocking()}

结果:

cron2
cron1
cron1
cron2
cron1
cron2
cron1
cron2