63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
// Copyright 2020-2024 guonaihong, antlabs. All rights reserved.
|
||
//
|
||
// mit license
|
||
package timer
|
||
|
||
import (
|
||
"time"
|
||
)
|
||
|
||
type minHeapNode struct {
|
||
callback func() // 用户的callback
|
||
absExpire time.Time // 绝对时间
|
||
userExpire time.Duration // 过期时间段
|
||
root *minHeap // 指向最小堆
|
||
next Next // 自定义下个触发的时间点, cronex项目用到了
|
||
index int32 // 在min heap中的索引，方便删除或者重新推入堆中
|
||
isSchedule bool // 是否是周期性任务
|
||
}
|
||
|
||
func (m *minHeapNode) Stop() bool {
|
||
m.root.removeTimeNode(m)
|
||
return true
|
||
}
|
||
func (m *minHeapNode) Reset(d time.Duration) bool {
|
||
m.root.resetTimeNode(m, d)
|
||
return true
|
||
}
|
||
|
||
func (m *minHeapNode) Next(now time.Time) time.Time {
|
||
if m.next != nil {
|
||
return (m.next).Next(now)
|
||
}
|
||
return now.Add(m.userExpire)
|
||
}
|
||
|
||
type minHeaps []*minHeapNode
|
||
|
||
func (m minHeaps) Len() int { return len(m) }
|
||
|
||
func (m minHeaps) Less(i, j int) bool { return m[i].absExpire.Before(m[j].absExpire) }
|
||
|
||
func (m minHeaps) Swap(i, j int) {
|
||
m[i], m[j] = m[j], m[i]
|
||
m[i].index = int32(i)
|
||
m[j].index = int32(j)
|
||
}
|
||
|
||
func (m *minHeaps) Push(x any) {
|
||
// Push and Pop use pointer receivers because they modify the slice's length,
|
||
// not just its contents.
|
||
*m = append(*m, x.(*minHeapNode))
|
||
lastIndex := int32(len(*m) - 1)
|
||
(*m)[lastIndex].index = lastIndex
|
||
}
|
||
|
||
func (m *minHeaps) Pop() any {
|
||
old := *m
|
||
n := len(old)
|
||
x := old[n-1]
|
||
*m = old[0 : n-1]
|
||
return x
|
||
}
|