http-server-status/main.go
2021-09-20 11:28:00 +02:00

56 lines
No EOL
1 KiB
Go

package main
import (
"encoding/json"
"github.com/rs/zerolog/log"
checks "http-server-status/internal/pkg/checks"
"net/http"
"sync"
)
var checkList []checks.Check
func init() {
checkList = append(checkList, checks.HDD{}, checks.Memory{}, checks.Load{})
}
func main() {
http.HandleFunc("/", handler)
err := http.ListenAndServe(":3003", nil)
log.Fatal().Err(err).Msg("Shutdown")
}
func handler(w http.ResponseWriter, r *http.Request) {
res, gloableRes := checkSystem()
resJson, err := json.Marshal(res)
if err != nil {
log.Fatal().Err(err).Msg("Check Error")
}
if gloableRes {
w.Write(resJson)
return
}
http.Error(w, string(resJson), http.StatusInternalServerError)
}
func checkSystem() (map[string]bool, bool) {
globaleResult := true
wg := sync.WaitGroup{}
res := make(map[string]bool)
for _, c := range checkList {
wg.Add(1)
go func(check checks.Check) {
defer wg.Done()
s, _ := check.Execute()
if s == false {
globaleResult = false
}
res[check.Name()] = s
}(c)
}
wg.Wait()
return res, globaleResult
}