mirror of
https://github.com/lukaszraczylo/traefikoidc.git
synced 2026-10-14 04:07:46 +00:00
R59: honor backend TTL on cache repopulation
This commit is contained in:
@@ -0,0 +1,63 @@
|
|||||||
|
package traefikoidc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// stubBackend implements backends.CacheBackend with a controllable TTL so the
|
||||||
|
// UniversalCache.Get re-population path can be tested deterministically.
|
||||||
|
type stubBackend struct {
|
||||||
|
val []byte
|
||||||
|
ttl time.Duration
|
||||||
|
exists bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *stubBackend) Set(ctx context.Context, key string, value []byte, ttl time.Duration) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
func (s *stubBackend) Get(ctx context.Context, key string) ([]byte, time.Duration, bool, error) {
|
||||||
|
return s.val, s.ttl, s.exists, nil
|
||||||
|
}
|
||||||
|
func (s *stubBackend) Delete(ctx context.Context, key string) (bool, error) { return true, nil }
|
||||||
|
func (s *stubBackend) Exists(ctx context.Context, key string) (bool, error) { return s.exists, nil }
|
||||||
|
func (s *stubBackend) Clear(ctx context.Context) error { return nil }
|
||||||
|
func (s *stubBackend) GetStats() map[string]interface{} { return nil }
|
||||||
|
func (s *stubBackend) Close() error { return nil }
|
||||||
|
func (s *stubBackend) Ping(ctx context.Context) error { return nil }
|
||||||
|
|
||||||
|
// TestUniversalCacheBackendRepopulateKeepsRealTTL guards against Get() re-caching a
|
||||||
|
// backend value under the federated DefaultTTL instead of the entry's real
|
||||||
|
// remaining TTL. The old code re-populated the local LRU with DefaultTTL
|
||||||
|
// (often 1h), so once the distributed backend became unreachable the local
|
||||||
|
// fast-path served data whose authoritative TTL had already lapsed (e.g. a
|
||||||
|
// 5-minute token served for up to an hour).
|
||||||
|
func TestUniversalCacheBackendRepopulateKeepsRealTTL(t *testing.T) {
|
||||||
|
backend := &stubBackend{
|
||||||
|
val: []byte("42"),
|
||||||
|
ttl: 5 * time.Second,
|
||||||
|
exists: true,
|
||||||
|
}
|
||||||
|
|
||||||
|
cache := NewUniversalCacheWithBackend(UniversalCacheConfig{
|
||||||
|
Type: CacheTypeMetadata,
|
||||||
|
DefaultTTL: time.Hour,
|
||||||
|
MaxSize: 100,
|
||||||
|
Logger: NewLogger(DefaultLogLevel),
|
||||||
|
}, backend)
|
||||||
|
defer cache.Close()
|
||||||
|
|
||||||
|
if _, ok := cache.Get("key"); !ok {
|
||||||
|
t.Fatalf("backend hit should return a value")
|
||||||
|
}
|
||||||
|
|
||||||
|
item := cache.items["key"]
|
||||||
|
if item == nil {
|
||||||
|
t.Fatalf("local cache should have been populated from backend")
|
||||||
|
}
|
||||||
|
remaining := time.Until(item.ExpiresAt)
|
||||||
|
if remaining > 10*time.Second || remaining <= 0 {
|
||||||
|
t.Fatalf("local cache TTL should track backend real TTL (~5s), got %v (old code used DefaultTTL 1h)", remaining)
|
||||||
|
}
|
||||||
|
}
|
||||||
+12
-6
@@ -340,7 +340,7 @@ func (c *UniversalCache) Get(key string) (interface{}, bool) {
|
|||||||
ctx, cancel := context.WithTimeout(context.Background(), 200*time.Millisecond)
|
ctx, cancel := context.WithTimeout(context.Background(), 200*time.Millisecond)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
|
|
||||||
data, _, exists, err := c.backend.Get(ctx, c.prefixKey(key))
|
data, ttl, exists, err := c.backend.Get(ctx, c.prefixKey(key))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.logger.Debugf("Backend get error for key %s: %v", key, err)
|
c.logger.Debugf("Backend get error for key %s: %v", key, err)
|
||||||
// Fall through to local cache
|
// Fall through to local cache
|
||||||
@@ -352,11 +352,17 @@ func (c *UniversalCache) Get(key string) (interface{}, bool) {
|
|||||||
// Fall through to local cache
|
// Fall through to local cache
|
||||||
} else {
|
} else {
|
||||||
atomic.AddInt64(&c.hits, 1)
|
atomic.AddInt64(&c.hits, 1)
|
||||||
// Update local cache with backend value synchronously.
|
// Re-populate local cache with the backend entry's REAL remaining
|
||||||
// Under yaegi, goroutine spawn is 5-10x costlier than compiled Go,
|
// TTL, not the federated DefaultTTL, so the in-memory copy
|
||||||
// and this path fires per-request on cold local cache.
|
// cannot outlive the authoritative backend entry. Previously a
|
||||||
// updateLocalCache is cheap (map write under mutex).
|
// short-TTL backend value (e.g. a 5-minute token) was cached
|
||||||
_ = c.updateLocalCache(key, value, c.config.DefaultTTL)
|
// locally for up to DefaultTTL (often 1h); if the backend then
|
||||||
|
// became unreachable, getLocal kept serving it well past its
|
||||||
|
// intended expiry.
|
||||||
|
if ttl <= 0 {
|
||||||
|
ttl = c.config.DefaultTTL
|
||||||
|
}
|
||||||
|
_ = c.updateLocalCache(key, value, ttl)
|
||||||
return value, true
|
return value, true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user