diff options
Diffstat (limited to 'cmd/resolverbench')
| -rw-r--r-- | cmd/resolverbench/main.go | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/cmd/resolverbench/main.go b/cmd/resolverbench/main.go index 0b517a2..a12679d 100644 --- a/cmd/resolverbench/main.go +++ b/cmd/resolverbench/main.go @@ -7,7 +7,7 @@ import ( "os" "time" - libdns "github.com/shuLhan/share/lib/dns" + "github.com/shuLhan/share/lib/dns" ) func usage() { @@ -22,30 +22,30 @@ func main() { log.SetFlags(0) - cl, err := libdns.NewUDPClient(os.Args[1]) + cl, err := dns.NewUDPClient(os.Args[1]) if err != nil { log.Fatal(err) } - msgs, err := libdns.HostsLoad(os.Args[2]) + hostsFile, err := dns.ParseHostsFile(os.Args[2]) if err != nil { log.Fatal(err) } var nfail int - fmt.Printf("= Benchmarking with %d messages\n", len(msgs)) + fmt.Printf("= Benchmarking with %d messages\n", len(hostsFile.Messages)) timeStart := time.Now() - for x := 0; x < len(msgs); x++ { - res, err := cl.Query(msgs[x]) + for x := 0; x < len(hostsFile.Messages); x++ { + res, err := cl.Query(hostsFile.Messages[x]) if err != nil { nfail++ log.Println("! Send error: ", err) continue } - exp := msgs[x].Answer[0].RData().([]byte) + exp := hostsFile.Messages[x].Answer[0].RData().([]byte) got := res.Answer[0].RData().([]byte) if !bytes.Equal(exp, got) { @@ -53,12 +53,12 @@ func main() { log.Printf(`! Answer not matched %s: expecting: %s got: %s -`, msgs[x].Question.String(), exp, got) +`, hostsFile.Messages[x].Question.String(), exp, got) } } timeEnd := time.Now() - fmt.Printf("= Total: %d\n", len(msgs)) + fmt.Printf("= Total: %d\n", len(hostsFile.Messages)) fmt.Printf("= Failed: %d\n", nfail) fmt.Printf("= Elapsed time: %v\n", timeEnd.Sub(timeStart)) } |