From a00e4405c707497938504432b89c56863e217ed4 Mon Sep 17 00:00:00 2001
From: Andre Medeiros <hello@andre.cool>
Date: Wed, 13 Jul 2022 15:02:38 -0400
Subject: [PATCH] tweak redis conns

---
 internal/cmd/api.go         | 2 +-
 internal/cmd/scheduler.go   | 2 +-
 internal/cmd/worker.go      | 2 +-
 internal/cmdutil/cmdutil.go | 4 ++--
 4 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/internal/cmd/api.go b/internal/cmd/api.go
index 344ed52..e2f9397 100644
--- a/internal/cmd/api.go
+++ b/internal/cmd/api.go
@@ -39,7 +39,7 @@ func APICmd(ctx context.Context) *cobra.Command {
 			}
 			defer db.Close()
 
-			redis, err := cmdutil.NewRedisClient(ctx)
+			redis, err := cmdutil.NewRedisClient(ctx, 16)
 			if err != nil {
 				return err
 			}
diff --git a/internal/cmd/scheduler.go b/internal/cmd/scheduler.go
index 94e888c..e610b0d 100644
--- a/internal/cmd/scheduler.go
+++ b/internal/cmd/scheduler.go
@@ -50,7 +50,7 @@ func SchedulerCmd(ctx context.Context) *cobra.Command {
 			}
 			defer db.Close()
 
-			redis, err := cmdutil.NewRedisClient(ctx)
+			redis, err := cmdutil.NewRedisClient(ctx, 2)
 			if err != nil {
 				return err
 			}
diff --git a/internal/cmd/worker.go b/internal/cmd/worker.go
index 943687f..0a93a57 100644
--- a/internal/cmd/worker.go
+++ b/internal/cmd/worker.go
@@ -54,7 +54,7 @@ func WorkerCmd(ctx context.Context) *cobra.Command {
 			}
 			defer db.Close()
 
-			redis, err := cmdutil.NewRedisClient(ctx)
+			redis, err := cmdutil.NewRedisClient(ctx, poolSize)
 			if err != nil {
 				return err
 			}
diff --git a/internal/cmdutil/cmdutil.go b/internal/cmdutil/cmdutil.go
index 2905b3a..a60aa65 100644
--- a/internal/cmdutil/cmdutil.go
+++ b/internal/cmdutil/cmdutil.go
@@ -35,12 +35,12 @@ func NewStatsdClient(tags ...string) (*statsd.Client, error) {
 	return statsd.New(os.Getenv("STATSD_URL"), statsd.WithTags(tags))
 }
 
-func NewRedisClient(ctx context.Context) (*redis.Client, error) {
+func NewRedisClient(ctx context.Context, maxConns int) (*redis.Client, error) {
 	opt, err := redis.ParseURL(os.Getenv("REDIS_URL"))
 	if err != nil {
 		return nil, err
 	}
-	opt.PoolSize = 16
+	opt.PoolSize = maxConns
 
 	client := redis.NewClient(opt)
 	if err := client.Ping(ctx).Err(); err != nil {