Add rudimentary support for dynamic graphs based on a metrics query.

[sysdb/webui.git] / server / graph.go

diff --git a/server/graph.go b/server/graph.go
index 7509485930e23f60f1dc140e8776d3d0b473c8b5..ddc8c4d17e5f6aec59531fbcf2a5e4f33b50c852 100644 (file)
--- a/server/graph.go
+++ b/server/graph.go
@@ -35,6 +35,7 @@ import (
        "time"
 
        "github.com/gonum/plot/vg"
+       "github.com/sysdb/go/sysdb"
        "github.com/sysdb/webui/graph"
 )
 
@@ -61,11 +62,20 @@ func (s *Server) graph(w http.ResponseWriter, req request) {
                        return
                }
        }
+
        g := &graph.Graph{
-               Start:   start,
-               End:     end,
-               Metrics: []graph.Metric{{Hostname: req.args[0], Identifier: req.args[1]}},
+               Start: start,
+               End:   end,
+       }
+       if req.args[0] == "q" {
+               if g.Metrics, err = s.queryMetrics(req.args[1]); err != nil {
+                       s.badrequest(w, fmt.Errorf("Failed to query metrics: %v", err))
+                       return
+               }
+       } else {
+               g.Metrics = []graph.Metric{{Hostname: req.args[0], Identifier: req.args[1]}}
        }
+
        p, err := g.Plot(s.c)
        if err != nil {
                s.internal(w, err)
@@ -88,4 +98,43 @@ func (s *Server) graph(w http.ResponseWriter, req request) {
        io.Copy(w, &buf)
 }
 
+func (s *Server) queryMetrics(q string) ([]graph.Metric, error) {
+       raw, err := parseQuery(q)
+       if err != nil {
+               return nil, err
+       }
+       if raw.typ != "" && raw.typ != "metrics" {
+               return nil, fmt.Errorf("Invalid object type %q for graphs", raw.typ)
+       }
+
+       var args string
+       for name, value := range raw.args {
+               if len(args) > 0 {
+                       args += " AND"
+               }
+
+               if name == "name" {
+                       args += fmt.Sprintf(" name =~ %s", value)
+               } else {
+                       args += fmt.Sprintf(" %s = %s", name, value)
+               }
+       }
+
+       res, err := s.c.Query("LOOKUP metrics MATCHING" + args)
+       if err != nil {
+               return nil, err
+       }
+       hosts, ok := res.([]sysdb.Host)
+       if !ok {
+               return nil, fmt.Errorf("LOOKUP did not return a list of hosts but %T", res)
+       }
+       var metrics []graph.Metric
+       for _, h := range hosts {
+               for _, m := range h.Metrics {
+                       metrics = append(metrics, graph.Metric{Hostname: h.Name, Identifier: m.Name})
+               }
+       }
+       return metrics, nil
+}
+
 // vim: set tw=78 sw=4 sw=4 noexpandtab :