From 548700540cb17755e14fae42413d28d462c01157 Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Wed, 7 Jun 2023 09:21:38 +0300 Subject: refactor --- internal/server/tcpserver.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'internal/server') diff --git a/internal/server/tcpserver.go b/internal/server/tcpserver.go index 30baa4a..983228b 100644 --- a/internal/server/tcpserver.go +++ b/internal/server/tcpserver.go @@ -44,9 +44,12 @@ func handleConnection(ctx context.Context, conf config.Config, conn net.Conn, ch chan<- vote.Vote) { defer conn.Close() - remoteAddr := conn.RemoteAddr().String() - reader := bufio.NewReader(conn) + var ( + remoteAddr = conn.RemoteAddr().String() + reader = bufio.NewReader(conn) + ) + for { select { case <-ctx.Done(): -- cgit v1.2.3