chore(api): add timeout for twitch user client

twirapp · Dec 3, 2023 · 6ae1420 · 6ae1420
1 parent 54d2756
commit 6ae1420
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/apps/api/internal/impl_unprotected/stats/stats.go b/apps/api/internal/impl_unprotected/stats/stats.go
@@ -168,7 +168,17 @@ func (c *Stats) cacheStreamers() {
 		user := user
 		streamersFollowersWg.Go(
 			func() {
-				userTwitchClient, err := twitch.NewUserClient(user.ID, c.Config, c.Grpc.Tokens)
+				userTwitchClientCtx, userTwitchClientCtxCancel := context.WithTimeout(
+					context.Background(),
+					500*time.Millisecond,
+				)
+				defer userTwitchClientCtxCancel()
+				userTwitchClient, err := twitch.NewUserClientWithContext(
+					userTwitchClientCtx,
+					user.ID,
+					c.Config,
+					c.Grpc.Tokens,
+				)
 				if err != nil {
 					c.Logger.Error("cannot create twitch client", slog.Any("err", err))
 					return