You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
842 B
Go
46 lines
842 B
Go
7 months ago
|
package store
|
||
|
|
||
|
import (
|
||
|
"sync"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
// AtomicTime is a time.Time with atomic operations.
|
||
|
type AtomicTime struct {
|
||
|
t time.Time
|
||
|
mu sync.RWMutex
|
||
|
}
|
||
|
|
||
|
// NewAtomicTime returns a new AtomicTime.
|
||
|
func NewAtomicTime() *AtomicTime {
|
||
|
return &AtomicTime{}
|
||
|
}
|
||
|
|
||
|
// Store stores a new time.
|
||
|
func (t *AtomicTime) Store(newTime time.Time) {
|
||
|
t.mu.Lock()
|
||
|
defer t.mu.Unlock()
|
||
|
t.t = newTime
|
||
|
}
|
||
|
|
||
|
// Load returns the stored time.
|
||
|
func (t *AtomicTime) Load() time.Time {
|
||
|
t.mu.RLock()
|
||
|
defer t.mu.RUnlock()
|
||
|
return t.t
|
||
|
}
|
||
|
|
||
|
// Add adds a duration to the stored time.
|
||
|
func (t *AtomicTime) Add(d time.Duration) {
|
||
|
t.mu.Lock()
|
||
|
defer t.mu.Unlock()
|
||
|
t.t = t.t.Add(d)
|
||
|
}
|
||
|
|
||
|
// Sub return the difference between the stored time and the given time.
|
||
|
func (t *AtomicTime) Sub(tt *AtomicTime) time.Duration {
|
||
|
t.mu.RLock()
|
||
|
defer t.mu.RUnlock()
|
||
|
return t.t.Sub(tt.t)
|
||
|
}
|