Merge pull request #9 from samcm/feat/eth-modules

feat(args): Add execution-modules argument

cmd/root.go

@@ -34,6 +34,7 @@ var (
 	executionUrl         string
 	consensusUrl         string
 	monitoredDirectories []string
+	executionModules     []string
 )
 
 // Execute adds all child commands to the root command and sets flags appropriately.
@@ -57,6 +58,7 @@ func init() {
 	rootCmd.PersistentFlags().StringVarP(&executionUrl, "execution-url", "", "", "(optional) URL to the execution node")
 	rootCmd.PersistentFlags().StringVarP(&consensusUrl, "consensus-url", "", "", "(optional) URL to the consensus node")
 	rootCmd.PersistentFlags().StringSliceVarP(&monitoredDirectories, "monitored-directories", "", []string{}, "(optional) directories to monitor for disk usage")
+	rootCmd.PersistentFlags().StringSliceVarP(&executionModules, "execution-modules", "", []string{}, "(optional) execution modules that are enabled on the node")
 
 	// Cobra also supports local flags, which will only run
 	// when this action is called directly.
@@ -108,6 +110,10 @@ func initCommon() {
 		config.DiskUsage.Directories = monitoredDirectories
 	}
 
+	if len(executionModules) > 0 {
+		config.Execution.Modules = executionModules
+	}
+
 	export = exporter.NewExporter(log, config)
 	if err := export.Init(ctx); err != nil {
 		logrus.Fatal(err)

pkg/exporter/disk/metrics.go

@@ -22,7 +22,7 @@ func NewMetrics(log logrus.FieldLogger, namespace string) Metrics {
 		diskUsage: prometheus.NewGaugeVec(
 			prometheus.GaugeOpts{
 				Namespace:   namespace,
-				Name:        "disk_usage",
+				Name:        "usage_bytes",
 				Help:        "How large the directory is (in bytes).",
 				ConstLabels: constLabels,
 			},