基于简单时间轮的定时任务功能

metabit

已于 2024-01-09 22:51:42 修改

阅读量94

点赞数

分类专栏：数据结构与算法文章标签：时间轮

于 2023-01-14 11:47:55 首次发布

本文链接：https://blog.csdn.net/dawnto/article/details/128683589

版权

数据结构与算法专栏收录该内容

185 篇文章 5 订阅

订阅专栏

文章介绍了一个使用Go语言实现的基于简单时间轮的定时任务系统，该系统利用原子操作更新当前时间，并在每个时间槽检查是否有任务需要执行。同时提到了其他系统如Nginx和Linux内核中定时任务的数据结构，如红黑树和小顶堆。

摘要由CSDN通过智能技术生成

定时任务

定时任务底层涉及到的数据结构与算法有很多种，如
nginx:红黑树，Go:小顶堆，Linux内核：时间轮

时间轮可以分为简单时间轮和层级时间轮，如下粗糙的实现了基于简单时间轮的定时任务功能。

package main

import (
	"fmt"
	"sync/atomic"
	"time"
)

const SECONDS = 60

type task []struct {
	stop   bool
	doTime uint32
	do     func()
}

type Timer struct {
	seconds     [SECONDS]task
	currentTime uint32
}

func NewTimer() *Timer {

	t := &Timer{
		seconds:     [60]task{},
		currentTime: 0,
	}
	go func() {
		for {
			for i := 0; i < 5; i++ {
				time.Sleep(time.Millisecond * 200)
			}
			atomic.StoreUint32(&t.currentTime, atomic.AddUint32(&t.currentTime, +1))
		}
	}()

	go func() {
		for {
			index := atomic.LoadUint32(&t.currentTime) % SECONDS
			tasks := t.seconds[index]
			for idx := range tasks {
				if tasks[idx].stop == false && tasks[idx].doTime <= atomic.LoadUint32(&t.currentTime) {
					tasks[idx].stop = true
					go tasks[idx].do()
				}
			}
			
			for i := 0; i < 3; i++ {
				time.Sleep(time.Millisecond * 200)
			}
		}
	}()
	return t
}

func (t *Timer) AfterFunc(second uint32, fn func()) {
	index := atomic.LoadUint32(&t.currentTime) + second
	t.seconds[index%SECONDS] = append(t.seconds[index%SECONDS], struct {
		stop   bool
		doTime uint32
		do     func()
	}{stop: false, doTime: index, do: fn})
}

func main() {
	t := NewTimer()

	t.AfterFunc(5, func() {
		fmt.Println("hello world")
	})

	t.AfterFunc(60, func() {
		fmt.Println("hello world1")
	})

	t.AfterFunc(10, func() {
		fmt.Println("hello 10s")
	})

	time.Sleep(time.Second * 100)
}