CI: Move grabpl publish-metrics (#55042)

* move grabpl publish-metrics

* grabpl -> ./bin/build

* fix lint error
This commit is contained in:
Kevin Minehart
2022-09-12 13:28:08 -05:00
committed by GitHub
parent d59bb1e4c2
commit d3af3e0431
5 changed files with 122 additions and 3 deletions

View File

@@ -108,6 +108,12 @@ func main() {
&noInstallDepsFlag,
},
},
{
Name: "publish-metrics",
Usage: "Publish a set of metrics from stdin",
ArgsUsage: "<api-key>",
Action: ArgCountWrapper(1, PublishMetrics),
},
}
if err := app.Run(os.Args); err != nil {

View File

@@ -0,0 +1,41 @@
package main
import (
"encoding/json"
"fmt"
"io"
"log"
"os"
"regexp"
"github.com/grafana/grafana/pkg/build/metrics"
"github.com/urfave/cli/v2"
)
func PublishMetrics(c *cli.Context) error {
apiKey := c.Args().Get(0)
input, err := io.ReadAll(os.Stdin)
if err != nil {
return cli.NewExitError(fmt.Sprintf("Reading from stdin failed: %s", err), 1)
}
reMetrics := regexp.MustCompile(`(?ms)^Metrics: (\{.+\})`)
ms := reMetrics.FindSubmatch(input)
if len(ms) == 0 {
return cli.NewExitError(fmt.Sprintf("Input on wrong format: %q", string(input)), 1)
}
m := map[string]string{}
if err := json.Unmarshal(ms[1], &m); err != nil {
return cli.NewExitError(fmt.Sprintf("decoding metrics failed: %s", err), 1)
}
log.Printf("Received metrics %+v", m)
if err := metrics.Publish(m, apiKey); err != nil {
return cli.NewExitError(fmt.Sprintf("publishing metrics failed: %s", err), 1)
}
return nil
}

View File

@@ -0,0 +1,72 @@
package metrics
import (
"bytes"
"encoding/json"
"fmt"
"log"
"strconv"
"time"
"net/http"
)
type payload struct {
Name string `json:"name"`
Value int `json:"value"`
Interval int `json:"interval"`
MType string `json:"mtype"`
Time int64 `json:"time"`
}
// Publish publishes a set of metrics.
func Publish(metrics map[string]string, apiKey string) error {
log.Println("Publishing metrics")
t := time.Now().Unix()
data := []payload{}
for k, vS := range metrics {
v, err := strconv.Atoi(vS)
if err != nil {
return fmt.Errorf("key %q has value on invalid format: %q", k, vS)
}
data = append(data, payload{
Name: k,
Value: v,
Interval: 60,
MType: "gauge",
Time: t,
})
}
buf := bytes.Buffer{}
enc := json.NewEncoder(&buf)
if err := enc.Encode(data); err != nil {
return err
}
log.Printf("Publishing metrics to https://<user>:<pass>@graphite-us-central1.grafana.net/metrics, JSON: %s",
buf.String())
u := fmt.Sprintf("https://6371:%s@graphite-us-central1.grafana.net/metrics", apiKey)
//nolint:gosec
resp, err := http.Post(u, "application/json", &buf)
if err != nil {
return fmt.Errorf("metrics publishing failed: %w", err)
}
defer func() {
if err := resp.Body.Close(); err != nil {
log.Println("Error closing HTTP body", err)
}
}()
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
return fmt.Errorf("metrics publishing failed with status code %d", resp.StatusCode)
}
log.Printf("Metrics successfully published")
return nil
}