mudhorn/node_exporter
1
0
Fork 0
mirror of https://github.com/prometheus/node_exporter.git synced 2025-03-05 21:00:12 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Allow the passing in of enabled collectors.

Browse source
This commit is contained in:
matt durham 2023-10-03 09:30:15 -04:00
parent e78bc3045a
commit 5bc1573cc4
No known key found for this signature in database
GPG key ID: A62E920AE398897B
2 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
collector/collector.go
View file

@ -58,7 +58,7 @@ var (
forcedCollectors = map[string]bool{} // collectors which have been explicitly enabled or disabled forcedCollectors = map[string]bool{} // collectors which have been explicitly enabled or disabled
) )
func getDefaults() map[string]bool { func GetDefaults() map[string]bool {
defaults := make(map[string]bool) defaults := make(map[string]bool)
for k, v := range collectorStateGlobal { for k, v := range collectorStateGlobal {
defaults[k] = *v defaults[k] = *v
@ -118,11 +118,10 @@ func collectorFlagAction(collector string) func(ctx *kingpin.ParseContext) error
} }
// NewNodeCollector creates a new NodeCollector. // NewNodeCollector creates a new NodeCollector.
func NewNodeCollector(config *NodeCollectorConfig, logger log.Logger, filters ...string) (*NodeCollector, error) { func NewNodeCollector(config *NodeCollectorConfig, enabledCollectors map[string]bool, logger log.Logger, filters ...string) (*NodeCollector, error) {
f := make(map[string]bool) f := make(map[string]bool)
defaults := getDefaults()
for _, filter := range filters { for _, filter := range filters {
enabled, exist := defaults[filter] enabled, exist := enabledCollectors[filter]
if !exist { if !exist {
return nil, fmt.Errorf("missing collector: %s", filter) return nil, fmt.Errorf("missing collector: %s", filter)
} }
@ -134,7 +133,7 @@ func NewNodeCollector(config *NodeCollectorConfig, logger log.Logger, filters ..
collectors := make(map[string]Collector) collectors := make(map[string]Collector)
initiatedCollectorsMtx.Lock() initiatedCollectorsMtx.Lock()
defer initiatedCollectorsMtx.Unlock() defer initiatedCollectorsMtx.Unlock()
for key, enabled := range defaults { for key, enabled := range enabledCollectors {
if !enabled || (len(f) > 0 && !f[key]) { if !enabled || (len(f) > 0 && !f[key]) {
continue continue
} }

2
node_exporter.go
View file

@ -102,7 +102,7 @@ func (h *handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
// (in which case it will log all the collectors enabled via command-line // (in which case it will log all the collectors enabled via command-line
// flags). // flags).
func (h *handler) innerHandler(filters ...string) (http.Handler, error) { func (h *handler) innerHandler(filters ...string) (http.Handler, error) {
nc, err := collector.NewNodeCollector(h.collectorConfig, h.logger, filters...) nc, err := collector.NewNodeCollector(h.collectorConfig, collector.GetDefaults(), h.logger, filters...)
if err != nil { if err != nil {
return nil, fmt.Errorf("couldn't create collector: %s", err) return nil, fmt.Errorf("couldn't create collector: %s", err)
} }