Add StandardGatherers utility to retrieve all the wired up gatherers

trento-project · Sep 27, 2022 · 1b71c7e · 1b71c7e
1 parent 22e9059
commit 1b71c7e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/cmd/facts.go b/cmd/facts.go
@@ -81,9 +81,7 @@ func gather(*cobra.Command, []string) {
 	var argument = viper.GetString("argument")
 	var pluginsFolder = viper.GetString("plugins-folder")
 
-	gathererManager := gatherers.NewManager(map[string]gatherers.FactGatherer{
-		gatherers.CorosyncFactKey: gatherers.NewDefaultCorosyncConfGatherer(),
-	})
+	gathererManager := gatherers.NewManager(gatherers.StandardGatherers())
 
 	log.Info("loading plugins")
 
@@ -137,9 +135,7 @@ func cleanupAndFatal(err error) {
 func list(*cobra.Command, []string) {
 	var pluginsFolder = viper.GetString("plugins-folder")
 
-	gathererManager := gatherers.NewManager(map[string]gatherers.FactGatherer{
-		gatherers.CorosyncFactKey: gatherers.NewDefaultCorosyncConfGatherer(),
-	})
+	gathererManager := gatherers.NewManager(gatherers.StandardGatherers())
 
 	log.Info("loading plugins")
 

diff --git a/internal/agent.go b/internal/agent.go
@@ -103,9 +103,7 @@ func (a *Agent) Start(ctx context.Context) error {
 
 	if a.config.FactsEngineEnabled {
 
-		gathererManager := gatherers.NewManager(map[string]gatherers.FactGatherer{
-			gatherers.CorosyncFactKey: gatherers.NewDefaultCorosyncConfGatherer(),
-		})
+		gathererManager := gatherers.NewManager(gatherers.StandardGatherers())
 
 		log.Info("loading plugins")
 

diff --git a/internal/factsengine/gatherers/gatherer.go b/internal/factsengine/gatherers/gatherer.go
@@ -5,3 +5,9 @@ import "github.com/trento-project/agent/internal/factsengine/entities"
 type FactGatherer interface {
 	Gather(factsRequests []entities.FactRequest) ([]entities.Fact, error)
 }
+
+func StandardGatherers() map[string]FactGatherer {
+	return map[string]FactGatherer{
+		CorosyncFactKey: NewDefaultCorosyncConfGatherer(),
+	}
+}