clientproxy: reorder imports

cmd/clientproxy/main.go

@@ -2,11 +2,12 @@ package main
 
 import (
 	"flag"
-	"github.com/miekg/dns"
-	log "github.com/sirupsen/logrus"
 	"net"
 	"os"
 
+	"github.com/miekg/dns"
+	log "github.com/sirupsen/logrus"
+
 	"github.com/natesales/doq/pkg/client"
 )
 
@@ -25,14 +26,6 @@ func main() {
 		log.SetLevel(log.DebugLevel)
 	}
 
-	// Create a new DoQ client
-	doqClient, err := client.New(*doqServer, *tlsInsecureSkipVerify, *tlsCompat)
-	if err != nil {
-		log.Fatal(doqClient)
-		os.Exit(1)
-	}
-	defer doqClient.Close()
-
 	// Create the UDP DNS listener
 	log.Debug("creating UDP listener")
 	pc, err := net.ListenPacket("udp", *listenAddr)
@@ -58,13 +51,23 @@ func main() {
 			log.Warn(err)
 		}
 
+		// Create a new DoQ client
+		doqClient, err := client.New(*doqServer, *tlsInsecureSkipVerify, *tlsCompat)
+		if err != nil {
+			log.Fatal(doqClient)
+			os.Exit(1)
+		}
+
 		log.Debugln("sending DoQ query")
 		resp, err := doqClient.SendQuery(msgIn)
 		if err != nil {
 			log.Fatal(err)
 			os.Exit(1)
 		}
 
+		log.Debugln("closing doq quic stream")
+		doqClient.Close()
+
 		packed, err := resp.Pack()
 		pc.WriteTo(packed, addr)
 		log.Debug("finished writing")