这是一个有趣的问题。我想到了一种解决方案,使用堆来维护待销毁物品的队列,并睡眠恰好的时间,直到下一个物品需要被销毁。我认为这样更有效率,但在某些情况下收益可能很小。无论如何,您可以在此处查看代码:
package main
import (
"container/heap"
"fmt"
"time"
)
type Item struct {
Expiration time.Time
Object interface{}
}
const MININT = 1 * time.Second
func deleteExpired(addCh chan Item) (quitCh chan bool) {
quitCh = make(chan bool)
go func() {
h := make(ExpHeap, 0)
var t *time.Timer
item := <-addCh
heap.Push(&h, &item)
t = time.NewTimer(time.Until(h[0].Expiration))
for {
for incoming := true; incoming; {
select {
case item := <-addCh:
heap.Push(&h, &item)
default:
incoming = false
}
}
if delta := time.Until(h[0].Expiration); delta >= MININT {
t.Reset(delta)
} else {
t.Reset(MININT)
}
select {
case <-quitCh:
return
case item := <-addCh:
heap.Push(&h, &item)
if item.Expiration.After(h[0].Expiration) {
continue
}
if delta := time.Until(item.Expiration); delta >= MININT {
t.Reset(delta)
} else {
t.Reset(MININT)
}
case <-t.C:
for !h[0].Expiration.After(time.Now()) {
item := heap.Pop(&h).(*Item)
destroy(item.Object)
}
if delta := time.Until(h[0].Expiration); delta >= MININT {
t.Reset(delta)
} else {
t.Reset(MININT)
}
}
}
}()
return quitCh
}
type ExpHeap []*Item
func (h ExpHeap) Len() int {
return len(h)
}
func (h ExpHeap) Swap(i, j int) {
h[i], h[j] = h[j], h[i]
}
func (h ExpHeap) Less(i, j int) bool {
return h[i].Expiration.Before(h[j].Expiration)
}
func (h *ExpHeap) Push(x interface{}) {
item := x.(*Item)
*h = append(*h, item)
}
func (h *ExpHeap) Pop() interface{} {
old, n := *h, len(*h)
item := old[n-1]
*h = old[:n-1]
return item
}
func destroy(x interface{}) {
fmt.Printf("%v @ %v\n", x, time.Now())
}
func main() {
addCh := make(chan Item)
quitCh := deleteExpired(addCh)
for i := 30; i > 0; i-- {
t := time.Now().Add(time.Duration(i) * time.Second / 2)
addCh <- Item{t, t}
}
time.Sleep(7 * time.Second)
quitCh <- true
}
游乐场: https://play.golang.org/p/JNV_6VJ_yfK
顺便提一下,有像cron
这样的作业管理包,但我不熟悉它们,所以无法对其效率发表评论。
编辑:
我仍然没有足够的声望来评论:(
关于性能:这段代码基本上具有更少的CPU使用量,因为它只在必要时唤醒自己,并且仅遍历需要销毁的项而不是整个列表。根据个人(实际上是ACM经验),大约现代CPU可以在1.2秒左右处理10^9的循环,这意味着在10^6的规模上,遍历整个列表需要超过1毫秒,不包括实际销毁代码和数据复制(平均情况下将花费大量时间超过数千次运行,达到100毫秒左右的规模)。我的代码方法是O(lg N),在10^6的规模上至少快了一千倍(考虑常数)。请再次注意,所有这些计算都是基于经验而不是基准测试的(有,但我无法提供)。
编辑2:
经过再次思考,我认为简单的解决方案可以使用一个简单的优化:
func deleteExpired(items []Item){
tail = len(items)
for index, v := range items { //better naming
if v.Expired(){
tail--
items[tail],items[index] = v,items[tail]
}
}
deleteditems := items[tail:]
items:=items[:tail]
}
通过这个改变,它不再低效地复制数据并且不会分配额外的空间。
编辑3:
从这里更改代码。
我测试了afterfunc的内存使用情况。在我的笔记本电脑上,每次调用它使用250字节,而在playground上是69字节(我很好奇原因)。使用我的代码,一个指针+一个time.Time是28字节。在一百万的规模下,差异很小。使用After Func是一个更好的选择。