Compare commits
2 commits
c713e16ed2
...
2a574d6f3b
Author | SHA1 | Date | |
---|---|---|---|
2a574d6f3b | |||
36e9aa49f7 |
2 changed files with 17 additions and 1 deletions
3
cache/cache.go
vendored
3
cache/cache.go
vendored
|
@ -69,12 +69,13 @@ func (sc *ScoreCache) Add(score float64, ip netip.Addr, ts time.Time) {
|
||||||
func (lc *ScoreCache) Get(ip netip.Addr) (ServerScore, error) {
|
func (lc *ScoreCache) Get(ip netip.Addr) (ServerScore, error) {
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
lc.mu.RLock()
|
lc.mu.RLock()
|
||||||
defer lc.mu.RUnlock()
|
|
||||||
|
|
||||||
cachedScore, ok := lc.scores[ip]
|
cachedScore, ok := lc.scores[ip]
|
||||||
if !ok {
|
if !ok {
|
||||||
|
lc.mu.RUnlock()
|
||||||
return ServerScore{}, newCacheMissError()
|
return ServerScore{}, newCacheMissError()
|
||||||
}
|
}
|
||||||
|
lc.mu.RUnlock()
|
||||||
|
|
||||||
if now.After(cachedScore.expiresAt) {
|
if now.After(cachedScore.expiresAt) {
|
||||||
lc.delete(ip)
|
lc.delete(ip)
|
||||||
|
|
15
cache/cache_test.go
vendored
15
cache/cache_test.go
vendored
|
@ -32,3 +32,18 @@ func TestCacheGet(t *testing.T) {
|
||||||
t.Fatalf("cache.Get(\"1.2.3.4\") = %f, want %f", score.Score, testFloat)
|
t.Fatalf("cache.Get(\"1.2.3.4\") = %f, want %f", score.Score, testFloat)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestCacheMiss(t *testing.T) {
|
||||||
|
now := time.Now()
|
||||||
|
const testFloat = 1.23456
|
||||||
|
|
||||||
|
expiredTime := now.Add(-(time.Minute * 10))
|
||||||
|
|
||||||
|
cache := NewLocalCache()
|
||||||
|
cache.Add(testFloat, netip.MustParseAddr("1.2.3.4"), expiredTime)
|
||||||
|
|
||||||
|
_, err := cache.Get(netip.MustParseAddr("1.2.3.4"))
|
||||||
|
if err == nil {
|
||||||
|
t.Fatalf("cache.Get(\"1.2.3.4\") = got nil, want error")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue