From 96e11c01a72f6006e08d8921d79931f40d0f921c Mon Sep 17 00:00:00 2001
From: AUTUMN <lyekumchew@gmail.com>
Date: Tue, 14 Jun 2022 00:03:36 +0800
Subject: [PATCH] Update doq.go

---
 pkg/resolvers/doq.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pkg/resolvers/doq.go b/pkg/resolvers/doq.go
index 7b0d0e2..99d9cf3 100644
--- a/pkg/resolvers/doq.go
+++ b/pkg/resolvers/doq.go
@@ -76,6 +76,7 @@ func (r *DOQResolver) Lookup(question dns.Question) (Response, error) {
 		if err != nil {
 			return rsp, err
 		}
+		// Make a QUIC request to the DNS server with the DNS message as wire format bytes in the body.
 		_, err = stream.Write(b)
 		if err != nil {
 			return rsp, err
@@ -96,6 +97,8 @@ func (r *DOQResolver) Lookup(question dns.Question) (Response, error) {
 		}
 		rtt := time.Since(now)
 
+		_ = stream.Close()
+
 		packetLen := binary.BigEndian.Uint16(buf[:2])
 		if packetLen != uint16(len(buf[2:])) {
 			return rsp, fmt.Errorf("packet length mismatch")
@@ -122,8 +125,6 @@ func (r *DOQResolver) Lookup(question dns.Question) (Response, error) {
 			// stop iterating the searchlist.
 			break
 		}
-
-		_ = stream.Close()
 	}
 	return rsp, nil
 }