add missing ttl to cache interface

Southclaws · Nov 19, 2024 · 63d89fb · 63d89fb
1 parent 854c1ea
commit 63d89fb
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/internal/infrastructure/cache/cache.go b/internal/infrastructure/cache/cache.go
@@ -16,7 +16,7 @@ import (
 
 type Store interface {
 	Get(ctx context.Context, key string) (string, error)
-	Set(ctx context.Context, key string, object string) error
+	Set(ctx context.Context, key string, object string, ttl time.Duration) error
 	Delete(ctx context.Context, key string) error
 
 	HIncrBy(ctx context.Context, key string, field string, incr int64) (int, error)

diff --git a/internal/infrastructure/cache/local/local.go b/internal/infrastructure/cache/local/local.go
@@ -40,9 +40,11 @@ func (c *LocalCache) Get(ctx context.Context, key string) (string, error) {
 	return v.Value.(string), nil
 }
 
-func (c *LocalCache) Set(ctx context.Context, key string, value string) error {
+func (c *LocalCache) Set(ctx context.Context, key string, value string, ttl time.Duration) error {
+	e := time.Now().Add(ttl)
 	c.local.Store(key, Entry{
-		Value: value,
+		Value:  value,
+		Expiry: &e,
 	})
 	return nil
 }

diff --git a/internal/infrastructure/cache/redis/redis_cache.go b/internal/infrastructure/cache/redis/redis_cache.go
@@ -36,12 +36,12 @@ func (c *RedisCache) Get(ctx context.Context, key string) (string, error) {
 	return str, err
 }
 
-func (c *RedisCache) Set(ctx context.Context, key string, value string) error {
+func (c *RedisCache) Set(ctx context.Context, key string, value string, ttl time.Duration) error {
 	cmd := c.client.B().
 		Set().
 		Key(key).
 		Value(value).
-		ExSeconds(int64(c.options.Expiration.Seconds())).
+		Ex(ttl).
 		Build()
 
 	err := c.client.Do(ctx, cmd).Error()