Merge pull request vpenso#50 from vpenso/gpus_acct

Enable GPUs accounting only via cmd line option (see vpenso#45)

Author: mtds

main.go

@@ -27,7 +27,6 @@ func init() {
 	// Metrics have to be registered to be exposed
 	prometheus.MustRegister(NewAccountsCollector())       // from accounts.go
 	prometheus.MustRegister(NewCPUsCollector())           // from cpus.go
-	prometheus.MustRegister(NewGPUsCollector())           // from gpus.go
 	prometheus.MustRegister(NewNodesCollector())          // from nodes.go
 	prometheus.MustRegister(NewNodeCollector())           // from node.go
 	prometheus.MustRegister(NewPartitionsCollector())     // from partitions.go
@@ -42,8 +41,19 @@ var listenAddress = flag.String(
 	":8080",
 	"The address to listen on for HTTP requests.")
 
+var gpuAcct = flag.Bool(
+	"gpus-acct",
+	false,
+	"Enable GPUs accounting")
+
 func main() {
 	flag.Parse()
+
+	// Turn on GPUs accounting only if the corresponding command line option is set to true.
+	if *gpuAcct {
+		prometheus.MustRegister(NewGPUsCollector())   // from gpus.go
+	}
+
 	// The Handler function provides a default handler to expose metrics
 	// via an HTTP server. "/metrics" is the usual endpoint for that.
 	log.Infof("Starting Server: %s", *listenAddress)