Rearrange kingpin
parent
5bc1573cc4
commit
d4879a5e87
|
@ -53,7 +53,8 @@ var (
|
|||
factories = make(map[string]func(config *NodeCollectorConfig, logger log.Logger) (Collector, error))
|
||||
initiatedCollectorsMtx = sync.Mutex{}
|
||||
initiatedCollectors = make(map[string]Collector)
|
||||
collectorStateGlobal = make(map[string]*bool)
|
||||
collectorStateGlobal = make(map[string]bool)
|
||||
collectorFlagState = make(map[string]*bool)
|
||||
availableCollectors = make([]string, 0)
|
||||
forcedCollectors = map[string]bool{} // collectors which have been explicitly enabled or disabled
|
||||
)
|
||||
|
@ -61,6 +62,14 @@ var (
|
|||
func GetDefaults() map[string]bool {
|
||||
defaults := make(map[string]bool)
|
||||
for k, v := range collectorStateGlobal {
|
||||
defaults[k] = v
|
||||
}
|
||||
return defaults
|
||||
}
|
||||
|
||||
func GetFlagDefaults() map[string]bool {
|
||||
defaults := make(map[string]bool)
|
||||
for k, v := range collectorFlagState {
|
||||
defaults[k] = *v
|
||||
}
|
||||
return defaults
|
||||
|
@ -84,7 +93,8 @@ func registerCollector(collector string, isDefaultEnabled bool, factory func(con
|
|||
defaultValue := fmt.Sprintf("%v", isDefaultEnabled)
|
||||
|
||||
flag := kingpin.Flag(flagName, flagHelp).Default(defaultValue).Action(collectorFlagAction(collector)).Bool()
|
||||
collectorStateGlobal[collector] = flag
|
||||
collectorStateGlobal[collector] = isDefaultEnabled
|
||||
collectorFlagState[collector] = flag
|
||||
|
||||
factories[collector] = factory
|
||||
}
|
||||
|
@ -98,9 +108,9 @@ type NodeCollector struct {
|
|||
// DisableDefaultCollectors sets the collector state to false for all collectors which
|
||||
// have not been explicitly enabled on the command line.
|
||||
func DisableDefaultCollectors() {
|
||||
for c := range collectorStateGlobal {
|
||||
for c := range collectorFlagState {
|
||||
if _, ok := forcedCollectors[c]; !ok {
|
||||
*collectorStateGlobal[c] = false
|
||||
*collectorFlagState[c] = false
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -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
|
||||
// flags).
|
||||
func (h *handler) innerHandler(filters ...string) (http.Handler, error) {
|
||||
nc, err := collector.NewNodeCollector(h.collectorConfig, collector.GetDefaults(), h.logger, filters...)
|
||||
nc, err := collector.NewNodeCollector(h.collectorConfig, collector.GetFlagDefaults(), h.logger, filters...)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("couldn't create collector: %s", err)
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue