add heartbeat

pkg/metrics/metrics.go

@@ -0,0 +1,31 @@
+package metrics
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"net/http"
+)
+
+type Metrics struct {
+	url string
+}
+
+func New(url string) *Metrics {
+	return &Metrics{url: url}
+}
+
+func (m *Metrics) Post(name string, attrs map[string]interface{}) error {
+	data, err := json.Marshal(attrs)
+	if err != nil {
+		return err
+	}
+
+	res, err := http.Post(fmt.Sprintf("%s/%s", m.url, name), "application/json", bytes.NewReader(data))
+	if err != nil {
+		return err
+	}
+	defer res.Body.Close()
+
+	return nil
+}

provider/aws/heartbeat.go

@@ -0,0 +1,21 @@
+package aws
+
+import (
+	"strings"
+
+	"github.com/convox/convox/pkg/common"
+)
+
+func (p *Provider) Heartbeat() (map[string]interface{}, error) {
+	data, err := common.Get("http://169.254.169.254/latest/meta-data/instance-type")
+	if err != nil {
+		return nil, err
+	}
+
+	hs := map[string]interface{}{
+		"instance_type": strings.TrimSpace(string(data)),
+		"region":        p.Region,
+	}
+
+	return hs, nil
+}

provider/gcp/heartbeat.go

@@ -0,0 +1,36 @@
+package gcp
+
+import (
+	"io/ioutil"
+	"net/http"
+	"strings"
+)
+
+func (p *Provider) Heartbeat() (map[string]interface{}, error) {
+	req, err := http.NewRequest("GET", "http://metadata.google.internal/computeMetadata/v1/instance/machine-type", nil)
+	if err != nil {
+		return nil, err
+	}
+
+	req.Header.Add("Metadata-Flavor", "Google")
+
+	res, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer res.Body.Close()
+
+	data, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	tparts := strings.Split(strings.TrimSpace(string(data)), "/")
+
+	hs := map[string]interface{}{
+		"instance_type": tparts[len(tparts)-1],
+		"region":        p.Region,
+	}
+
+	return hs, nil
+}

provider/k8s/k8s.go

@@ -212,7 +212,7 @@ func (p *Provider) heartbeat() error {
 		ms[k] = v
 	}
 
-	if err := p.metrics.Post("heartbeat", hs); err != nil {
+	if err := p.metrics.Post("heartbeat", ms); err != nil {
 		return err
 	}