timer_heap.go

// Copyright 2020 lesismal. All rights reserved.
// Use of this source code is governed by an MIT-style
// license that can be found in the LICENSE file.

package nbio

import (
	"math"
	"time"
)

const (
	timeForever = time.Duration(math.MaxInt64)
)

// Timer type for export.
type Timer struct {
	*htimer
}

// heap timer item.
type htimer struct {
	index  int
	expire time.Time
	f      func()
	parent *Gopher
}

// cancel timer.
func (it *htimer) Stop() {
	it.parent.removeTimer(it)
}

// reset timer.
func (it *htimer) Reset(timeout time.Duration) {
	it.expire = time.Now().Add(timeout)
	it.parent.resetTimer(it)
}

type timerHeap []*htimer

func (h timerHeap) Len() int           { return len(h) }
func (h timerHeap) Less(i, j int) bool { return h[i].expire.Before(h[j].expire) }
func (h timerHeap) Swap(i, j int) {
	h[i], h[j] = h[j], h[i]
	h[i].index = i
	h[j].index = j
}

func (h *timerHeap) Push(x interface{}) {
	*h = append(*h, x.(*htimer))
	n := len(*h)
	(*h)[n-1].index = n - 1
}
func (h *timerHeap) Pop() interface{} {
	old := *h
	n := len(old)
	x := old[n-1]
	old[n-1] = nil // avoid memory leak
	*h = old[0 : n-1]
	return x
}