simplify close condition check

mholt · Oct 30, 2024 · 1025d32 · 1025d32
1 parent c422a1b
commit 1025d32
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/layer4/listener.go b/layer4/listener.go
@@ -134,13 +134,8 @@ func (l *listener) Close() error {
 func (l *listener) loop() {
 	for {
 		conn, err := l.Listener.Accept()
-		// listener closed
-		if l.closed.Load() {
-			break
-		}
-
 		var nerr net.Error
-		if errors.As(err, &nerr) && nerr.Temporary() {
+		if errors.As(err, &nerr) && nerr.Temporary() && !l.closed.Load() {
 			l.logger.Error("temporary error accepting connection", zap.Error(err))
 			continue
 		}