mirror of
https://github.com/EchoVault/SugarDB.git
synced 2025-10-05 16:06:57 +08:00
42 lines
695 B
Go
42 lines
695 B
Go
package clock
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Clock interface {
|
|
Now() time.Time
|
|
After(d time.Duration) <-chan time.Time
|
|
}
|
|
|
|
func NewClock() Clock {
|
|
// If we're in a test environment, return the mock clock.
|
|
if strings.Contains(os.Args[0], ".test") {
|
|
return MockClock{}
|
|
}
|
|
return RealClock{}
|
|
}
|
|
|
|
type RealClock struct{}
|
|
|
|
func (RealClock) Now() time.Time {
|
|
return time.Now()
|
|
}
|
|
|
|
func (RealClock) After(d time.Duration) <-chan time.Time {
|
|
return time.After(d)
|
|
}
|
|
|
|
type MockClock struct{}
|
|
|
|
func (MockClock) Now() time.Time {
|
|
t, _ := time.Parse(time.RFC3339, "2006-01-02T15:04:05+07:00")
|
|
return t
|
|
}
|
|
|
|
func (MockClock) After(d time.Duration) <-chan time.Time {
|
|
return time.After(d)
|
|
}
|