Adding initial retry layer version (#14954)

* Adding initial retry layer version

* Some simplification around the generated code

* Generating retry layer again

* Improving naming generation in store generated layers

* Address PR review comments

* Updating store layers

* Addressing PR review comments

* fixing lint errors

* Updating store layers

* Adding license header

* Applying the retry layer to the reaction_store

* Regenerating retry layer

store/layer_generators/main.go

@@ -28,6 +28,10 @@ func isError(typeName string) bool {
 	return strings.Contains(typeName, APP_ERROR_TYPE) || strings.Contains(typeName, ERROR_TYPE)
 }
 
+func isAppError(typeName string) bool {
+	return strings.Contains(typeName, APP_ERROR_TYPE)
+}
+
 func main() {
 	if err := buildTimerLayer(); err != nil {
 		log.Fatal(err)
@@ -35,6 +39,22 @@ func main() {
 	if err := buildOpenTracingLayer(); err != nil {
 		log.Fatal(err)
 	}
+	if err := buildRetryLayer(); err != nil {
+		log.Fatal(err)
+	}
+}
+
+func buildRetryLayer() error {
+	code, err := generateLayer("RetryLayer", "retry_layer.go.tmpl")
+	if err != nil {
+		return err
+	}
+	formatedCode, err := format.Source(code)
+	if err != nil {
+		return err
+	}
+
+	return ioutil.WriteFile(path.Join("retrylayer/retrylayer.go"), formatedCode, 0644)
 }
 
 func buildTimerLayer() error {
@@ -213,19 +233,33 @@ func generateLayer(name, templateFile string) ([]byte, error) {
 		},
 		"genResultsVars": func(results []string) string {
 			vars := []string{}
-			for i := range results {
-				vars = append(vars, fmt.Sprintf("resultVar%d", i))
+			for i, typeName := range results {
+				if isError(typeName) {
+					vars = append(vars, "err")
+				} else if i == 0 {
+					vars = append(vars, "result")
+				} else {
+					vars = append(vars, fmt.Sprintf("resultVar%d", i))
+				}
 			}
 			return strings.Join(vars, ", ")
 		},
 		"errorToBoolean": func(results []string) string {
-			for i, typeName := range results {
+			for _, typeName := range results {
 				if isError(typeName) {
-					return fmt.Sprintf("resultVar%d == nil", i)
+					return fmt.Sprintf("err == nil")
 				}
 			}
 			return "true"
 		},
+		"isAppError": func(results []string) bool {
+			for _, typeName := range results {
+				if isAppError(typeName) {
+					return true
+				}
+			}
+			return false
+		},
 		"errorPresent": func(results []string) bool {
 			for _, typeName := range results {
 				if isError(typeName) {
@@ -235,9 +269,9 @@ func generateLayer(name, templateFile string) ([]byte, error) {
 			return false
 		},
 		"errorVar": func(results []string) string {
-			for i, typeName := range results {
+			for _, typeName := range results {
 				if isError(typeName) {
-					return fmt.Sprintf("resultVar%d", i)
+					return "err"
 				}
 			}
 			return ""
@@ -262,6 +296,19 @@ func generateLayer(name, templateFile string) ([]byte, error) {
 			}
 			return strings.Join(paramsWithType, ", ")
 		},
+		"joinParamsWithTypeOutsideStore": func(params []methodParam) string {
+			paramsWithType := []string{}
+			for _, param := range params {
+				if param.Type == "ChannelSearchOpts" || param.Type == "UserGetByIdsOpts" {
+					paramsWithType = append(paramsWithType, fmt.Sprintf("%s store.%s", param.Name, param.Type))
+				} else if param.Type == "*UserGetByIdsOpts" {
+					paramsWithType = append(paramsWithType, fmt.Sprintf("%s *store.UserGetByIdsOpts", param.Name))
+				} else {
+					paramsWithType = append(paramsWithType, fmt.Sprintf("%s %s", param.Name, param.Type))
+				}
+			}
+			return strings.Join(paramsWithType, ", ")
+		},
 	}
 
 	t := template.Must(template.New(templateFile).Funcs(myFuncs).ParseFiles("layer_generators/" + templateFile))