107 lines
2.7 KiB
Go
107 lines
2.7 KiB
Go
// Copyright 2015 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 (
|
|
"net/http"
|
|
|
|
"github.com/golang/protobuf/proto"
|
|
|
|
"github.com/prometheus/prometheus/promql"
|
|
"github.com/prometheus/prometheus/storage/metric"
|
|
|
|
"github.com/prometheus/common/expfmt"
|
|
"github.com/prometheus/common/model"
|
|
|
|
dto "github.com/prometheus/client_model/go"
|
|
)
|
|
|
|
func (h *Handler) federation(w http.ResponseWriter, req *http.Request) {
|
|
h.mtx.RLock()
|
|
defer h.mtx.RUnlock()
|
|
|
|
req.ParseForm()
|
|
|
|
metrics := map[model.Fingerprint]metric.Metric{}
|
|
|
|
for _, s := range req.Form["match[]"] {
|
|
matchers, err := promql.ParseMetricSelector(s)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
for fp, met := range h.storage.MetricsForLabelMatchers(matchers...) {
|
|
metrics[fp] = met
|
|
}
|
|
}
|
|
|
|
format := expfmt.Negotiate(req.Header)
|
|
w.Header().Set("Content-Type", string(format))
|
|
|
|
enc := expfmt.NewEncoder(w, format)
|
|
|
|
protMetric := &dto.Metric{
|
|
Label: []*dto.LabelPair{},
|
|
Untyped: &dto.Untyped{},
|
|
}
|
|
protMetricFam := &dto.MetricFamily{
|
|
Metric: []*dto.Metric{protMetric},
|
|
Type: dto.MetricType_UNTYPED.Enum(),
|
|
}
|
|
|
|
for fp, met := range metrics {
|
|
globalUsed := map[model.LabelName]struct{}{}
|
|
|
|
sp := h.storage.LastSamplePairForFingerprint(fp)
|
|
if sp == nil {
|
|
continue
|
|
}
|
|
|
|
// Reset label slice.
|
|
protMetric.Label = protMetric.Label[:0]
|
|
|
|
for ln, lv := range met.Metric {
|
|
if ln == model.MetricNameLabel {
|
|
protMetricFam.Name = proto.String(string(lv))
|
|
continue
|
|
}
|
|
protMetric.Label = append(protMetric.Label, &dto.LabelPair{
|
|
Name: proto.String(string(ln)),
|
|
Value: proto.String(string(lv)),
|
|
})
|
|
if _, ok := h.globalLabels[ln]; ok {
|
|
globalUsed[ln] = struct{}{}
|
|
}
|
|
}
|
|
|
|
// Attach global labels if they do not exist yet.
|
|
for ln, lv := range h.globalLabels {
|
|
if _, ok := globalUsed[ln]; !ok {
|
|
protMetric.Label = append(protMetric.Label, &dto.LabelPair{
|
|
Name: proto.String(string(ln)),
|
|
Value: proto.String(string(lv)),
|
|
})
|
|
}
|
|
}
|
|
|
|
protMetric.TimestampMs = proto.Int64(int64(sp.Timestamp))
|
|
protMetric.Untyped.Value = proto.Float64(float64(sp.Value))
|
|
|
|
if err := enc.Encode(protMetricFam); err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
}
|