prometheus/web/api/query.go

90 lines
2 KiB
Go
Raw Normal View History

package api
import (
"code.google.com/p/gorest"
"encoding/json"
2013-01-17 16:54:26 -08:00
"errors"
"github.com/prometheus/prometheus/rules"
"github.com/prometheus/prometheus/rules/ast"
"log"
"net/http"
2013-01-17 15:07:00 -08:00
"sort"
"time"
)
2013-01-15 02:30:55 -08:00
func (serv MetricsService) Query(Expr string, Json string) (result string) {
exprNode, err := rules.LoadExprFromString(Expr)
if err != nil {
2013-01-17 16:54:26 -08:00
return ast.ErrorToJSON(err)
}
timestamp := serv.time.Now()
rb := serv.ResponseBuilder()
var format ast.OutputFormat
if Json != "" {
format = ast.JSON
rb.SetContentType(gorest.Application_Json)
} else {
format = ast.TEXT
rb.SetContentType(gorest.Text_Plain)
}
return ast.EvalToString(exprNode, &timestamp, format)
}
2013-01-15 02:30:55 -08:00
func (serv MetricsService) QueryRange(Expr string, End int64, Range int64, Step int64) string {
exprNode, err := rules.LoadExprFromString(Expr)
if err != nil {
2013-01-17 16:54:26 -08:00
return ast.ErrorToJSON(err)
2013-01-15 02:30:55 -08:00
}
2013-01-17 15:07:00 -08:00
if exprNode.Type() != ast.VECTOR {
2013-01-17 16:54:26 -08:00
return ast.ErrorToJSON(errors.New("Expression does not evaluate to vector type"))
2013-01-17 15:07:00 -08:00
}
2013-01-15 02:30:55 -08:00
rb := serv.ResponseBuilder()
2013-01-17 15:07:00 -08:00
rb.SetContentType(gorest.Application_Json)
2013-01-15 02:30:55 -08:00
2013-01-17 15:07:00 -08:00
if End == 0 {
End = serv.time.Now().Unix()
2013-01-17 15:07:00 -08:00
}
2013-01-15 02:30:55 -08:00
2013-01-17 15:07:00 -08:00
if Step < 1 {
Step = 1
}
2013-01-15 02:30:55 -08:00
2013-01-17 15:07:00 -08:00
if End-Range < 0 {
Range = End
}
2013-01-15 02:30:55 -08:00
2013-01-17 15:07:00 -08:00
// Align the start to step "tick" boundary.
End -= End % Step
2013-01-15 02:30:55 -08:00
2013-01-17 15:07:00 -08:00
matrix := ast.EvalVectorRange(
exprNode.(ast.VectorNode),
time.Unix(End-Range, 0),
time.Unix(End, 0),
time.Duration(Step)*time.Second)
2013-01-15 02:30:55 -08:00
2013-01-17 15:07:00 -08:00
sort.Sort(matrix)
return ast.TypedValueToJSON(matrix, "matrix")
2013-01-15 02:30:55 -08:00
}
func (serv MetricsService) Metrics() string {
metricNames, err := serv.persistence.GetAllMetricNames()
rb := serv.ResponseBuilder()
rb.SetContentType(gorest.Application_Json)
if err != nil {
log.Printf("Error loading metric names: %v", err)
rb.SetResponseCode(http.StatusInternalServerError)
return err.Error()
}
sort.Strings(metricNames)
resultBytes, err := json.Marshal(metricNames)
if err != nil {
log.Printf("Error marshalling metric names: %v", err)
rb.SetResponseCode(http.StatusInternalServerError)
return err.Error()
}
return string(resultBytes)
}