Made clients for each protocol to reuse connections

internal/protocols/do53/do53.go

@@ -4,60 +4,62 @@ import (
 	"fmt"
 	"net"
 
-	"golang.org/x/net/dns/dnsmessage"
+	"github.com/miekg/dns"
 )
 
-func Run(domain, queryType, dest string, dnssec bool) error {
+type Do53Client struct {
+	udpConn *net.UDPConn
+}
 
-	message, err := MakeDNSMessage(domain, queryType)
-	if err != nil {
-		return err
-	}
+func New(dest string) (*Do53Client, error) {
 
 	udpAddr, err := net.ResolveUDPAddr("udp", dest)
 	if err != nil {
-		return fmt.Errorf("failed to resolve UDP address: %v", err)
+		return nil, fmt.Errorf("failed to resolve UDP address: %v", err)
 	}
 
 	udpConn, err := net.DialUDP("udp", nil, udpAddr)
 	if err != nil {
-		return fmt.Errorf("failed to dial UDP connection: %v", err)
+		return nil, fmt.Errorf("failed to dial UDP connection: %v", err)
 	}
-	defer udpConn.Close()
+	return &Do53Client{udpConn: udpConn}, nil
+}
 
-	_, err = udpConn.Write(message)
+func (c *Do53Client) Close() {
+	if c.udpConn != nil {
+		c.udpConn.Close()
+	}
+}
+
+func (c *Do53Client) Query(domain, queryType, dest string, dnssec bool) error {
+
+	message, err := NewDNSMessage(domain, queryType)
+	if err != nil {
+		return err
+	}
+
+	_, err = c.udpConn.Write(message)
 	if err != nil {
 		return fmt.Errorf("failed to send DNS query: %v", err)
 	}
 
 	buf := make([]byte, 4096)
-	n, err := udpConn.Read(buf)
+	n, err := c.udpConn.Read(buf)
 	if err != nil {
 		return fmt.Errorf("failed to read DNS response: %v", err)
 	}
 
-	var parser dnsmessage.Parser
-	_, err = parser.Start(buf[:n])
+	recvMsg := new(dns.Msg)
+	err = recvMsg.Unpack(buf[:n])
 	if err != nil {
 		return fmt.Errorf("failed to parse DNS response: %v", err)
 	}
 
 	// TODO: Check if the response had no errors or TD bit set
 
-	err = parser.SkipAllQuestions()
-	if err != nil {
-		return fmt.Errorf("failed to skip questions: %v", err)
-	}
-
-	answers, err := parser.AllAnswers()
-	if err != nil {
-		return err
-	}
-
-	for _, answer := range answers {
-		fmt.Println(answer.GoString())
+	for _, answer := range recvMsg.Answer {
+		fmt.Println(answer.String())
 	}
 
 	return nil
 }
-