Revert "each consumer gets their reddit client"

This reverts commit 5714fbaad7.
This commit is contained in:
Andre Medeiros 2022-11-02 19:02:25 -04:00
parent 5714fbaad7
commit c6903d2b5f

View file

@ -39,6 +39,7 @@ type notificationsWorker struct {
db *pgxpool.Pool db *pgxpool.Pool
redis *redis.Client redis *redis.Client
queue rmq.Connection queue rmq.Connection
reddit *reddit.Client
apns *token.Token apns *token.Token
consumers int consumers int
@ -48,6 +49,14 @@ 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 { 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 var apns *token.Token
{ {
authKey, err := token.AuthKeyFromFile(os.Getenv("APPLE_KEY_PATH")) authKey, err := token.AuthKeyFromFile(os.Getenv("APPLE_KEY_PATH"))
@ -69,6 +78,7 @@ func NewNotificationsWorker(ctx context.Context, logger *zap.Logger, statsd *sta
db, db,
redis, redis,
queue, queue,
reddit,
apns, apns,
consumers, consumers,
@ -109,23 +119,13 @@ func (nw *notificationsWorker) Stop() {
type notificationsConsumer struct { type notificationsConsumer struct {
*notificationsWorker *notificationsWorker
reddit *reddit.Client
tag int tag int
apns *apns2.Client apns *apns2.Client
} }
func NewNotificationsConsumer(nw *notificationsWorker, tag int) *notificationsConsumer { 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 &notificationsConsumer{ return &notificationsConsumer{
nw, nw,
reddit,
tag, tag,
apns2.NewTokenClient(nw.apns).Production(), apns2.NewTokenClient(nw.apns).Production(),
} }