prometheus/web/web.go

424 lines
11 KiB
Go
Raw Normal View History

// Copyright 2013 The Prometheus Authors
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package web
import (
"encoding/json"
"fmt"
2015-06-15 03:50:53 -07:00
"io"
"io/ioutil"
"net/http"
2015-06-15 03:50:53 -07:00
"net/url"
"os"
2015-06-15 03:50:53 -07:00
"path/filepath"
"sort"
"strings"
2015-06-15 03:50:53 -07:00
"sync"
"time"
2015-06-15 03:50:53 -07:00
template_std "html/template"
pprof_runtime "runtime/pprof"
clientmodel "github.com/prometheus/client_golang/model"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/log"
2015-06-15 03:50:53 -07:00
"github.com/prometheus/prometheus/config"
"github.com/prometheus/prometheus/promql"
"github.com/prometheus/prometheus/retrieval"
"github.com/prometheus/prometheus/rules"
"github.com/prometheus/prometheus/storage/local"
"github.com/prometheus/prometheus/template"
"github.com/prometheus/prometheus/util/route"
"github.com/prometheus/prometheus/version"
"github.com/prometheus/prometheus/web/api/legacy"
"github.com/prometheus/prometheus/web/api/v1"
"github.com/prometheus/prometheus/web/blob"
)
var localhostRepresentations = []string{"127.0.0.1", "localhost"}
2015-06-15 03:50:53 -07:00
// Handler serves various HTTP endpoints of the Prometheus server
type Handler struct {
ruleManager *rules.Manager
queryEngine *promql.Engine
apiV1 *v1.API
apiLegacy *legacy.API
federation *Federation
2015-06-15 03:50:53 -07:00
router *route.Router
quitCh chan struct{}
options *Options
statusInfo *PrometheusStatus
2015-06-15 03:50:53 -07:00
muAlerts sync.Mutex
}
2013-02-08 06:38:50 -08:00
2015-06-15 03:50:53 -07:00
// PrometheusStatus contains various information about the status
// of the running Prometheus process.
type PrometheusStatus struct {
Birth time.Time
Flags map[string]string
Config string
2015-06-15 03:50:53 -07:00
// A function that returns the current scrape targets pooled
// by their job name.
TargetPools func() map[string][]*retrieval.Target
// A function that returns all loaded rules.
Rules func() []rules.Rule
mu sync.RWMutex
}
2015-06-15 03:50:53 -07:00
// ApplyConfig updates the status state as the new config requires.
// Returns true on success.
func (s *PrometheusStatus) ApplyConfig(conf *config.Config) bool {
s.mu.Lock()
s.Config = conf.String()
s.mu.Unlock()
return true
}
2015-06-15 03:50:53 -07:00
// Options for the web Handler.
type Options struct {
PathPrefix string
ListenAddress string
Hostname string
MetricsPath string
UseLocalAssets bool
UserAssetsPath string
ConsoleTemplatesPath string
ConsoleLibrariesPath string
EnableQuit bool
}
// New initializes a new web Handler.
func New(st local.Storage, qe *promql.Engine, rm *rules.Manager, status *PrometheusStatus, o *Options) *Handler {
router := route.New()
2015-06-15 03:50:53 -07:00
h := &Handler{
router: router,
quitCh: make(chan struct{}),
options: o,
statusInfo: status,
2015-06-15 03:50:53 -07:00
ruleManager: rm,
queryEngine: qe,
apiV1: &v1.API{
QueryEngine: qe,
Storage: st,
},
apiLegacy: &legacy.API{
QueryEngine: qe,
Storage: st,
Now: clientmodel.Now,
},
federation: &Federation{
Storage: st,
},
}
if o.PathPrefix != "" {
2015-06-15 03:50:53 -07:00
// If the prefix is missing for the root path, prepend it.
router.Get("/", func(w http.ResponseWriter, r *http.Request) {
2015-06-15 03:50:53 -07:00
http.Redirect(w, r, o.PathPrefix, http.StatusFound)
})
router = router.WithPrefix(o.PathPrefix)
}
2015-06-15 03:50:53 -07:00
instrf := prometheus.InstrumentHandlerFunc
instrh := prometheus.InstrumentHandler
router.Get("/", instrf("status", h.status))
2015-06-15 03:50:53 -07:00
router.Get("/alerts", instrf("alerts", h.alerts))
router.Get("/graph", instrf("graph", h.graph))
router.Get("/version", instrf("version", h.version))
2015-06-15 03:50:53 -07:00
router.Get("/heap", instrf("heap", dumpHeap))
router.Get("/federate", instrh("federate", h.federation))
2015-06-15 03:50:53 -07:00
router.Get(o.MetricsPath, prometheus.Handler().ServeHTTP)
2015-06-15 03:50:53 -07:00
h.apiLegacy.Register(router.WithPrefix("/api"))
h.apiV1.Register(router.WithPrefix("/api/v1"))
2015-06-15 03:50:53 -07:00
router.Get("/consoles/*filepath", instrf("consoles", h.consoles))
2015-06-15 03:50:53 -07:00
if o.UseLocalAssets {
router.Get("/static/*filepath", instrf("static", route.FileServe("web/blob/static")))
} else {
2015-06-15 03:50:53 -07:00
router.Get("/static/*filepath", instrh("static", blob.Handler{}))
}
2015-06-15 03:50:53 -07:00
if o.UserAssetsPath != "" {
router.Get("/user/*filepath", instrf("user", route.FileServe(o.UserAssetsPath)))
}
2015-06-15 03:50:53 -07:00
if o.EnableQuit {
router.Post("/-/quit", h.quit)
}
2015-06-15 03:50:53 -07:00
return h
}
// Quit returns the receive-only quit channel.
func (h *Handler) Quit() <-chan struct{} {
return h.quitCh
}
// Run serves the HTTP endpoints.
2015-06-15 03:50:53 -07:00
func (h *Handler) Run() {
log.Infof("Listening on %s", h.options.ListenAddress)
2015-05-19 05:38:50 -07:00
// If we cannot bind to a port, retry after 30 seconds.
for {
2015-06-15 03:50:53 -07:00
err := http.ListenAndServe(h.options.ListenAddress, h.router)
2015-05-19 05:38:50 -07:00
if err != nil {
2015-06-15 03:50:53 -07:00
log.Errorf("Could not listen on %s: %s", h.options.ListenAddress, err)
2015-05-19 05:38:50 -07:00
}
time.Sleep(30 * time.Second)
}
}
2015-06-15 03:50:53 -07:00
func (h *Handler) alerts(w http.ResponseWriter, r *http.Request) {
h.muAlerts.Lock()
defer h.muAlerts.Unlock()
alerts := h.ruleManager.AlertingRules()
alertsSorter := byAlertStateSorter{alerts: alerts}
sort.Sort(alertsSorter)
alertStatus := AlertStatus{
AlertingRules: alertsSorter.alerts,
AlertStateToRowClass: map[rules.AlertState]string{
rules.StateInactive: "success",
rules.StatePending: "warning",
rules.StateFiring: "danger",
},
}
h.executeTemplate(w, "alerts", alertStatus)
}
func (h *Handler) consoles(w http.ResponseWriter, r *http.Request) {
ctx := route.Context(r)
name := route.Param(ctx, "filepath")
file, err := http.Dir(h.options.ConsoleTemplatesPath).Open(name)
if err != nil {
http.Error(w, err.Error(), http.StatusNotFound)
return
}
text, err := ioutil.ReadAll(file)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
// Provide URL parameters as a map for easy use. Advanced users may have need for
// parameters beyond the first, so provide RawParams.
rawParams, err := url.ParseQuery(r.URL.RawQuery)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
params := map[string]string{}
for k, v := range rawParams {
params[k] = v[0]
}
data := struct {
RawParams url.Values
Params map[string]string
Path string
}{
RawParams: rawParams,
Params: params,
Path: name,
}
tmpl := template.NewTemplateExpander(string(text), "__console_"+name, data, clientmodel.Now(), h.queryEngine, h.options.PathPrefix)
filenames, err := filepath.Glob(h.options.ConsoleLibrariesPath + "/*.lib")
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
result, err := tmpl.ExpandHTML(filenames)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
io.WriteString(w, result)
}
func (h *Handler) graph(w http.ResponseWriter, r *http.Request) {
h.executeTemplate(w, "graph", nil)
}
func (h *Handler) status(w http.ResponseWriter, r *http.Request) {
h.statusInfo.mu.RLock()
defer h.statusInfo.mu.RUnlock()
2015-06-15 03:50:53 -07:00
h.executeTemplate(w, "status", struct {
Status *PrometheusStatus
Info map[string]string
}{
Status: h.statusInfo,
Info: version.Map,
})
2015-06-15 03:50:53 -07:00
}
func (h *Handler) version(w http.ResponseWriter, r *http.Request) {
dec := json.NewEncoder(w)
if err := dec.Encode(version.Map); err != nil {
http.Error(w, fmt.Sprintf("error encoding JSON: %s", err), http.StatusInternalServerError)
}
}
2015-06-15 03:50:53 -07:00
func (h *Handler) quit(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "Requesting termination... Goodbye!")
close(h.quitCh)
}
2015-06-15 03:50:53 -07:00
func (h *Handler) getTemplateFile(name string) (string, error) {
if h.options.UseLocalAssets {
file, err := ioutil.ReadFile(fmt.Sprintf("web/blob/templates/%s.html", name))
if err != nil {
log.Errorf("Could not read %s template: %s", name, err)
return "", err
}
return string(file), nil
}
file, err := blob.GetFile(blob.TemplateFiles, name+".html")
if err != nil {
log.Errorf("Could not read %s template: %s", name, err)
return "", err
}
return string(file), nil
}
2015-06-15 03:50:53 -07:00
func (h *Handler) getConsoles() string {
if _, err := os.Stat(h.options.ConsoleTemplatesPath + "/index.html"); !os.IsNotExist(err) {
return h.options.PathPrefix + "/consoles/index.html"
}
2015-06-15 03:50:53 -07:00
if h.options.UserAssetsPath != "" {
if _, err := os.Stat(h.options.UserAssetsPath + "/index.html"); !os.IsNotExist(err) {
return h.options.PathPrefix + "/user/index.html"
}
}
return ""
}
2015-06-15 03:50:53 -07:00
func (h *Handler) getTemplate(name string) (*template_std.Template, error) {
t := template_std.New("_base")
var err error
2015-06-15 03:50:53 -07:00
t.Funcs(template_std.FuncMap{
"since": time.Since,
2015-06-15 03:50:53 -07:00
"getConsoles": h.getConsoles,
"pathPrefix": func() string { return h.options.PathPrefix },
"stripLabels": func(lset clientmodel.LabelSet, labels ...clientmodel.LabelName) clientmodel.LabelSet {
for _, ln := range labels {
delete(lset, ln)
}
return lset
},
"globalURL": func(u *url.URL) *url.URL {
for _, lhr := range localhostRepresentations {
if strings.HasPrefix(u.Host, lhr+":") {
u.Host = strings.Replace(u.Host, lhr+":", h.options.Hostname+":", 1)
}
}
return u
},
2015-06-15 03:50:53 -07:00
"healthToClass": func(th retrieval.TargetHealth) string {
switch th {
case retrieval.HealthUnknown:
return "warning"
case retrieval.HealthGood:
return "success"
default:
return "danger"
}
},
})
2015-06-15 03:50:53 -07:00
file, err := h.getTemplateFile("_base")
if err != nil {
log.Errorln("Could not read base template:", err)
return nil, err
}
t, err = t.Parse(file)
if err != nil {
log.Errorln("Could not parse base template:", err)
}
2015-06-15 03:50:53 -07:00
file, err = h.getTemplateFile(name)
if err != nil {
log.Error("Could not read template %s: %s", name, err)
return nil, err
}
t, err = t.Parse(file)
if err != nil {
log.Errorf("Could not parse template %s: %s", name, err)
}
return t, err
}
2015-06-15 03:50:53 -07:00
func (h *Handler) executeTemplate(w http.ResponseWriter, name string, data interface{}) {
tpl, err := h.getTemplate(name)
if err != nil {
log.Error("Error preparing layout template: ", err)
return
}
err = tpl.Execute(w, data)
if err != nil {
log.Error("Error executing template: ", err)
}
}
func dumpHeap(w http.ResponseWriter, r *http.Request) {
target := fmt.Sprintf("/tmp/%d.heap", time.Now().Unix())
f, err := os.Create(target)
if err != nil {
log.Error("Could not dump heap: ", err)
}
fmt.Fprintf(w, "Writing to %s...", target)
defer f.Close()
pprof_runtime.WriteHeapProfile(f)
fmt.Fprintf(w, "Done")
}
2015-06-15 03:50:53 -07:00
// AlertStatus bundles alerting rules and the mapping of alert states to row classes.
type AlertStatus struct {
AlertingRules []*rules.AlertingRule
AlertStateToRowClass map[rules.AlertState]string
}
2015-06-15 03:50:53 -07:00
type byAlertStateSorter struct {
alerts []*rules.AlertingRule
}
func (s byAlertStateSorter) Len() int {
return len(s.alerts)
}
func (s byAlertStateSorter) Less(i, j int) bool {
return s.alerts[i].State() > s.alerts[j].State()
}
func (s byAlertStateSorter) Swap(i, j int) {
s.alerts[i], s.alerts[j] = s.alerts[j], s.alerts[i]
}