Go language timer pool

pool.go

package timers

import (
	"sync"
	"time"
)

var timerPool sync.Pool

// AcquireTimer from pool.
func AcquireTimer(d time.Duration) *PoolTimer {
	v := timerPool.Get()
	if v == nil {
		return newTimer(d)
	}
	tm := v.(*PoolTimer)
	if tm.Reset(d) {
		panic("Received an active timer from the pool!")
	}
	return tm
}

// ReleaseTimer to pool.
func ReleaseTimer(tm *PoolTimer) {
	timerPool.Put(tm)
}

// PoolTimer ...
type PoolTimer struct {
	*time.Timer
	read bool
}

// MarkRead must be called after receiving value from timer chan.
func (t *PoolTimer) MarkRead() {
	t.read = true
}

// Reset timer safely.
func (t *PoolTimer) Reset(d time.Duration) bool {
	stopped := t.Stop()
	if !stopped && !t.read {
		<-t.C
	}
	t.Timer.Reset(d)
	t.read = false
	return stopped
}

func newTimer(d time.Duration) *PoolTimer {
	return &PoolTimer{
		Timer: time.NewTimer(d),
	}
}

pool_test.go

package timers


import (
	"testing"
	"time"
)

func BenchmarkPool(b *testing.B) {
	for i := 0; i < b.N; i++ {
		tm := AcquireTimer(time.Microsecond)
		<-tm.C
		tm.MarkRead()
		ReleaseTimer(tm)
	}
	b.StopTimer()
	b.ReportAllocs()
}

func BenchmarkStd(b *testing.B) {
	for i := 0; i < b.N; i++ {
		tm := time.NewTimer(time.Microsecond)
		<-tm.C
	}
	b.StopTimer()
	b.ReportAllocs()
}