-
Notifications
You must be signed in to change notification settings - Fork 672
/
clock.go
40 lines (32 loc) · 948 Bytes
/
clock.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package timer
import (
"time"
)
// MaxTime was taken from https://stackoverflow.com/questions/25065055/what-is-the-maximum-time-time-in-go/32620397#32620397
var MaxTime = time.Unix(1<<63-62135596801, 0) // 0 is used because we drop the nano-seconds
// Clock acts as a thin wrapper around global time that allows for easy testing
type Clock struct {
faked bool
time time.Time
}
// Set the time on the clock
func (c *Clock) Set(time time.Time) { c.faked = true; c.time = time }
// Sync this clock with global time
func (c *Clock) Sync() { c.faked = false }
// Time returns the time on this clock
func (c *Clock) Time() time.Time {
if c.faked {
return c.time
}
return time.Now()
}
// Unix returns the unix time on this clock.
func (c *Clock) Unix() uint64 {
unix := c.Time().Unix()
if unix < 0 {
unix = 0
}
return uint64(unix)
}