Remove extraction.Result type, simplify code.

This commit is contained in:
Julius Volz 2014-12-31 14:01:19 +01:00
parent 9a4ca68a61
commit d6b9e97655
6 changed files with 28 additions and 46 deletions

13
main.go
View file

@ -23,7 +23,6 @@ import (
"time" "time"
"github.com/golang/glog" "github.com/golang/glog"
"github.com/prometheus/client_golang/extraction"
clientmodel "github.com/prometheus/client_golang/model" clientmodel "github.com/prometheus/client_golang/model"
registry "github.com/prometheus/client_golang/prometheus" registry "github.com/prometheus/client_golang/prometheus"
@ -81,7 +80,7 @@ var (
) )
type prometheus struct { type prometheus struct {
unwrittenSamples chan *extraction.Result unwrittenSamples chan clientmodel.Samples
ruleManager manager.RuleManager ruleManager manager.RuleManager
targetManager retrieval.TargetManager targetManager retrieval.TargetManager
@ -102,7 +101,7 @@ func NewPrometheus() *prometheus {
glog.Fatalf("Error loading configuration from %s: %v", *configFile, err) glog.Fatalf("Error loading configuration from %s: %v", *configFile, err)
} }
unwrittenSamples := make(chan *extraction.Result, *samplesQueueCapacity) unwrittenSamples := make(chan clientmodel.Samples, *samplesQueueCapacity)
ingester := &retrieval.MergeLabelsIngester{ ingester := &retrieval.MergeLabelsIngester{
Labels: conf.GlobalLabels(), Labels: conf.GlobalLabels(),
@ -214,11 +213,11 @@ func (p *prometheus) Serve() {
} }
}() }()
for block := range p.unwrittenSamples { for samples := range p.unwrittenSamples {
if block.Err == nil && len(block.Samples) > 0 { if len(samples) > 0 {
p.storage.AppendSamples(block.Samples) p.storage.AppendSamples(samples)
if p.remoteTSDBQueue != nil { if p.remoteTSDBQueue != nil {
p.remoteTSDBQueue.Queue(block.Samples) p.remoteTSDBQueue.Queue(samples)
} }
} }
} }

View file

@ -14,11 +14,11 @@
package retrieval package retrieval
import ( import (
"github.com/prometheus/client_golang/extraction" clientmodel "github.com/prometheus/client_golang/model"
) )
type nopIngester struct{} type nopIngester struct{}
func (i nopIngester) Ingest(*extraction.Result) error { func (i nopIngester) Ingest(clientmodel.Samples) error {
return nil return nil
} }

View file

@ -31,19 +31,19 @@ type MergeLabelsIngester struct {
// Ingest ingests the provided extraction result by merging in i.Labels and then // Ingest ingests the provided extraction result by merging in i.Labels and then
// handing it over to i.Ingester. // handing it over to i.Ingester.
func (i *MergeLabelsIngester) Ingest(r *extraction.Result) error { func (i *MergeLabelsIngester) Ingest(samples clientmodel.Samples) error {
for _, s := range r.Samples { for _, s := range samples {
s.Metric.MergeFromLabelSet(i.Labels, i.CollisionPrefix) s.Metric.MergeFromLabelSet(i.Labels, i.CollisionPrefix)
} }
return i.Ingester.Ingest(r) return i.Ingester.Ingest(samples)
} }
// ChannelIngester feeds results into a channel without modifying them. // ChannelIngester feeds results into a channel without modifying them.
type ChannelIngester chan<- *extraction.Result type ChannelIngester chan<- clientmodel.Samples
// Ingest ingests the provided extraction result by sending it to i. // Ingest ingests the provided extraction result by sending it to i.
func (i ChannelIngester) Ingest(r *extraction.Result) error { func (i ChannelIngester) Ingest(s clientmodel.Samples) error {
i <- r i <- s
return nil return nil
} }

View file

@ -205,10 +205,7 @@ func (t *target) recordScrapeHealth(ingester extraction.Ingester, timestamp clie
Value: clientmodel.SampleValue(float64(scrapeDuration) / float64(time.Second)), Value: clientmodel.SampleValue(float64(scrapeDuration) / float64(time.Second)),
} }
ingester.Ingest(&extraction.Result{ ingester.Ingest(clientmodel.Samples{healthSample, durationSample})
Err: nil,
Samples: clientmodel.Samples{healthSample, durationSample},
})
} }
// RunScraper implements Target. // RunScraper implements Target.

View file

