Merge pull request #20 from LambdaTest/MLE-17521-new

Mle 17521 new

Author: chandrajeetn

localEvaluation/localEvaluation.go

@@ -19,6 +19,7 @@ var (
 	LocalEvaluationConfigPollInterval         = 120
 	LocalEvaluationConfigPollerRequestTimeout = 60
 	LocalEvaluationDeploymentKey              = "server-jAqqJaX3l8PgNiJpcv9j20ywPzANQQFh"
+	retries                                   = 5
 )
 
 type Variant struct {
@@ -42,7 +43,7 @@ type UserProperties struct {
 	TemplateId       string `json:"template_id,omitempty"`
 }
 
-func init() {
+func Init() {
 	err := godotenv.Load()
 	if err != nil {
 		fmt.Printf("No .env file found")
@@ -68,14 +69,24 @@ func init() {
 }
 
 func Initialize() {
+	Init()
 	config := local.Config{
 		Debug:                          LocalEvaluationConfigDebug,
 		ServerUrl:                      LocalEvaluationConfigServerUrl,
 		FlagConfigPollerInterval:       time.Duration(LocalEvaluationConfigPollInterval) * time.Second,
 		FlagConfigPollerRequestTimeout: time.Duration(LocalEvaluationConfigPollerRequestTimeout) * time.Second,
 	}
 	client = local.Initialize(LocalEvaluationDeploymentKey, &config)
-	err := client.Start()
+	var err error
+	for i := 0; i < retries; i++ {
+		err = client.Start()
+		if err != nil {
+			err = fmt.Errorf("unable to create local evaluation client with given config %+v attempt:%v with error %s", config, i+1, err.Error())
+			continue
+		} else {
+			break
+		}
+	}
 	if err != nil {
 		err = fmt.Errorf("unable to create local evaluation client with given config %+v with error %s", config, err.Error())
 		panic(err)
@@ -84,7 +95,16 @@ func Initialize() {
 
 func InitializeWithConfig(conf local.Config, deploymentKey string) {
 	client = local.Initialize(deploymentKey, &conf)
-	err := client.Start()
+	var err error
+	for i := 0; i < retries; i++ {
+		err = client.Start()
+		if err != nil {
+			err = fmt.Errorf("unable to create local evaluation client with given config %+v attempt:%v with error %s", conf, i+1, err.Error())
+			continue
+		} else {
+			break
+		}
+	}
 	if err != nil {
 		err = fmt.Errorf("unable to create local evaluation client with given config %+v with error %s", conf, err.Error())
 		panic(err)