A few parameterizations.

This commit is contained in:
Matt T. Proud 2013-01-25 13:45:21 +01:00 committed by Julius Volz
parent ea1e6471de
commit b2df1c375a
2 changed files with 13 additions and 7 deletions

14
main.go
View file

@ -32,8 +32,11 @@ import (
// Commandline flags.
var (
configFile = flag.String("configFile", "prometheus.conf", "Prometheus configuration file name.")
metricsStoragePath = flag.String("metricsStoragePath", "/tmp/metrics", "Base path for metrics storage.")
configFile = flag.String("configFile", "prometheus.conf", "Prometheus configuration file name.")
metricsStoragePath = flag.String("metricsStoragePath", "/tmp/metrics", "Base path for metrics storage.")
scrapeResultsQueueCapacity = flag.Int("scrapeResultsQueueCapacity", 4096, "The size of the scrape results queue.")
ruleResultsQueueCapacity = flag.Int("ruleResultsQueueCapacity", 4096, "The size of the rule results queue.")
concurrentRetrievalAllowance = flag.Int("concurrentRetrievalAllowance", 15, "The number of concurrent metrics retrieval requests allowed.")
)
func main() {
@ -59,12 +62,13 @@ func main() {
defer persistence.Close()
scrapeResults := make(chan format.Result, 4096)
// Queue depth will need to be exposed
scrapeResults := make(chan format.Result, *scrapeResultsQueueCapacity)
targetManager := retrieval.NewTargetManager(scrapeResults, 1)
targetManager := retrieval.NewTargetManager(scrapeResults, *concurrentRetrievalAllowance)
targetManager.AddTargetsFromConfig(conf)
ruleResults := make(chan *rules.Result, 4096)
ruleResults := make(chan *rules.Result, *ruleResultsQueueCapacity)
ast.SetPersistence(persistence, nil)
ruleManager := rules.NewRuleManager(ruleResults, conf.Global.EvaluationInterval)

View file

@ -134,6 +134,10 @@ func (t *target) Scrape(earliest time.Time, results chan format.Result) (err err
done := make(chan bool)
request := func() {
defer func() {
done <- true
}()
resp, err := http.Get(t.Address())
if err != nil {
return
@ -157,8 +161,6 @@ func (t *target) Scrape(earliest time.Time, results chan format.Result) (err err
if err != nil {
return
}
done <- true
}
accumulator := func(d time.Duration) {