From c153bcfa22f1806debf98755364d156a5cb310cb Mon Sep 17 00:00:00 2001 From: Miek Gieben Date: Tue, 20 Dec 2011 13:05:42 +0100 Subject: [PATCH] Make axfr example much more robust --- _examples/axfr/axfr.go | 23 +++++++++++++---------- 1 file changed, 13 insertions(+), 10 deletions(-) diff --git a/_examples/axfr/axfr.go b/_examples/axfr/axfr.go index 16d7cf1c..b469bd3d 100644 --- a/_examples/axfr/axfr.go +++ b/_examples/axfr/axfr.go @@ -21,14 +21,17 @@ func main() { } else { m.SetAxfr(zone) } - if err := client.XfrReceive(m, *nameserver); err == nil { - for r := range client.ReplyChan { - if r.Error == dns.ErrXfrLast { - break - } - fmt.Printf("%v\n", r.Reply) - } - } else { - fmt.Printf("Error %v\n", err) - } + if err := client.XfrReceive(m, *nameserver); err == nil { + for r := range client.ReplyChan { + if r.Error != nil { + if r.Error == dns.ErrXfrLast { + fmt.Printf("%v\n", r.Reply) + } + break + } + fmt.Printf("%v\n", r.Reply) + } + } else { + fmt.Printf("Error %v\n", err) + } }