astra/owpengram-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

perf: sync protocol and core hardening updates

Browse source
This commit is contained in:
A 2026-07-11 19:48:26 +08:00
parent 152fed3b87
commit 4390ebf5a9
283 changed files with 29231 additions and 2295 deletions
Download patch file Download diff file Expand all files Collapse all files

17
internal/store/redisstore/allocator_integration_test.go
View file

@ -158,4 +158,21 @@ func TestRedisRateLimiterWindow(t *testing.T) {
if allowed || retry <= 0 {
t.Fatalf("AllowN second allowed=%v retry=%d, want limited with retry", allowed, retry)
}
// Heal a counter left without TTL by an older split INCRBY→EXPIRE writer.
// Without this branch one transient crash could permanently deny login-code
// issuance for the affected phone/auth-key limiter dimension.
orphanKey := key + ":orphan-no-ttl"
redisOrphanKey := rateLimitKey(orphanKey)
t.Cleanup(func() { _ = c.Del(ctx, redisOrphanKey).Err() })
if err := c.Set(ctx, redisOrphanKey, 100, 0).Err(); err != nil {
t.Fatalf("seed no-TTL counter: %v", err)
}
allowed, retry, err = limiter.Allow(ctx, orphanKey, 1, 5*time.Second)
if err != nil || allowed || retry <= 0 || retry > 5 {
t.Fatalf("heal no-TTL counter allowed=%v retry=%d err=%v", allowed, retry, err)
}
if ttl, err := c.PTTL(ctx, redisOrphanKey).Result(); err != nil || ttl <= 0 || ttl > 5*time.Second {
t.Fatalf("healed counter TTL=%v err=%v, want (0,5s]", ttl, err)
}
}