mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-01-11 13:57:36 -08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Use templates.TemplateExpander for all page templates.

Browse source 
This will allow us to use all of our shiny templating functions in the
main page templates as well.
This commit is contained in:
Julius Volz 2015-06-23 17:46:50 +02:00
parent 7d0850571d
commit d7ea0397d4
1 changed files with 35 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

69
web/web.go
View file

@ -27,8 +27,8 @@ import (
"sync" "sync"
"time" "time"
template_std "html/template"
pprof_runtime "runtime/pprof" pprof_runtime "runtime/pprof"
template_text "text/template"
clientmodel "github.com/prometheus/client_golang/model" clientmodel "github.com/prometheus/client_golang/model"
"github.com/prometheus/client_golang/prometheus" "github.com/prometheus/client_golang/prometheus"
@ -320,11 +320,26 @@ func (h *Handler) getConsoles() string {
return "" return ""
} }
func (h *Handler) getTemplate(name string) (*template_std.Template, error) { func (h *Handler) getTemplate(name string) (string, error) {
t := template_std.New("_base") baseTmpl, err := h.getTemplateFile("_base")
var err error if err != nil {
return "", fmt.Errorf("Error reading base template: %s", err)
}
pageTmpl, err := h.getTemplateFile(name)
if err != nil {
return "", fmt.Errorf("Error reading page template %s: %s", name, err)
}
return baseTmpl + pageTmpl, nil
}
t.Funcs(template_std.FuncMap{ func (h *Handler) executeTemplate(w http.ResponseWriter, name string, data interface{}) {
text, err := h.getTemplate(name)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
tmpl := template.NewTemplateExpander(text, name, data, clientmodel.Now(), h.queryEngine, h.options.PathPrefix)
tmpl.Funcs(template_text.FuncMap{
"since": time.Since, "since": time.Since,
"getConsoles": h.getConsoles, "getConsoles": h.getConsoles,
"pathPrefix": func() string { return h.options.PathPrefix }, "pathPrefix": func() string { return h.options.PathPrefix },
@ -352,40 +367,26 @@ func (h *Handler) getTemplate(name string) (*template_std.Template, error) {
return "danger" return "danger"
} }
}, },
"alertStateToClass": func(as rules.AlertState) string {
switch as {
case rules.StateInactive:
return "success"
case rules.StatePending:
return "warning"
case rules.StateFiring:
return "danger"
default:
panic("unknown alert state")
}
},
}) })
file, err := h.getTemplateFile("_base") result, err := tmpl.ExpandHTML(nil)
if err != nil { if err != nil {
log.Errorln("Could not read base template:", err) http.Error(w, err.Error(), http.StatusInternalServerError)
return nil, err
}
t, err = t.Parse(file)
if err != nil {
log.Errorln("Could not parse base template:", err)
}
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
}
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 return
} }
err = tpl.Execute(w, data) io.WriteString(w, result)
if err != nil {
log.Error("Error executing template: ", err)
}
} }
func dumpHeap(w http.ResponseWriter, r *http.Request) { func dumpHeap(w http.ResponseWriter, r *http.Request) {