Retry five times getcurrentip before error

This commit is contained in:
Laurent Le Houerou 2019-07-03 12:46:36 +04:00
parent 32ac90376f
commit 64d066ba4b
3 changed files with 40 additions and 19 deletions

1
go.mod
View File

@ -3,6 +3,7 @@ module gogs.lehouerou.net/Laurent/dnsupdater
go 1.12 go 1.12
require ( require (
github.com/matryer/try v0.0.0-20161228173917-9ac251b645a2
github.com/sirupsen/logrus v1.4.2 github.com/sirupsen/logrus v1.4.2
github.com/urfave/cli v1.20.0 github.com/urfave/cli v1.20.0
gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df

2
go.sum
View File

@ -2,6 +2,8 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/konsorten/go-windows-terminal-sequences v1.0.1 h1:mweAR1A6xJ3oS2pRaGiHgQ4OO8tzTaLawm8vnODuwDk= github.com/konsorten/go-windows-terminal-sequences v1.0.1 h1:mweAR1A6xJ3oS2pRaGiHgQ4OO8tzTaLawm8vnODuwDk=
github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
github.com/matryer/try v0.0.0-20161228173917-9ac251b645a2 h1:JAEbJn3j/FrhdWA9jW8B5ajsLIjeuEHLi8xE4fk997o=
github.com/matryer/try v0.0.0-20161228173917-9ac251b645a2/go.mod h1:0KeJpeMD6o+O4hW7qJOT7vyQPKrWmj26uf5wMc/IiIs=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/sirupsen/logrus v1.4.2 h1:SPIRibHv4MatM3XXNO2BJeFLZwZ2LvZgfQ5+UNI2im4= github.com/sirupsen/logrus v1.4.2 h1:SPIRibHv4MatM3XXNO2BJeFLZwZ2LvZgfQ5+UNI2im4=

24
main.go
View File

@ -4,6 +4,7 @@ import (
"bytes" "bytes"
"encoding/json" "encoding/json"
"fmt" "fmt"
"github.com/matryer/try"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"github.com/urfave/cli" "github.com/urfave/cli"
"gopkg.in/gomail.v2" "gopkg.in/gomail.v2"
@ -12,6 +13,7 @@ import (
"os" "os"
"path" "path"
"regexp" "regexp"
"time"
) )
type Configuration struct { type Configuration struct {
@ -115,7 +117,7 @@ func update() error {
continue continue
} }
log.Infoln("%s -> Ip has changed %s -> %s\n", domain, currentIp, newIp) log.Infoln(fmt.Sprintf("%s -> Ip has changed %s -> %s", domain, currentIp, newIp))
err = SetCurrentIp(newIp, domain, *configuration) err = SetCurrentIp(newIp, domain, *configuration)
if err != nil { if err != nil {
@ -144,7 +146,8 @@ func SendStatusMail(messageText string) {
d := gomail.NewDialer(settings.SmtpHost, settings.SmtpPort, settings.SmtpLogin, settings.SmtpPassword) d := gomail.NewDialer(settings.SmtpHost, settings.SmtpPort, settings.SmtpLogin, settings.SmtpPassword)
s, err := d.Dial() s, err := d.Dial()
if err != nil { if err != nil {
log.Errorln("error while dialing smtp server : %v", err) log.Errorln(fmt.Sprintf("error while dialing smtp server : %v", err))
return
} }
m := gomail.NewMessage() m := gomail.NewMessage()
@ -155,13 +158,17 @@ func SendStatusMail(messageText string) {
m.SetBody("text/plain", messageText) m.SetBody("text/plain", messageText)
if err := gomail.Send(s, m); err != nil { if err := gomail.Send(s, m); err != nil {
log.Warnln("could not send email to %q: %v", r, err) log.Warnln(fmt.Sprintf("could not send email to %q: %v", r, err))
} }
m.Reset() m.Reset()
} }
} }
func GetCurrentIp(configuration Configuration, domain string) (string, error) { func GetCurrentIp(configuration Configuration, domain string) (string, error) {
var value string
err := try.Do(func(attempt int) (bool, error) {
var err error
value, err = func(configuration Configuration, domain string) (string, error) {
url := fmt.Sprintf(GandiARecordUrl, domain) url := fmt.Sprintf(GandiARecordUrl, domain)
req, err := http.NewRequest("GET", url, nil) req, err := http.NewRequest("GET", url, nil)
if err != nil { if err != nil {
@ -182,6 +189,17 @@ func GetCurrentIp(configuration Configuration, domain string) (string, error) {
return "", err return "", err
} }
return dnsresponse.Values[0], nil return dnsresponse.Values[0], nil
}(configuration, domain)
if err != nil {
log.Warnln(fmt.Sprintf("Error while getting current ip from gandi : %v", err))
time.Sleep(30 * time.Second)
}
return attempt < 5, err
})
if err != nil {
return "", err
}
return value, nil
} }
func SetCurrentIp(newip string, domain string, configuration Configuration) error { func SetCurrentIp(newip string, domain string, configuration Configuration) error {