chore: add some logging

steadybit · Jul 5, 2024 · 15dff94 · 15dff94
1 parent 079c3d0
commit 15dff94
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 2 deletions.
diff --git a/extloadtest/discovery.go b/extloadtest/discovery.go
@@ -3,6 +3,7 @@ package extloadtest
 import (
 	"context"
 	"fmt"
+	"github.com/rs/zerolog/log"
 	"github.com/steadybit/action-kit/go/action_kit_api/v2"
 	"github.com/steadybit/action-kit/go/action_kit_sdk"
 	"github.com/steadybit/discovery-kit/go/discovery_kit_api"
@@ -78,6 +79,39 @@ func NewTargetData() *TargetData {
 		azureKubernetesNodes = createKubernetesNodeTargets(azureKubernetesContainers)
 		azureContainers = createContainerTargets(azureKubernetesContainers)
 	}
+	// count generated targets
+	log.Info().Msgf("Generated %d hosts", len(ec2Hosts)+len(gcpHosts)+len(azureHosts))
+	log.Info().Msgf("Generated %d ec2 instances", len(ec2Instances))
+	log.Info().Msgf("Generated %d gcp instances", len(gcpInstances))
+	log.Info().Msgf("Generated %d azure instances", len(azureInstances))
+	log.Info().Msgf("Generated %d kubernetes clusters", len(kubernetesClusters))
+	log.Info().Msgf("Generated %d kubernetes deployments", len(ec2KubernetesDeployments)+len(gcpKubernetesDeployments)+len(azureKubernetesDeployments))
+	log.Info().Msgf("Generated %d kubernetes pods", len(ec2KubernetesPods)+len(gcpKubernetesPods)+len(azureKubernetesPods))
+	log.Info().Msgf("Generated %d kubernetes containers", len(ec2KubernetesContainers)+len(gcpKubernetesContainers)+len(azureKubernetesContainers))
+	log.Info().Msgf("Generated %d kubernetes nodes", len(ec2KubernetesNodes)+len(gcpKubernetesNodes)+len(azureKubernetesNodes))
+	log.Info().Msgf("Generated %d containers", len(ec2Containers)+len(gcpContainers)+len(azureContainers))
+
+	var targetsAvailable = 0;
+	if !config.Config.DisableHostDiscovery {
+		targetsAvailable += len(ec2Hosts) + len(gcpHosts) + len(azureHosts)
+	}
+	if !config.Config.DisableAWSDiscovery {
+		targetsAvailable += len(ec2Instances)
+	}
+	if !config.Config.DisableGCPDiscovery {
+		targetsAvailable += len(gcpInstances)
+	}
+	if !config.Config.DisableAzureDiscovery {
+		targetsAvailable += len(azureInstances)
+	}
+	if !config.Config.DisableKubernetesDiscovery {
+		targetsAvailable += len(kubernetesClusters) + len(ec2KubernetesDeployments) + len(gcpKubernetesDeployments) + len(azureKubernetesDeployments) + len(ec2KubernetesPods) + len(gcpKubernetesPods) + len(azureKubernetesPods) + len(ec2KubernetesContainers) + len(gcpKubernetesContainers) + len(azureKubernetesContainers) + len(ec2KubernetesNodes) + len(gcpKubernetesNodes) + len(azureKubernetesNodes)
+	}
+	if !config.Config.DisableContainerDiscovery {
+		targetsAvailable += len(ec2Containers) + len(gcpContainers) + len(azureContainers)
+	}
+	log.Info().Msgf("Total targets available: %d", targetsAvailable)
+
 	return &TargetData{
 		hosts:                 append(append(ec2Hosts, gcpHosts...), azureHosts...),
 		ec2Instances:          ec2Instances,

diff --git a/extloadtest/discovery_kubernetes_node.go b/extloadtest/discovery_kubernetes_node.go
@@ -1,7 +1,6 @@
 package extloadtest
 
 import (
-	"fmt"
 	"github.com/steadybit/discovery-kit/go/discovery_kit_api"
 	"github.com/steadybit/discovery-kit/go/discovery_kit_commons"
 	"github.com/steadybit/extension-kit/extutil"
@@ -49,7 +48,7 @@ func createKubernetesNodeTargets(containerTargets []discovery_kit_api.Enrichment
 
 	var result = make([]discovery_kit_api.Target, 0, len(data))
 	for hostName, nodeData := range data {
-		fmt.Println(hostName, nodeData)
+		//fmt.Println(hostName, nodeData)
 
 		target := discovery_kit_api.Target{
 			Id:         hostName,