mirror of https://github.com/prometheus/prometheus
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
705 lines
20 KiB
705 lines
20 KiB
// Copyright 2013 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 scrape |
|
|
|
import ( |
|
"context" |
|
"net/http" |
|
"strconv" |
|
"testing" |
|
"time" |
|
|
|
"github.com/prometheus/common/model" |
|
"github.com/stretchr/testify/require" |
|
"gopkg.in/yaml.v2" |
|
|
|
"github.com/prometheus/prometheus/config" |
|
"github.com/prometheus/prometheus/discovery" |
|
"github.com/prometheus/prometheus/discovery/targetgroup" |
|
"github.com/prometheus/prometheus/model/labels" |
|
"github.com/prometheus/prometheus/model/relabel" |
|
) |
|
|
|
func TestPopulateLabels(t *testing.T) { |
|
cases := []struct { |
|
in labels.Labels |
|
cfg *config.ScrapeConfig |
|
noDefaultPort bool |
|
res labels.Labels |
|
resOrig labels.Labels |
|
err string |
|
}{ |
|
// Regular population of scrape config options. |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:1000", |
|
"custom": "value", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
res: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:1000", |
|
model.InstanceLabel: "1.2.3.4:1000", |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
"custom": "value", |
|
}), |
|
resOrig: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:1000", |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
"custom": "value", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
}), |
|
}, |
|
// Pre-define/overwrite scrape config labels. |
|
// Leave out port and expect it to be defaulted to scheme. |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4", |
|
model.SchemeLabel: "http", |
|
model.MetricsPathLabel: "/custom", |
|
model.JobLabel: "custom-job", |
|
model.ScrapeIntervalLabel: "2s", |
|
model.ScrapeTimeoutLabel: "2s", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
res: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:80", |
|
model.InstanceLabel: "1.2.3.4:80", |
|
model.SchemeLabel: "http", |
|
model.MetricsPathLabel: "/custom", |
|
model.JobLabel: "custom-job", |
|
model.ScrapeIntervalLabel: "2s", |
|
model.ScrapeTimeoutLabel: "2s", |
|
}), |
|
resOrig: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4", |
|
model.SchemeLabel: "http", |
|
model.MetricsPathLabel: "/custom", |
|
model.JobLabel: "custom-job", |
|
model.ScrapeIntervalLabel: "2s", |
|
model.ScrapeTimeoutLabel: "2s", |
|
}), |
|
}, |
|
// Provide instance label. HTTPS port default for IPv6. |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "[::1]", |
|
model.InstanceLabel: "custom-instance", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
res: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "[::1]:443", |
|
model.InstanceLabel: "custom-instance", |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
}), |
|
resOrig: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "[::1]", |
|
model.InstanceLabel: "custom-instance", |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
}), |
|
}, |
|
// Address label missing. |
|
{ |
|
in: labels.FromStrings("custom", "value"), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
res: labels.EmptyLabels(), |
|
resOrig: labels.EmptyLabels(), |
|
err: "no address", |
|
}, |
|
// Address label missing, but added in relabelling. |
|
{ |
|
in: labels.FromStrings("custom", "host:1234"), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
RelabelConfigs: []*relabel.Config{ |
|
{ |
|
Action: relabel.Replace, |
|
Regex: relabel.MustNewRegexp("(.*)"), |
|
SourceLabels: model.LabelNames{"custom"}, |
|
Replacement: "${1}", |
|
TargetLabel: string(model.AddressLabel), |
|
}, |
|
}, |
|
}, |
|
res: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "host:1234", |
|
model.InstanceLabel: "host:1234", |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
"custom": "host:1234", |
|
}), |
|
resOrig: labels.FromMap(map[string]string{ |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
"custom": "host:1234", |
|
}), |
|
}, |
|
// Address label missing, but added in relabelling. |
|
{ |
|
in: labels.FromStrings("custom", "host:1234"), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
RelabelConfigs: []*relabel.Config{ |
|
{ |
|
Action: relabel.Replace, |
|
Regex: relabel.MustNewRegexp("(.*)"), |
|
SourceLabels: model.LabelNames{"custom"}, |
|
Replacement: "${1}", |
|
TargetLabel: string(model.AddressLabel), |
|
}, |
|
}, |
|
}, |
|
res: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "host:1234", |
|
model.InstanceLabel: "host:1234", |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
"custom": "host:1234", |
|
}), |
|
resOrig: labels.FromMap(map[string]string{ |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
"custom": "host:1234", |
|
}), |
|
}, |
|
// Invalid UTF-8 in label. |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:1000", |
|
"custom": "\xbd", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
res: labels.EmptyLabels(), |
|
resOrig: labels.EmptyLabels(), |
|
err: "invalid label value for \"custom\": \"\\xbd\"", |
|
}, |
|
// Invalid duration in interval label. |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:1000", |
|
model.ScrapeIntervalLabel: "2notseconds", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
res: labels.EmptyLabels(), |
|
resOrig: labels.EmptyLabels(), |
|
err: "error parsing scrape interval: unknown unit \"notseconds\" in duration \"2notseconds\"", |
|
}, |
|
// Invalid duration in timeout label. |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:1000", |
|
model.ScrapeTimeoutLabel: "2notseconds", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
res: labels.EmptyLabels(), |
|
resOrig: labels.EmptyLabels(), |
|
err: "error parsing scrape timeout: unknown unit \"notseconds\" in duration \"2notseconds\"", |
|
}, |
|
// 0 interval in timeout label. |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:1000", |
|
model.ScrapeIntervalLabel: "0s", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
res: labels.EmptyLabels(), |
|
resOrig: labels.EmptyLabels(), |
|
err: "scrape interval cannot be 0", |
|
}, |
|
// 0 duration in timeout label. |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:1000", |
|
model.ScrapeTimeoutLabel: "0s", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
res: labels.EmptyLabels(), |
|
resOrig: labels.EmptyLabels(), |
|
err: "scrape timeout cannot be 0", |
|
}, |
|
// Timeout less than interval. |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:1000", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "2s", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
res: labels.EmptyLabels(), |
|
resOrig: labels.EmptyLabels(), |
|
err: "scrape timeout cannot be greater than scrape interval (\"2s\" > \"1s\")", |
|
}, |
|
// Don't attach default port. |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
noDefaultPort: true, |
|
res: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4", |
|
model.InstanceLabel: "1.2.3.4", |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
}), |
|
resOrig: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4", |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
}), |
|
}, |
|
// Remove default port (http). |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:80", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "http", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
noDefaultPort: true, |
|
res: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4", |
|
model.InstanceLabel: "1.2.3.4:80", |
|
model.SchemeLabel: "http", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
}), |
|
resOrig: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:80", |
|
model.SchemeLabel: "http", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
}), |
|
}, |
|
// Remove default port (https). |
|
{ |
|
in: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:443", |
|
}), |
|
cfg: &config.ScrapeConfig{ |
|
Scheme: "https", |
|
MetricsPath: "/metrics", |
|
JobName: "job", |
|
ScrapeInterval: model.Duration(time.Second), |
|
ScrapeTimeout: model.Duration(time.Second), |
|
}, |
|
noDefaultPort: true, |
|
res: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4", |
|
model.InstanceLabel: "1.2.3.4:443", |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
}), |
|
resOrig: labels.FromMap(map[string]string{ |
|
model.AddressLabel: "1.2.3.4:443", |
|
model.SchemeLabel: "https", |
|
model.MetricsPathLabel: "/metrics", |
|
model.JobLabel: "job", |
|
model.ScrapeIntervalLabel: "1s", |
|
model.ScrapeTimeoutLabel: "1s", |
|
}), |
|
}, |
|
} |
|
for _, c := range cases { |
|
in := c.in.Copy() |
|
|
|
res, orig, err := PopulateLabels(labels.NewBuilder(c.in), c.cfg, c.noDefaultPort) |
|
if c.err != "" { |
|
require.EqualError(t, err, c.err) |
|
} else { |
|
require.NoError(t, err) |
|
} |
|
require.Equal(t, c.in, in) |
|
require.Equal(t, c.res, res) |
|
require.Equal(t, c.resOrig, orig) |
|
} |
|
} |
|
|
|
func loadConfiguration(t testing.TB, c string) *config.Config { |
|
t.Helper() |
|
|
|
cfg := &config.Config{} |
|
if err := yaml.UnmarshalStrict([]byte(c), cfg); err != nil { |
|
t.Fatalf("Unable to load YAML config: %s", err) |
|
} |
|
return cfg |
|
} |
|
|
|
func noopLoop() loop { |
|
return &testLoop{ |
|
startFunc: func(interval, timeout time.Duration, errc chan<- error) {}, |
|
stopFunc: func() {}, |
|
} |
|
} |
|
|
|
func TestManagerApplyConfig(t *testing.T) { |
|
// Valid initial configuration. |
|
cfgText1 := ` |
|
scrape_configs: |
|
- job_name: job1 |
|
static_configs: |
|
- targets: ["foo:9090"] |
|
` |
|
// Invalid configuration. |
|
cfgText2 := ` |
|
scrape_configs: |
|
- job_name: job1 |
|
scheme: https |
|
static_configs: |
|
- targets: ["foo:9090"] |
|
tls_config: |
|
ca_file: /not/existing/ca/file |
|
` |
|
// Valid configuration. |
|
cfgText3 := ` |
|
scrape_configs: |
|
- job_name: job1 |
|
scheme: https |
|
static_configs: |
|
- targets: ["foo:9090"] |
|
` |
|
var ( |
|
cfg1 = loadConfiguration(t, cfgText1) |
|
cfg2 = loadConfiguration(t, cfgText2) |
|
cfg3 = loadConfiguration(t, cfgText3) |
|
|
|
ch = make(chan struct{}, 1) |
|
) |
|
|
|
opts := Options{} |
|
scrapeManager := NewManager(&opts, nil, nil) |
|
newLoop := func(scrapeLoopOptions) loop { |
|
ch <- struct{}{} |
|
return noopLoop() |
|
} |
|
sp := &scrapePool{ |
|
appendable: &nopAppendable{}, |
|
activeTargets: map[uint64]*Target{ |
|
1: {}, |
|
}, |
|
loops: map[uint64]loop{ |
|
1: noopLoop(), |
|
}, |
|
newLoop: newLoop, |
|
logger: nil, |
|
config: cfg1.ScrapeConfigs[0], |
|
client: http.DefaultClient, |
|
} |
|
scrapeManager.scrapePools = map[string]*scrapePool{ |
|
"job1": sp, |
|
} |
|
|
|
// Apply the initial configuration. |
|
if err := scrapeManager.ApplyConfig(cfg1); err != nil { |
|
t.Fatalf("unable to apply configuration: %s", err) |
|
} |
|
select { |
|
case <-ch: |
|
t.Fatal("reload happened") |
|
default: |
|
} |
|
|
|
// Apply a configuration for which the reload fails. |
|
if err := scrapeManager.ApplyConfig(cfg2); err == nil { |
|
t.Fatalf("expecting error but got none") |
|
} |
|
select { |
|
case <-ch: |
|
t.Fatal("reload happened") |
|
default: |
|
} |
|
|
|
// Apply a configuration for which the reload succeeds. |
|
if err := scrapeManager.ApplyConfig(cfg3); err != nil { |
|
t.Fatalf("unable to apply configuration: %s", err) |
|
} |
|
select { |
|
case <-ch: |
|
default: |
|
t.Fatal("reload didn't happen") |
|
} |
|
|
|
// Re-applying the same configuration shouldn't trigger a reload. |
|
if err := scrapeManager.ApplyConfig(cfg3); err != nil { |
|
t.Fatalf("unable to apply configuration: %s", err) |
|
} |
|
select { |
|
case <-ch: |
|
t.Fatal("reload happened") |
|
default: |
|
} |
|
} |
|
|
|
func TestManagerTargetsUpdates(t *testing.T) { |
|
opts := Options{} |
|
m := NewManager(&opts, nil, nil) |
|
|
|
ts := make(chan map[string][]*targetgroup.Group) |
|
go m.Run(ts) |
|
defer m.Stop() |
|
|
|
tgSent := make(map[string][]*targetgroup.Group) |
|
for x := 0; x < 10; x++ { |
|
|
|
tgSent[strconv.Itoa(x)] = []*targetgroup.Group{ |
|
{ |
|
Source: strconv.Itoa(x), |
|
}, |
|
} |
|
|
|
select { |
|
case ts <- tgSent: |
|
case <-time.After(10 * time.Millisecond): |
|
t.Error("Scrape manager's channel remained blocked after the set threshold.") |
|
} |
|
} |
|
|
|
m.mtxScrape.Lock() |
|
tsetActual := m.targetSets |
|
m.mtxScrape.Unlock() |
|
|
|
// Make sure all updates have been received. |
|
require.Equal(t, tgSent, tsetActual) |
|
|
|
select { |
|
case <-m.triggerReload: |
|
default: |
|
t.Error("No scrape loops reload was triggered after targets update.") |
|
} |
|
} |
|
|
|
func TestSetOffsetSeed(t *testing.T) { |
|
getConfig := func(prometheus string) *config.Config { |
|
cfgText := ` |
|
global: |
|
external_labels: |
|
prometheus: '` + prometheus + `' |
|
` |
|
|
|
cfg := &config.Config{} |
|
if err := yaml.UnmarshalStrict([]byte(cfgText), cfg); err != nil { |
|
t.Fatalf("Unable to load YAML config cfgYaml: %s", err) |
|
} |
|
|
|
return cfg |
|
} |
|
|
|
opts := Options{} |
|
scrapeManager := NewManager(&opts, nil, nil) |
|
|
|
// Load the first config. |
|
cfg1 := getConfig("ha1") |
|
if err := scrapeManager.setOffsetSeed(cfg1.GlobalConfig.ExternalLabels); err != nil { |
|
t.Error(err) |
|
} |
|
offsetSeed1 := scrapeManager.offsetSeed |
|
|
|
if offsetSeed1 == 0 { |
|
t.Error("Offset seed has to be a hash of uint64") |
|
} |
|
|
|
// Load the first config. |
|
cfg2 := getConfig("ha2") |
|
if err := scrapeManager.setOffsetSeed(cfg2.GlobalConfig.ExternalLabels); err != nil { |
|
t.Error(err) |
|
} |
|
offsetSeed2 := scrapeManager.offsetSeed |
|
|
|
if offsetSeed1 == offsetSeed2 { |
|
t.Error("Offset seed should not be the same on different set of external labels") |
|
} |
|
} |
|
|
|
func TestManagerScrapePools(t *testing.T) { |
|
cfgText1 := ` |
|
scrape_configs: |
|
- job_name: job1 |
|
static_configs: |
|
- targets: ["foo:9090"] |
|
- job_name: job2 |
|
static_configs: |
|
- targets: ["foo:9091", "foo:9092"] |
|
` |
|
cfgText2 := ` |
|
scrape_configs: |
|
- job_name: job1 |
|
static_configs: |
|
- targets: ["foo:9090", "foo:9094"] |
|
- job_name: job3 |
|
static_configs: |
|
- targets: ["foo:9093"] |
|
` |
|
var ( |
|
cfg1 = loadConfiguration(t, cfgText1) |
|
cfg2 = loadConfiguration(t, cfgText2) |
|
) |
|
|
|
reload := func(scrapeManager *Manager, cfg *config.Config) { |
|
newLoop := func(scrapeLoopOptions) loop { |
|
return noopLoop() |
|
} |
|
scrapeManager.scrapePools = map[string]*scrapePool{} |
|
for _, sc := range cfg.ScrapeConfigs { |
|
_, cancel := context.WithCancel(context.Background()) |
|
defer cancel() |
|
sp := &scrapePool{ |
|
appendable: &nopAppendable{}, |
|
activeTargets: map[uint64]*Target{}, |
|
loops: map[uint64]loop{ |
|
1: noopLoop(), |
|
}, |
|
newLoop: newLoop, |
|
logger: nil, |
|
config: sc, |
|
client: http.DefaultClient, |
|
cancel: cancel, |
|
} |
|
for _, c := range sc.ServiceDiscoveryConfigs { |
|
staticConfig := c.(discovery.StaticConfig) |
|
for _, group := range staticConfig { |
|
for i := range group.Targets { |
|
sp.activeTargets[uint64(i)] = &Target{} |
|
} |
|
} |
|
} |
|
scrapeManager.scrapePools[sc.JobName] = sp |
|
} |
|
} |
|
|
|
opts := Options{} |
|
scrapeManager := NewManager(&opts, nil, nil) |
|
|
|
reload(scrapeManager, cfg1) |
|
require.ElementsMatch(t, []string{"job1", "job2"}, scrapeManager.ScrapePools()) |
|
|
|
reload(scrapeManager, cfg2) |
|
require.ElementsMatch(t, []string{"job1", "job3"}, scrapeManager.ScrapePools()) |
|
}
|
|
|