feat: improve registration of checks by config

pkg/checks/checks.go

@@ -6,6 +6,13 @@ import (
 	"time"
 )
 
+// Available Checks will be registered in this map
+// The key is the name of the Check
+// The name needs to map the configuration item key
+var RegisteredChecks = map[string]func(string) Check{
+	"rtt": GetRoundtripCheck,
+}
+
 type Check interface {
 	// Run is called once per check interval
 	// this should error if there is a problem running the check

pkg/checks/roundtrip.go

@@ -18,6 +18,13 @@ type RoundTrip struct {
 	config roundTripConfig
 }
 
+// Constructor for the RoundtripCheck
+func GetRoundtripCheck(name string) Check {
+	return &RoundTrip{
+		name: name,
+	}
+}
+
 func (rt *RoundTrip) Run(ctx context.Context) (Result, error) {
 	return Result{}, nil
 }

pkg/sparrow/run.go

@@ -17,7 +17,6 @@ type Sparrow struct {
 func New(config *Config) *Sparrow {
 	// TODO read this from config file
 	return &Sparrow{
-
 		config: config,
 		c:      make(chan checks.Result),
 	}
@@ -41,15 +40,10 @@ func (s *Sparrow) Run(ctx context.Context) error {
 		}
 	}()
 
-	for k, checkConfig := range s.config.Checks {
-		var check checks.Check
-		switch k {
-		case "rtt":
-			check = &checks.RoundTrip{}
-		default:
-			continue
-		}
+	for checkName, checkConfig := range s.config.Checks {
+		check := checks.RegisteredChecks[checkName](checkName)
 		s.checks = append(s.checks, check)
+
 		err := check.SetConfig(ctx, checkConfig)
 		if err != nil {
 			return fmt.Errorf("failed to set config for check %s: %w", check.Name(), err)