From 9fb789ee2675fec29783026e33cba07d4a4070dd Mon Sep 17 00:00:00 2001 From: Simon Ser Date: Fri, 15 Apr 2022 09:49:19 +0200 Subject: [PATCH] Add timeout for upstream connection registration --- user.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/user.go b/user.go index e2fe9aa..2b3b45e 100644 --- a/user.go +++ b/user.go @@ -191,6 +191,9 @@ func (net *network) runConn(ctx context.Context) error { net.user.srv.metrics.upstreams.Add(1) defer net.user.srv.metrics.upstreams.Add(-1) + ctx, cancel := context.WithTimeout(ctx, time.Minute) + defer cancel() + uc, err := connectToUpstream(ctx, net) if err != nil { return fmt.Errorf("failed to connect: %w", err)