TheArchive/tailscale
1
0
Fork 0
mirror of https://github.com/tailscale/tailscale.git synced 2025-04-21 22:21:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

testy: make safe for concurrent use.

Browse Source 
Signed-off-by: David Anderson <dave@natulte.net>
This commit is contained in:
David Anderson 2020-03-27 03:05:25 -07:00
parent 1fdadf06f1
commit 80261b02ba
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
testy/clock.go
View File

@ -4,7 +4,10 @@
package testy package testy
import "time" import (
"sync"
"time"
)
// Clock is a testing clock that advances every time its Now method is // Clock is a testing clock that advances every time its Now method is
// called, beginning at Start. // called, beginning at Start.
@ -18,11 +21,15 @@ type Clock struct {
Step time.Duration Step time.Duration
// Present is the time that the next Now call will receive. // Present is the time that the next Now call will receive.
Present time.Time Present time.Time
sync.Mutex
} }
// Now returns the virtual clock's current time, and avances it // Now returns the virtual clock's current time, and avances it
// according to its step configuration. // according to its step configuration.
func (c *Clock) Now() time.Time { func (c *Clock) Now() time.Time {
c.Lock()
defer c.Unlock()
if c.Start.IsZero() { if c.Start.IsZero() {
c.Start = time.Now() c.Start = time.Now()
c.Present = c.Start c.Present = c.Start
@ -35,5 +42,7 @@ func (c *Clock) Now() time.Time {
// Reset rewinds the virtual clock to its start time. // Reset rewinds the virtual clock to its start time.
func (c *Clock) Reset() { func (c *Clock) Reset() {
c.Lock()
defer c.Unlock()
c.Present = c.Start c.Present = c.Start
} }