Add possibility to overwrite redis client options

pace · Oct 7, 2020 · c81840c · c81840c
1 parent 419321a
commit c81840c
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/backend/redis/redis.go b/backend/redis/redis.go
@@ -84,8 +84,8 @@ func init() {
 }
 
 // Client with environment based configuration
-func Client() *redis.Client {
-	return CustomClient(&redis.Options{
+func Client(overwriteOpts ...func(*redis.Options)) *redis.Client {
+	opts := &redis.Options{
 		Addr:               cfg.Addrs[0],
 		Password:           cfg.Password,
 		DB:                 cfg.DB,
@@ -101,7 +101,13 @@ func Client() *redis.Client {
 		PoolTimeout:        cfg.PoolTimeout,
 		IdleTimeout:        cfg.IdleTimeout,
 		IdleCheckFrequency: cfg.IdleCheckFrequency,
-	})
+	}
+
+	for _, o := range overwriteOpts {
+		o(opts)
+	}
+
+	return CustomClient(opts)
 }
 
 // CustomClient with passed configuration