[receiver/purefa] Adds a new scrapper type directories (open-telemetr…

…y#16590)
frzifus · Dec 5, 2022 · b0de453 · b0de453
1 parent 761def3
commit b0de453
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/receiver/purefareceiver/config.go b/receiver/purefareceiver/config.go
@@ -39,15 +39,19 @@ type Config struct {
 
  // Hosts represents the list of hosts to query
  Hosts []internal.ScraperConfig `mapstructure:"hosts"`
+
+ // Directories represents the list of directories to query
+ Directories []internal.ScraperConfig `mapstructure:"directories"`
 }
 
 type Settings struct {
  ReloadIntervals *ReloadIntervals `mapstructure:"reload_intervals"`
 }
 
 type ReloadIntervals struct {
- Array time.Duration `mapstructure:"array"`
- Host time.Duration `mapstructure:"host"`
+ Array time.Duration `mapstructure:"array"`
+ Host time.Duration `mapstructure:"host"`
+ Directories time.Duration `mapstructure:"directories"`
 }
 
 func (c *Config) Validate() error {

diff --git a/receiver/purefareceiver/internal/scraper.go b/receiver/purefareceiver/internal/scraper.go
@@ -34,8 +34,9 @@ type Scraper interface {
 type ScraperType string
 
 const (
- ScraperTypeArray ScraperType = "array"
- ScraperTypeHost ScraperType = "host"
+ ScraperTypeArray ScraperType = "array"
+ ScraperTypeHost ScraperType = "host"
+ ScraperTypeDirectories ScraperType = "directories"
 )
 
 type scraper struct {

diff --git a/receiver/purefareceiver/receiver.go b/receiver/purefareceiver/receiver.go
@@ -61,6 +61,13 @@ func (r *purefaReceiver) Start(ctx context.Context, compHost component.Host) err
  return err
  }
 
+ directoriesScraper := internal.NewScraper(ctx, internal.ScraperTypeDirectories, r.cfg.Endpoint, r.cfg.Hosts, r.cfg.Settings.ReloadIntervals.Host)
+ if scCfgs, err := directoriesScraper.ToPrometheusReceiverConfig(compHost, fact); err == nil {
+ scrapeCfgs = append(scrapeCfgs, scCfgs...)
+ } else {
+ return err
+ }
+
  promRecvCfg := fact.CreateDefaultConfig().(*prometheusreceiver.Config)
  promRecvCfg.PrometheusConfig = &config.Config{ScrapeConfigs: scrapeCfgs}