forked from mamu/spacepanel_aggregator
73 lines
1.1 KiB
Go
73 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"errors"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/valyala/fastjson"
|
|
)
|
|
|
|
const (
|
|
SleepTime = 60 * time.Minute
|
|
)
|
|
|
|
var (
|
|
insecureTransport = &http.Transport{
|
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
|
}
|
|
|
|
client = &http.Client{
|
|
Transport: insecureTransport,
|
|
}
|
|
)
|
|
|
|
func StartPollWorker(space *Space) {
|
|
for {
|
|
state, err := requestSpaceState(space.URL)
|
|
if err != nil {
|
|
logrus.Errorf("URL: %s, Err: %v", space.URL, err)
|
|
}
|
|
space.State = state
|
|
time.Sleep(SleepTime)
|
|
}
|
|
}
|
|
|
|
func requestSpaceState(url string) (State, error) {
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
return Unknown, err
|
|
}
|
|
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
return Unknown, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
data, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return Unknown, err
|
|
}
|
|
|
|
value, err := fastjson.ParseBytes(data)
|
|
if err != nil {
|
|
return Unknown, err
|
|
}
|
|
|
|
if !value.Exists("state", "open") {
|
|
return Unknown, errors.New("no space state existing")
|
|
}
|
|
|
|
state := value.GetBool("state", "open")
|
|
|
|
if state {
|
|
return Open, nil
|
|
}
|
|
|
|
return Close, nil
|
|
}
|