func (p *promH) graphCmd(ctx context.Context, w hugot.ResponseWriter, m *hugot.Message) error {
defText, defGraph := false, false
if !hugot.IsTextOnly(w) {
defText = true
defGraph = false
} else {
defText = false
defGraph = true
}
_ = m.Bool("t", defText, "Render the graphs as text sparkline.")
webGraph := m.Bool("g", defGraph, "Render the graphs as a png.")
dur := m.Duration("d", 15*time.Minute, "how far back to render")
if err := m.Parse(); err != nil {
return err
}
if len(m.Args()) == 0 {
return fmt.Errorf("you need to give a query")
}
q := strings.Join(m.Args(), " ")
s := time.Now().Add(-1 * *dur)
e := time.Now()
if *webGraph {
nu := *p.URL()
nu.Path = nu.Path + "graph/thing.png"
qs := nu.Query()
qs.Set("q", q)
qs.Set("s", fmt.Sprintf("%d", s.UnixNano()))
qs.Set("e", fmt.Sprintf("%d", e.UnixNano()))
nu.RawQuery = qs.Encode()
m := hugot.Message{
Channel: m.Channel,
Attachments: []hugot.Attachment{
{
Fallback: "fallback",
Pretext: "image",
ImageURL: nu.String(),
},
},
}
w.Send(ctx, &m)
return nil
}
qapi := prom.NewQueryAPI(*p.client)
d, err := qapi.QueryRange(ctx, q, prom.Range{s, e, 15 * time.Second})
if err != nil {
return err
}
switch d.Type() {
case model.ValScalar:
m := d.(*model.Scalar)
glog.Infof("scalar %v", m)
case model.ValVector:
m := d.(model.Vector)
glog.Infof("vector %v", m)
case model.ValMatrix:
mx := d.(model.Matrix)
sort.Sort(mx)
for _, ss := range mx {
l := line(ss.Values)
fmt.Fprintf(w, "%v\n%s", ss.Metric, l)
}
case model.ValString:
m := d.(*model.String)
glog.Infof("matrix %v", m)
case model.ValNone:
}
return nil
}
func (p *promH) graphHook(w http.ResponseWriter, r *http.Request) {
q, ok := r.URL.Query()["q"]
if !ok || len(q) != 1 {
w.WriteHeader(http.StatusBadRequest)
return
}
s, ok := r.URL.Query()["s"]
if !ok || len(s) != 1 {
w.WriteHeader(http.StatusBadRequest)
return
}
e, ok := r.URL.Query()["e"]
if !ok || len(e) != 1 {
w.WriteHeader(http.StatusBadRequest)
return
}
st, _ := strconv.Atoi(s[0])
et, _ := strconv.Atoi(e[0])
ctx := r.Context()
qapi := prom.NewQueryAPI(*p.client)
d, err := qapi.QueryRange(ctx, q[0], prom.Range{time.Unix(0, int64(st)), time.Unix(0, int64(et)), 15 * time.Second})
if err != nil {
w.WriteHeader(http.StatusBadRequest)
return
}
var mx model.Matrix
switch d.Type() {
case model.ValMatrix:
mx = d.(model.Matrix)
sort.Sort(mx)
default:
w.WriteHeader(http.StatusBadRequest)
return
}
plt, err := plot.New()
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
return
}
plt.Title.Text = q[0]
plt.X.Tick.Marker = plot.UnixTimeTicks{Format: "Mon 15:04:05"}
for _, ss := range mx {
for _, sps := range mx {
err = plotutil.AddLinePoints(plt, ss.Metric.String(), modelToPlot(sps.Values))
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
return
}
}
}
// Save the plot to a PNG file.
var wt io.WriterTo
if wt, err = plt.WriterTo(4*vg.Inch, 2*vg.Inch, "png"); err != nil {
w.WriteHeader(http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "image/png")
if _, err := wt.WriteTo(w); err != nil {
w.WriteHeader(http.StatusInternalServerError)
return
}
}