simplify error type check

pion · May 2, 2024 · 9f5ddeb · 9f5ddeb
1 parent 3f0bd2a
commit 9f5ddeb
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/conn.go b/conn.go
@@ -625,10 +625,6 @@ func (c *Conn) readAndBuffer(ctx context.Context) error {
 			hasHandshake = true
 		}
 
-		var e *alertError
-		if errors.As(err, &e) && e.IsFatalOrCloseNotify() {
-			return e
-		}
 		if err != nil {
 			return err
 		}
@@ -660,10 +656,11 @@ func (c *Conn) handleQueuedPackets(ctx context.Context) error {
 			}
 		}
 		var e *alertError
-		if errors.As(err, &e) && e.IsFatalOrCloseNotify() {
-			return e
-		}
-		if err != nil {
+		if errors.As(err, &e) {
+			if e.IsFatalOrCloseNotify() {
+				return e
+			}
+		} else if err != nil {
 			return err
 		}
 	}