延迟队列实现(2)

  1. 定义任务结构体 首先,我们需要定义任务结构体,用于存储任务相关的信息。任务结构体至少应该包含以下几个字段:
  • ID:任务唯一标识符 *
  • Topic:任务主题,用于区分不同类型的任务*
  • Delay:任务延迟时间,即从任务加入队列到任务执行之间的时间间隔*
  • TTR(Time To Run):任务最长执行时间,即任务从开始执行到执行结束的时间间隔上限*
  • Body:任务内容,即需要执行的具体操作*
  1. 定义延迟队列 其次,我们需要定义延迟队列结构体,用于存储所有的任务。延迟队列应该包括以下几个组件:
  • 任务队列:一个优先队列,用于存储所有任务。按照任务的延迟时间进行排序,最先执行的任务排在队列前面。*
  • 延迟桶:一个时间轮,用于管理任务的延迟时间。我们可以设置多个桶,每个桶表示一个时间段,比如1秒、10秒、1分钟等等。每个桶内部维护一个任务队列,用于存储所有延迟时间在该桶时间范围内的任务。每次时间轮tick时,我们需要将当前桶的所有任务添加到任务队列中,以便执行。*
  • 定时器:一个定时器,用于定时执行任务队列中的任务。每次定时器tick时,我们从任务队列中取出最先执行的任务,并执行之。如果任务执行超时,我们可以将任务重新加入队列,并根据重试次数设置不同的延迟时间。*
  1. 添加任务 当有新任务需要添加到延迟队列中时,我们需要将任务加入到延迟桶中。首先,我们根据任务的延迟时间计算出应该加入到哪个桶中。然后,将任务加入到该桶的任务队列中。如果任务队列为空,则需要重新设置定时器的时间。

  2. 执行任务 当定时器tick时,我们从任务队列中取出最先执行的任务,并执行之。如果任务执行成功,则不需要进行任何操作;如果任务执行失败,则可以将任务重新加入队列,并根据重试次数设置不同的延迟时间。在任务执行完成后,我们需要重新设置定时器的时间。

  3. 删除任务 当某个任务需要从延迟队列中删除时,我们需要在任务队列和对应的延迟桶中删除该任务。

以上是一个完整的延迟队列系统的实现步骤。具体实现时,我们可以根据需求进行一些调整和优化,比如设置任务优先级、增加任务超时监控、支持任务重试、支持任务取消等等。

package main

import (
    "container/heap"
    "container/list"
    "fmt"
    "sync"
    "time"
)

// 任务结构体
type Task struct {
    ID      uint64    // 任务ID
    Delay   time.Time // 执行时间
    Handler func()    // 任务处理函数
}

// 实现 heap.Interface 接口,用于构建优先队列
type TaskQueue []*Task

func (t TaskQueue) Len() int {
    return len(t)
}

func (t TaskQueue) Less(i, j int) bool {
    return t[i].Delay.Before(t[j].Delay)
}

func (t TaskQueue) Swap(i, j int) {
    t[i], t[j] = t[j], t[i]
}

func (t *TaskQueue) Push(x interface{}) {
    *t = append(*t, x.(*Task))
}

func (t *TaskQueue) Pop() interface{} {
    old := *t
    n := len(old)
    x := old[n-1]
    *t = old[:n-1]
    return x
}

// 延迟队列结构体
type DelayQueue struct {
    buckets   map[time.Duration]*list.List // 延迟桶
    taskQueue TaskQueue                    // 任务队列
    mutex     sync.Mutex                   // 互斥锁
}

// 创建延迟队列
func NewDelayQueue() *DelayQueue {
    dq := &DelayQueue{
        buckets: make(map[time.Duration]*list.List),
    }
    heap.Init(&dq.taskQueue)
    go dq.loop()
    return dq
}

// 循环执行任务
func (dq *DelayQueue) loop() {
    for {
        time.Sleep(time.Second)
        dq.mutex.Lock()
        for {
            now := time.Now()
            if dq.taskQueue.Len() == 0 {
                break
            }
            if dq.taskQueue[0].Delay.After(now) {
                break
            }
            task := heap.Pop(&dq.taskQueue).(*Task)
            go task.Handler()
            dq.removeFromBucket(task)
        }
        dq.mutex.Unlock()
    }
}

// 添加任务到延迟队列中
func (dq *DelayQueue) AddTask(task *Task) {
    task.Delay = time.Now().Add(task.Delay.Sub(time.Now()))
    bucket := dq.getBucket(task.Delay.Sub(time.Now()))
    bucket.PushBack(task)
    heap.Push(&dq.taskQueue, task)
}

// 从延迟桶中移除任务
func (dq *DelayQueue) removeFromBucket(task *Task) {
    delay := task.Delay.Sub(time.Now())
    bucket := dq.getBucket(delay)
    for e := bucket.Front(); e != nil; e = e.Next() {
        if e.Value == task {
            bucket.Remove(e)
            break
        }
    }
}

// 获取延迟时间所对应的桶
func (dq *DelayQueue) getBucket(delay time.Duration) *list.List {
    for d := range dq.buckets {
        if d == delay {
            return dq.buckets[d]
        }
        if d > delay {
            break
        }
    }
    bucket := list.New()
    dq.buckets[delay] = bucket
    return bucket
}

// 测试代码
func main() {
    dq := NewDelayQueue()
    task1 := &Task{
        ID:    1,
        Delay: time.Now().Add(time.Second * 3),
        // 处理函数
        Handler: func() {
            fmt.Println("Task 1 is executed.")
        },
    }
    task2 := &Task{
        ID:    2,
        Delay: time.Now().Add(time.Second * 5),
        Handler: func() {
            fmt.Println("Task 2 is executed.")
        },
    }
    task3 := &Task{
        ID:    3,
        Delay: time.Now().Add(time.Second * 2),
        Handler: func() {
            fmt.Println("Task 3 is executed.")
        },
    }
    dq.AddTask(task1)
    dq.AddTask(task2)
    dq.AddTask(task3)
    time.Sleep(time.Second * 10)
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 223,343评论 6 521
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 95,508评论 3 400
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 170,182评论 0 366
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 60,374评论 1 300
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 69,372评论 6 398
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 52,926评论 1 314
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 41,333评论 3 426
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 40,285评论 0 277
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,813评论 1 321
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,887评论 3 343
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 41,016评论 1 354
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 36,677评论 5 351
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 42,350评论 3 336
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,843评论 0 25
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,960评论 1 275
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 49,493评论 3 379
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 46,038评论 2 361

推荐阅读更多精彩内容