implement querier and add comments to satisfy linters

Signed-off-by: Bruno Schaatsbergen <[email protected]>

Author: bschaatsbergen

cmd/root.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"os"
 
+	"github.com/hashicorp/go-multierror"
 	"github.com/spf13/cobra"
 	"github.com/znscli/zns/internal/query"
 )
@@ -23,6 +24,30 @@ var (
 				fmt.Println("See 'zns -h' for help and examples")
 				os.Exit(1)
 			}
+
+			// Initialize a new querier with the specified DNS server
+			querier := query.NewQuerier("1.1.1.1:53", nil)
+
+			qtypes := make([]uint16, 0, len(query.QueryTypes))
+			for _, qtype := range query.QueryTypes {
+				qtypes = append(qtypes, qtype)
+			}
+
+			messages, err := querier.QueryTypes(args[0], qtypes)
+			if err != nil {
+				if merr, ok := err.(*multierror.Error); ok {
+					for _, e := range merr.Errors {
+						fmt.Println(e)
+					}
+				} else { // This should never happen since all errors in zns are multierrors, but we handle it anyway.
+					fmt.Println(err)
+				}
+				os.Exit(1)
+			}
+
+			for _, msg := range messages {
+				fmt.Println(msg)
+			}
 		},
 	}
 )
@@ -38,4 +63,3 @@ func Execute() {
 		os.Exit(1)
 	}
 }
-

internal/query/query.go

@@ -22,18 +22,19 @@ var (
 	}
 )
 
-// Querier is an interface for querying DNS records
+// Querier is an interface for querying DNS records.
 type Querier interface {
 	Query(domain string, qtype uint16) (*dns.Msg, error)
+	QueryTypes(domain string, qtypes []uint16) ([]*dns.Msg, error)
 }
 
-// Query implements the Querier interface
+// Query implements the Querier interface.
 type Query struct {
 	Server string
 	hclog.Logger
 }
 
-// NewQuerier creates a new Querier with the specified server
+// NewQuerier creates a new Querier with the specified server.
 func NewQuerier(server string, logger hclog.Logger) Querier {
 	if logger == nil {
 		logger = hclog.New(&hclog.LoggerOptions{
@@ -47,7 +48,7 @@ func NewQuerier(server string, logger hclog.Logger) Querier {
 	return &Query{Server: server, Logger: logger}
 }
 
-// QueryTypes performs DNS queries for multiple types concurrently
+// QueryTypes performs DNS queries for multiple types concurrently.
 func (q *Query) QueryTypes(domain string, qtypes []uint16) ([]*dns.Msg, error) {
 	var errors *multierror.Error
 	var wg sync.WaitGroup
@@ -68,7 +69,7 @@ func (q *Query) QueryTypes(domain string, qtypes []uint16) ([]*dns.Msg, error) {
 	return messages, errors.ErrorOrNil()
 }
 
-// Query performs the DNS query and returns the response and any error encountered
+// Query performs the DNS query and returns the response and any error encountered.
 func (q *Query) Query(domain string, qtype uint16) (*dns.Msg, error) {
 	msg := new(dns.Msg)
 	msg.SetQuestion(dns.Fqdn(domain), qtype)