set retry for any non 2xx response

marigold-dev · Dec 20, 2023 · cbfcbe7 · cbfcbe7
1 parent 4e45d8c
commit cbfcbe7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/middlewares/retry.go b/middlewares/retry.go
@@ -1,6 +1,7 @@
 package middlewares
 
 import (
+	"strconv"
 	"strings"
 
 	"github.com/labstack/echo/v4"
@@ -15,15 +16,13 @@ func Retry(config *utils.Config) echo.MiddlewareFunc {
 			}
 			err = next(c)
 
-			status := c.Response().Status
-			if status == 404 || status == 410 &&
-				strings.Contains(c.Request().URL.Path, "blocks") {
+			status := strconv.Itoa(c.Response().Status)
+			if !strings.HasPrefix(status, "2") {
 				c.Set("retry", true)
 				return next(c)
-			} else {
-				c.Set("retry", false)
 			}
 
+			c.Set("retry", false)
 			return err
 		}
 	}

diff --git a/utils/config.go b/utils/config.go
@@ -256,7 +256,7 @@ func NewConfig() *Config {
 		store = &memoryStore
 	}
 
-	balancer := balancers.NewSameNodeBalancer(targets, retryTarget, configFile.LoadBalancer.TTL, store)
+	balancer := balancers.NewIPHashBalancer(targets, retryTarget, configFile.LoadBalancer.TTL, store)
 
 	transport := &http.Transport{
 		Proxy: http.ProxyFromEnvironment,