From f490fa923aa8cb257625a8a687db05f361243c43 Mon Sep 17 00:00:00 2001
From: dyhkwong <50692134+dyhkwong@users.noreply.github.com>
Date: Mon, 18 Nov 2024 12:59:27 +0800
Subject: [PATCH] fix QUIC stream close

---
 quic/transport_quic.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/quic/transport_quic.go b/quic/transport_quic.go
index b5b765f..7bb2082 100644
--- a/quic/transport_quic.go
+++ b/quic/transport_quic.go
@@ -153,12 +153,12 @@ func (t *Transport) exchange(ctx context.Context, message *mDNS.Msg, conn quic.C
 	if err != nil {
 		return nil, err
 	}
-	defer stream.Close()
-	defer stream.CancelRead(0)
 	_, err = stream.Write(buffer.Bytes())
 	if err != nil {
+		stream.Close()
 		return nil, err
 	}
+	stream.Close()
 	buffer.Reset()
 	_, err = buffer.ReadFullFrom(stream, 2)
 	if err != nil {