@ -22,17 +22,15 @@ import (
clientmodel "github.com/prometheus/client_golang/model" clientmodel "github.com/prometheus/client_golang/model"
"github.com/prometheus/client_golang/extraction"
"github.com/prometheus/prometheus/utility" "github.com/prometheus/prometheus/utility"
) )
type collectResultIngester struct { type collectResultIngester struct {
result *extraction.Result result clientmodel.Samples
} }
func (i *collectResultIngester) Ingest(r *extraction.Result) error { func (i *collectResultIngester) Ingest(s clientmodel.Samples) error {
i.result = r i.result = s
return nil return nil
} }
@ -57,15 +55,15 @@ func TestTargetRecordScrapeHealth(t *testing.T) {
now := clientmodel.Now() now := clientmodel.Now()
ingester := &collectResultIngester{} ingester := &collectResultIngester{}
testTarget.recordScrapeHealth(ingester, now, true, 2 * time.Second) testTarget.recordScrapeHealth(ingester, now, true, 2*time.Second)
result := ingester.result result := ingester.result
if len(result.Samples) != 2 { if len(result) != 2 {
t.Fatalf("Expected two samples, got %d", len(result.Samples)) t.Fatalf("Expected two samples, got %d", len(result))
} }
actual := result.Samples[0] actual := result[0]
expected := &clientmodel.Sample{ expected := &clientmodel.Sample{
Metric: clientmodel.Metric{ Metric: clientmodel.Metric{
clientmodel.MetricNameLabel: scrapeHealthMetricName, clientmodel.MetricNameLabel: scrapeHealthMetricName,
@ -76,15 +74,11 @@ func TestTargetRecordScrapeHealth(t *testing.T) {
Value: 1, Value: 1,
} }
if result.Err != nil {
t.Fatalf("Got unexpected error: %v", result.Err)
}
if !actual.Equal(expected) { if !actual.Equal(expected) {
t.Fatalf("Expected and actual samples not equal. Expected: %v, actual: %v", expected, actual) t.Fatalf("Expected and actual samples not equal. Expected: %v, actual: %v", expected, actual)
} }
actual = result.Samples[1] actual = result[1]
expected = &clientmodel.Sample{ expected = &clientmodel.Sample{
Metric: clientmodel.Metric{ Metric: clientmodel.Metric{
clientmodel.MetricNameLabel: scrapeDurationMetricName, clientmodel.MetricNameLabel: scrapeDurationMetricName,
@ -95,10 +89,6 @@ func TestTargetRecordScrapeHealth(t *testing.T) {
Value: 2.0, Value: 2.0,
} }
if result.Err != nil {
t.Fatalf("Got unexpected error: %v", result.Err)
}
if !actual.Equal(expected) { if !actual.Equal(expected) {
t.Fatalf("Expected and actual samples not equal. Expected: %v, actual: %v", expected, actual) t.Fatalf("Expected and actual samples not equal. Expected: %v, actual: %v", expected, actual)
} }

View file

@ -19,7 +19,6 @@ import (
"time" "time"
"github.com/golang/glog" "github.com/golang/glog"
"github.com/prometheus/client_golang/extraction"
"github.com/prometheus/client_golang/prometheus" "github.com/prometheus/client_golang/prometheus"
clientmodel "github.com/prometheus/client_golang/model" clientmodel "github.com/prometheus/client_golang/model"
@ -95,7 +94,7 @@ type ruleManager struct {
interval time.Duration interval time.Duration
storage local.Storage storage local.Storage
results chan<- *extraction.Result results chan<- clientmodel.Samples
notificationHandler *notification.NotificationHandler notificationHandler *notification.NotificationHandler
prometheusURL string prometheusURL string
@ -107,7 +106,7 @@ type RuleManagerOptions struct {
Storage local.Storage Storage local.Storage
NotificationHandler *notification.NotificationHandler NotificationHandler *notification.NotificationHandler
Results chan<- *extraction.Result Results chan<- clientmodel.Samples
PrometheusURL string PrometheusURL string
} }
@ -210,7 +209,7 @@ func (m *ruleManager) queueAlertNotifications(rule *rules.AlertingRule, timestam
m.notificationHandler.SubmitReqs(notifications) m.notificationHandler.SubmitReqs(notifications)
} }
func (m *ruleManager) runIteration(results chan<- *extraction.Result) { func (m *ruleManager) runIteration(results chan<- clientmodel.Samples) {
now := clientmodel.Now() now := clientmodel.Now()
wg := sync.WaitGroup{} wg := sync.WaitGroup{}
@ -240,11 +239,8 @@ func (m *ruleManager) runIteration(results chan<- *extraction.Result) {
if err != nil { if err != nil {
evalFailures.Inc() evalFailures.Inc()
} } else {
m.results <- samples
m.results <- &extraction.Result{
Samples: samples,
Err: err,
} }
switch r := rule.(type) { switch r := rule.(type) {