tyler
/
armbian-router
1
0
Fork 0
Code Issues Pull Requests Projects Releases 17 Wiki Activity

Cleanup status checks, debugging

This commit is contained in:
Tyler 2022-01-11 01:10:36 -05:00
parent e4a14fb9fd
commit 516390b17d
1 changed files with 54 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

92
servers.go
View File

@ -27,6 +27,59 @@ type Server struct {
Redirects prometheus.Counter Redirects prometheus.Counter
} }
func (server *Server) checkStatus() {
req, err := http.NewRequest(http.MethodGet, "https://"+server.Host+"/"+strings.TrimLeft(server.Path, "/"), nil)
req.Header.Set("User-Agent", "ArmbianRouter/1.0 (Go "+runtime.Version()+")")
if err != nil {
// This should never happen.
log.WithFields(log.Fields{
"server": server.Host,
"error": err,
}).Warning("Invalid request! This should not happen, please check config.")
return
}
res, err := checkClient.Do(req)
if err != nil {
if server.Available {
log.WithFields(log.Fields{
"server": server.Host,
"error": err,
}).Info("Server went offline")
server.Available = false
} else {
log.WithFields(log.Fields{
"server": server.Host,
"error": err,
}).Debug("Server is still offline")
}
return
}
responseFields := log.Fields{
"server": server.Host,
"responseCode": res.StatusCode,
}
if res.StatusCode == http.StatusOK || res.StatusCode == http.StatusMovedPermanently || res.StatusCode == http.StatusFound || res.StatusCode == http.StatusNotFound {
if !server.Available {
server.Available = true
log.WithFields(responseFields).Info("Server is online")
}
} else {
log.WithFields(responseFields).Debug("Server status not known")
if server.Available {
log.WithFields(responseFields).Info("Server went offline")
server.Available = false
}
}
}
type ServerList []*Server type ServerList []*Server
func (s ServerList) checkLoop() { func (s ServerList) checkLoop() {
@ -50,44 +103,7 @@ func (s ServerList) Check() {
go func(server *Server) { go func(server *Server) {
defer wg.Done() defer wg.Done()
req, err := http.NewRequest(http.MethodGet, "https://"+server.Host+"/"+strings.TrimLeft(server.Path, "/"), nil) server.checkStatus()
req.Header.Set("User-Agent", "ArmbianRouter/1.0 (Go "+runtime.Version()+")")
if err != nil {
// This should never happen.
log.WithError(err).Warning("Invalid request! This should not happen, please check config.")
return
}
res, err := checkClient.Do(req)
if err != nil {
if server.Available {
log.WithField("server", server.Host).Info("Server went offline")
server.Available = false
} else {
log.WithField("server", server.Host).Info("Server is still offline")
}
return
}
if res.StatusCode == http.StatusOK || res.StatusCode == http.StatusMovedPermanently || res.StatusCode == http.StatusFound || res.StatusCode == http.StatusNotFound {
if !server.Available {
server.Available = true
log.WithField("server", server.Host).Info("Server is online")
}
} else {
log.WithFields(log.Fields{
"server": server.Host,
"responseCode": res.StatusCode,
}).Debug("Server status not known")
if server.Available {
log.WithField("server", server.Host).Info("Server went offline")
server.Available = false
}
}
}(server) }(server)
} }