feat: add infinite retry mechanism for ThreadWinds registration

Author: JocLRojas

threadwinds-ingestion/main.go

@@ -11,6 +11,7 @@ import (
 	"github.com/utmstack/UTMStack/threadwinds-ingestion/config"
 	"github.com/utmstack/UTMStack/threadwinds-ingestion/internal/client"
 	"github.com/utmstack/UTMStack/threadwinds-ingestion/internal/scheduler"
+	"github.com/utmstack/UTMStack/threadwinds-ingestion/utils"
 )
 
 func main() {
@@ -57,14 +58,27 @@ func main() {
 	}
 
 	if twConfig.APIKey == "" || twConfig.APISecret == "" {
-		catcher.Info("ThreadWinds not configured, registering in platform...", nil)
+		catcher.Info("ThreadWinds not configured, will attempt registration with retry...", nil)
 
-		regResp, err := cmClient.RegisterUserReporter(adminEmail)
+		var regResp *client.RegistrationResponse
+
+		registerFunc := func() error {
+			resp, err := cmClient.RegisterUserReporter(adminEmail)
+			if err != nil {
+				return err
+			}
+			regResp = resp
+			return nil
+		}
+
+		err = utils.InfiniteRetryIfXError(registerFunc, "404", "Not Found", "connection refused")
 		if err != nil {
-			catcher.Error("failed to register in ThreadWinds Platform", err, nil)
+			catcher.Error("failed to register in ThreadWinds Platform after retries", err, nil)
 			os.Exit(1)
 		}
 
+		catcher.Info("ThreadWinds registration successful", nil)
+
 		err = backendClient.SaveThreadWindsCredentials(ctx,
 			regResp.APIKey,
 			regResp.APISecret,

threadwinds-ingestion/utils/check.go

@@ -0,0 +1,39 @@
+package utils
+
+import (
+	"strings"
+	"time"
+
+	"github.com/threatwinds/go-sdk/catcher"
+)
+
+const (
+	wait = 5 * time.Second
+)
+
+func InfiniteRetryIfXError(f func() error, exceptions ...string) error {
+	var xErrorWasLogged bool
+
+	for {
+		err := f()
+		if err != nil && is(err, exceptions...) {
+			if !xErrorWasLogged {
+				_ = catcher.Error("An error occurred (%s), will keep retrying indefinitely...", err, nil)
+				xErrorWasLogged = true
+			}
+			time.Sleep(wait)
+			continue
+		}
+
+		return err
+	}
+}
+
+func is(e error, args ...string) bool {
+	for _, arg := range args {
+		if strings.Contains(e.Error(), arg) {
+			return true
+		}
+	}
+	return false
+}