zeroclock.go

package main

import (
	"github.com/jonboulle/clockwork"
	"time"
)

type ZeroClock struct {
	clock clockwork.Clock
}

func (z ZeroClock) Sleep(d time.Duration) {
	if z.clock == nil {
		time.Sleep(d)
	} else {
		z.clock.Sleep(d)
	}
}

func (z ZeroClock) Now() time.Time {
	if z.clock == nil {
		return time.Now()
	}
	return z.clock.Now()
}