fix: return ErrConnClosed if write frame failed (yomorun#674)

wujunzhuo · Nov 17, 2023 · ccfcc5a · ccfcc5a
1 parent 04ecc45
commit ccfcc5a
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pkg/listener/quic/quic.go b/pkg/listener/quic/quic.go
@@ -156,10 +156,12 @@ func (p *FrameConn) WriteFrame(f frame.Frame) error {
 	default:
 		b, err := p.codec.Encode(f)
 		if err != nil {
-			return err
+			return &ErrConnClosed{err.Error()}
 		}
-
-		return p.prw.WritePacket(p.stream, f.Type(), b)
+		if err := p.prw.WritePacket(p.stream, f.Type(), b); err != nil {
+			return &ErrConnClosed{err.Error()}
+		}
+		return nil
 	}
 }