mirror of https://github.com/prometheus/prometheus
146 lines
3.8 KiB
Go
146 lines
3.8 KiB
Go
// Copyright 2013 Prometheus Team
|
|
// 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 rules
|
|
|
|
import (
|
|
"github.com/prometheus/prometheus/model"
|
|
"github.com/prometheus/prometheus/rules/ast"
|
|
"github.com/prometheus/prometheus/storage/metric"
|
|
"time"
|
|
)
|
|
|
|
var testDuration5m = time.Duration(5) * time.Minute
|
|
var testStartTime = time.Time{}
|
|
|
|
func getTestValueStream(startVal model.SampleValue,
|
|
endVal model.SampleValue,
|
|
stepVal model.SampleValue) (resultValues []model.SamplePair) {
|
|
currentTime := testStartTime
|
|
for currentVal := startVal; currentVal <= endVal; currentVal += stepVal {
|
|
sample := model.SamplePair{
|
|
Value: currentVal,
|
|
Timestamp: currentTime,
|
|
}
|
|
resultValues = append(resultValues, sample)
|
|
currentTime = currentTime.Add(testDuration5m)
|
|
}
|
|
return resultValues
|
|
}
|
|
|
|
func getTestVectorFromTestMatrix(matrix ast.Matrix) ast.Vector {
|
|
vector := ast.Vector{}
|
|
for _, sampleSet := range matrix {
|
|
lastSample := sampleSet.Values[len(sampleSet.Values)-1]
|
|
vector = append(vector, model.Sample{
|
|
Metric: sampleSet.Metric,
|
|
Value: lastSample.Value,
|
|
Timestamp: lastSample.Timestamp,
|
|
})
|
|
}
|
|
return vector
|
|
}
|
|
|
|
func storeMatrix(storage metric.Storage, matrix ast.Matrix) error {
|
|
for _, sampleSet := range matrix {
|
|
for _, sample := range sampleSet.Values {
|
|
err := storage.AppendSample(model.Sample{
|
|
Metric: sampleSet.Metric,
|
|
Value: sample.Value,
|
|
Timestamp: sample.Timestamp,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var testMatrix = ast.Matrix{
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "http_requests",
|
|
"job": "api-server",
|
|
"instance": "0",
|
|
"group": "production",
|
|
},
|
|
Values: getTestValueStream(0, 100, 10),
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "http_requests",
|
|
"job": "api-server",
|
|
"instance": "1",
|
|
"group": "production",
|
|
},
|
|
Values: getTestValueStream(0, 200, 20),
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "http_requests",
|
|
"job": "api-server",
|
|
"instance": "0",
|
|
"group": "canary",
|
|
},
|
|
Values: getTestValueStream(0, 300, 30),
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "http_requests",
|
|
"job": "api-server",
|
|
"instance": "1",
|
|
"group": "canary",
|
|
},
|
|
Values: getTestValueStream(0, 400, 40),
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "http_requests",
|
|
"job": "app-server",
|
|
"instance": "0",
|
|
"group": "production",
|
|
},
|
|
Values: getTestValueStream(0, 500, 50),
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "http_requests",
|
|
"job": "app-server",
|
|
"instance": "1",
|
|
"group": "production",
|
|
},
|
|
Values: getTestValueStream(0, 600, 60),
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "http_requests",
|
|
"job": "app-server",
|
|
"instance": "0",
|
|
"group": "canary",
|
|
},
|
|
Values: getTestValueStream(0, 700, 70),
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "http_requests",
|
|
"job": "app-server",
|
|
"instance": "1",
|
|
"group": "canary",
|
|
},
|
|
Values: getTestValueStream(0, 800, 80),
|
|
},
|
|
}
|
|
|
|
var testVector = getTestVectorFromTestMatrix(testMatrix)
|