luzifer/rootzone
1
0
Fork 0
mirror of https://github.com/Luzifer/rootzone.git synced 2024-12-20 19:31:20 +00:00
Code Issues Releases Wiki Activity

Add retries to all DNS calls

Browse source 
Signed-off-by: Knut Ahlers <knut@ahlers.me>
This commit is contained in:
Knut Ahlers 2019-06-22 20:03:17 +02:00
parent 7896e20f7f
commit 213e724e1b
Signed by: luzifer
GPG key ID: DC2729FDD34BE99E
3 changed files with 61 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
iana.go
View file

@ -42,9 +42,16 @@ func getIANAZoneMasters(tld string) ([]string, error) {
m := new(dns.Msg) m := new(dns.Msg)
m.SetQuestion(tld, dns.TypeNS) m.SetQuestion(tld, dns.TypeNS)
r, _, err := c.Exchange(m, getRandomInternicRoot()) var (
if err != nil { err error
return nil, errors.Wrap(err, "Could not query nameservers") r *dns.Msg
)
if err = retry(func() error {
r, _, err = c.Exchange(m, getRandomInternicRoot())
return errors.Wrap(err, "Could not query nameservers")
}); err != nil {
return nil, err
} }
if r.Rcode != dns.RcodeSuccess { if r.Rcode != dns.RcodeSuccess {

34
opennic.go
View file

@ -13,9 +13,16 @@ func getOpenNICTLDs() ([]string, error) {
m := new(dns.Msg) m := new(dns.Msg)
m.SetQuestion("tlds.opennic.glue.", dns.TypeTXT) m.SetQuestion("tlds.opennic.glue.", dns.TypeTXT)
r, _, err := c.Exchange(m, cfg.OpenNICRoot+":53") var (
if err != nil { err error
return nil, errors.Wrap(err, "Could not query nameservers") r *dns.Msg
)
if err = retry(func() error {
r, _, err = c.Exchange(m, cfg.OpenNICRoot+":53")
return errors.Wrap(err, "Could not query nameservers")
}); err != nil {
return nil, err
} }
if r.Rcode != dns.RcodeSuccess { if r.Rcode != dns.RcodeSuccess {
@ -42,9 +49,16 @@ func getOpenNICZoneMasters(tld string) ([]string, error) {
m := new(dns.Msg) m := new(dns.Msg)
m.SetQuestion(tld+"opennic.glue.", dns.TypeCNAME) m.SetQuestion(tld+"opennic.glue.", dns.TypeCNAME)
r, _, err := c.Exchange(m, cfg.OpenNICRoot+":53") var (
if err != nil { err error
return nil, errors.Wrap(err, "Could not query nameservers") r *dns.Msg
)
if err = retry(func() error {
r, _, err = c.Exchange(m, cfg.OpenNICRoot+":53")
return errors.Wrap(err, "Could not query nameservers")
}); err != nil {
return nil, err
} }
if r.Rcode != dns.RcodeSuccess { if r.Rcode != dns.RcodeSuccess {
@ -65,9 +79,11 @@ func getOpenNICZoneMasters(tld string) ([]string, error) {
m = new(dns.Msg) m = new(dns.Msg)
m.SetQuestion(master, dns.TypeA) m.SetQuestion(master, dns.TypeA)
r, _, err := c.Exchange(m, cfg.OpenNICRoot+":53") if err = retry(func() error {
if err != nil { r, _, err = c.Exchange(m, cfg.OpenNICRoot+":53")
return nil, errors.Wrap(err, "Could not query nameservers") return errors.Wrap(err, "Could not query nameservers")
}); err != nil {
return nil, err
} }
if r.Rcode != dns.RcodeSuccess { if r.Rcode != dns.RcodeSuccess {

26
retry.go Normal file
View file

@ -0,0 +1,26 @@
package main
import (
"math"
"time"
)
const (
maxRetries = 5
retryDelay = 1.2
)
func retry(f func() error) error {
var err error
for i := 1; i <= maxRetries; i++ {
if err = f(); err == nil {
return nil
}
sleep := time.Duration(math.Pow(retryDelay, float64(i)) * float64(time.Second))
time.Sleep(sleep)
}
return err
}