add service cmd flag and custom labels

Signed-off-by: Peekjef72 <[email protected]>

collector/collector.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/alecthomas/kingpin/v2"
 	"github.com/leoluk/perflib_exporter/perflib"
+	"github.com/prometheus-community/windows_exporter/config"
 	"github.com/prometheus-community/windows_exporter/log"
 	"github.com/prometheus/client_golang/prometheus"
 	"golang.org/x/sys/windows/registry"
@@ -58,11 +59,15 @@ type perfCounterNamesBuilder func() []string
 var (
 	builders                = make(map[string]collectorBuilder)
 	perfCounterDependencies = make(map[string]string)
+	config_hooks            = make(map[string]config.CfgHook)
 )
 
-func registerCollector(name string, builder collectorBuilder, perfCounterNames ...string) {
+func registerCollector(name string, builder collectorBuilder, hooks map[string]config.CfgHook, perfCounterNames ...string) {
 	builders[name] = builder
 	addPerfCounterDependencies(name, perfCounterNames)
+	for k, v := range hooks {
+		config_hooks[k] = v
+	}
 }
 
 func addPerfCounterDependencies(name string, perfCounterNames []string) {
@@ -80,10 +85,13 @@ func Available() []string {
 	}
 	return cs
 }
+func CfgHooks() map[string]config.CfgHook {
+	return config_hooks
+}
 func Build(collector string) (Collector, error) {
 	builder, exists := builders[collector]
 	if !exists {
-		return nil, fmt.Errorf("Unknown collector %q", collector)
+		return nil, fmt.Errorf("unknown collector %q", collector)
 	}
 	return builder()
 }

collector/init.go

@@ -1,6 +1,9 @@
 package collector
 
-import "github.com/alecthomas/kingpin/v2"
+import (
+	"github.com/alecthomas/kingpin/v2"
+	"github.com/prometheus-community/windows_exporter/config"
+)
 
 // collectorInit represents the required initialisation config for a collector.
 type collectorInit struct {
@@ -13,6 +16,8 @@ type collectorInit struct {
 	// Perflib counter names for the collector.
 	// These will be included in the Perflib scrape scope by the exporter.
 	perfCounterFunc perfCounterNamesBuilder
+	// builder function to intercept parameters for a collector
+	config_hooks map[string]config.CfgHook
 }
 
 func getDFSRCollectorDeps() []string {
@@ -307,6 +312,7 @@ var collectors = []collectorInit{
 		flags:           newServiceCollectorFlags,
 		builder:         newserviceCollector,
 		perfCounterFunc: nil,
+		config_hooks:    ServiceBuildHook(),
 	},
 	{
 		name:    "smtp",
@@ -402,6 +408,6 @@ func RegisterCollectors() {
 			perfCounterNames = v.perfCounterFunc()
 		}
 
-		registerCollector(v.name, v.builder, perfCounterNames...)
+		registerCollector(v.name, v.builder, v.config_hooks, perfCounterNames...)
 	}
 }