Go编写定时器与定时任务详解(附第三方库gocron用法)

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()
}

结果: