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