Migrate to k8s.io/utils/clock in apimachinery

This commit is contained in:
wojtekt 2021-09-15 10:05:55 +02:00
parent 859a98c035
commit adf82f050c
3 changed files with 9 additions and 9 deletions

View File

@ -21,17 +21,17 @@ import (
"sync"
"time"
utilclock "k8s.io/apimachinery/pkg/util/clock"
"k8s.io/utils/clock"
)
// NewExpiring returns an initialized expiring cache.
func NewExpiring() *Expiring {
return NewExpiringWithClock(utilclock.RealClock{})
return NewExpiringWithClock(clock.RealClock{})
}
// NewExpiringWithClock is like NewExpiring but allows passing in a custom
// clock for testing.
func NewExpiringWithClock(clock utilclock.Clock) *Expiring {
func NewExpiringWithClock(clock clock.Clock) *Expiring {
return &Expiring{
clock: clock,
cache: make(map[interface{}]entry),
@ -40,7 +40,7 @@ func NewExpiringWithClock(clock utilclock.Clock) *Expiring {
// Expiring is a map whose entries expire after a per-entry timeout.
type Expiring struct {
clock utilclock.Clock
clock clock.Clock
// mu protects the below fields
mu sync.RWMutex

View File

@ -25,7 +25,7 @@ import (
"github.com/google/uuid"
utilclock "k8s.io/apimachinery/pkg/util/clock"
testingclock "k8s.io/utils/clock/testing"
)
func TestExpiringCache(t *testing.T) {
@ -58,7 +58,7 @@ func TestExpiringCache(t *testing.T) {
}
func TestExpiration(t *testing.T) {
fc := &utilclock.FakeClock{}
fc := &testingclock.FakeClock{}
c := NewExpiringWithClock(fc)
c.Set("a", "a", time.Second)
@ -104,7 +104,7 @@ func TestExpiration(t *testing.T) {
}
func TestGarbageCollection(t *testing.T) {
fc := &utilclock.FakeClock{}
fc := &testingclock.FakeClock{}
type entry struct {
key, val string

View File

@ -21,7 +21,7 @@ import (
"time"
"github.com/google/go-cmp/cmp"
"k8s.io/apimachinery/pkg/util/clock"
testingclock "k8s.io/utils/clock/testing"
)
func expectEntry(t *testing.T, c *LRUExpireCache, key interface{}, value interface{}) {
@ -68,7 +68,7 @@ func TestSimpleRemove(t *testing.T) {
}
func TestExpiredGet(t *testing.T) {
fakeClock := clock.NewFakeClock(time.Now())
fakeClock := testingclock.NewFakeClock(time.Now())
c := NewLRUExpireCacheWithClock(10, fakeClock)
c.Add("short-lived", "12345", 1*time.Millisecond)
// ensure the entry expired