From 556830e8e02c861d6ed5b3f6e526c7db7f74bf55 Mon Sep 17 00:00:00 2001 From: Samantha Date: Mon, 23 Sep 2024 17:41:59 -0400 Subject: [PATCH] ratelimits: Add TTLs for Redis source --- ratelimits/source_redis.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ratelimits/source_redis.go b/ratelimits/source_redis.go index 83f7046280d..cba7aab9532 100644 --- a/ratelimits/source_redis.go +++ b/ratelimits/source_redis.go @@ -89,7 +89,9 @@ func (r *RedisSource) BatchSet(ctx context.Context, buckets map[string]time.Time pipeline := r.client.Pipeline() for bucketKey, tat := range buckets { - pipeline.Set(ctx, bucketKey, tat.UTC().UnixNano(), 0) + // Set a TTL of TAT + 10 minutes to account for clock skew. + ttl := tat.UTC().Sub(r.clk.Now()) + time.Minute*10 + pipeline.Set(ctx, bucketKey, tat.UTC().UnixNano(), ttl) } _, err := pipeline.Exec(ctx) if err != nil {