86 lines
1.8 KiB
Go
86 lines
1.8 KiB
Go
package cachet
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/tls"
|
|
"encoding/json"
|
|
"errors"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
)
|
|
|
|
type CachetAPI struct {
|
|
URL string `json:"url"`
|
|
Token string `json:"token"`
|
|
Insecure bool `json:"insecure"`
|
|
}
|
|
|
|
type CachetResponse struct {
|
|
Data json.RawMessage `json:"data"`
|
|
}
|
|
|
|
// TODO: test
|
|
func (api CachetAPI) Ping() error {
|
|
resp, _, err := api.NewRequest("GET", "/ping", nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if resp.StatusCode != 200 {
|
|
return errors.New("API Responded with non-200 status code")
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
|
|
return nil
|
|
}
|
|
|
|
// SendMetric adds a data point to a cachet monitor
|
|
func (api CachetAPI) SendMetric(id int, lag int64) {
|
|
logrus.Debugf("Sending lag metric ID:%d RTT %vms", id, lag)
|
|
|
|
jsonBytes, _ := json.Marshal(map[string]interface{}{
|
|
"value": lag,
|
|
"timestamp": time.Now().Unix(),
|
|
})
|
|
|
|
resp, _, err := api.NewRequest("POST", "/metrics/"+strconv.Itoa(id)+"/points", jsonBytes)
|
|
if err != nil || resp.StatusCode != 200 {
|
|
logrus.Warnf("Could not log metric! ID: %d, err: %v", id, err)
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
}
|
|
|
|
// TODO: test
|
|
// NewRequest wraps http.NewRequest
|
|
func (api CachetAPI) NewRequest(requestType, url string, reqBody []byte) (*http.Response, CachetResponse, error) {
|
|
req, err := http.NewRequest(requestType, api.URL+url, bytes.NewBuffer(reqBody))
|
|
|
|
req.Header.Set("Content-Type", "application/json")
|
|
req.Header.Set("X-Cachet-Token", api.Token)
|
|
|
|
transport := http.DefaultTransport.(*http.Transport)
|
|
transport.TLSClientConfig = &tls.Config{InsecureSkipVerify: api.Insecure}
|
|
client := &http.Client{
|
|
Transport: transport,
|
|
}
|
|
|
|
res, err := client.Do(req)
|
|
if err != nil {
|
|
return nil, CachetResponse{}, err
|
|
}
|
|
|
|
var body struct {
|
|
Data json.RawMessage `json:"data"`
|
|
}
|
|
err = json.NewDecoder(res.Body).Decode(&body)
|
|
|
|
defer req.Body.Close()
|
|
|
|
return res, body, err
|
|
}
|