From 5714fbaad7f7dd1cbcc219b319acd5da9fdc0b21 Mon Sep 17 00:00:00 2001 From: Andre Medeiros Date: Wed, 2 Nov 2022 18:51:16 -0400 Subject: [PATCH] each consumer gets their reddit client --- internal/worker/notifications.go | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/internal/worker/notifications.go b/internal/worker/notifications.go index eaec291..32d2c56 100644 --- a/internal/worker/notifications.go +++ b/internal/worker/notifications.go @@ -39,7 +39,6 @@ type notificationsWorker struct { db *pgxpool.Pool redis *redis.Client queue rmq.Connection - reddit *reddit.Client apns *token.Token consumers int @@ -49,14 +48,6 @@ type notificationsWorker struct { } func NewNotificationsWorker(ctx context.Context, logger *zap.Logger, statsd *statsd.Client, db *pgxpool.Pool, redis *redis.Client, queue rmq.Connection, consumers int) Worker { - reddit := reddit.NewClient( - os.Getenv("REDDIT_CLIENT_ID"), - os.Getenv("REDDIT_CLIENT_SECRET"), - statsd, - redis, - consumers, - ) - var apns *token.Token { authKey, err := token.AuthKeyFromFile(os.Getenv("APPLE_KEY_PATH")) @@ -78,7 +69,6 @@ func NewNotificationsWorker(ctx context.Context, logger *zap.Logger, statsd *sta db, redis, queue, - reddit, apns, consumers, @@ -119,13 +109,23 @@ func (nw *notificationsWorker) Stop() { type notificationsConsumer struct { *notificationsWorker - tag int - apns *apns2.Client + reddit *reddit.Client + tag int + apns *apns2.Client } func NewNotificationsConsumer(nw *notificationsWorker, tag int) *notificationsConsumer { + reddit := reddit.NewClient( + os.Getenv("REDDIT_CLIENT_ID"), + os.Getenv("REDDIT_CLIENT_SECRET"), + nw.statsd, + nw.redis, + 1, + ) + return ¬ificationsConsumer{ nw, + reddit, tag, apns2.NewTokenClient(nw.apns).Production(), }