bug fix in main receive loop (#85)

before fix goroutine did not finish after TCP connection was closed Fixes: #84
nemith · Nov 11, 2024 · 059ddfd · 059ddfd
1 parent 497e3b4
commit 059ddfd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/session.go b/session.go
@@ -238,9 +238,11 @@ func (s *Session) recvMsg() error {
 // interleaved messages (like notifications).
 func (s *Session) recv() {
 	var err error
+	var opErr *net.OpError
+
 	for {
 		err = s.recvMsg()
-		if errors.Is(err, io.EOF) || errors.Is(err, io.ErrUnexpectedEOF) {
+		if errors.Is(err, io.EOF) || errors.Is(err, io.ErrUnexpectedEOF) || errors.As(err, &opErr) {
 			break
 		}
 		if err != nil {