ratelimits: Add TTLs for Redis source

letsencrypt · Sep 23, 2024 · 556830e · 556830e
1 parent 990ad07
commit 556830e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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 {