Implement a retry mechanism in case of very slow response from the htmx routes

1 file · +11 -4

M internal/httpserve/middleware/requirelogin.go +11 -4


@@ -15,14 +15,21 @@ import (
15	15		func RequireLogin(a *server.App) echo.MiddlewareFunc {
16	16		return func(next echo.HandlerFunc) echo.HandlerFunc {
17	17		return func(c echo.Context) error {
18		-	if err := validateSessionAndUser(c, a); err != nil {
19		-	return c.JSON(401, map[string]string{"error": err.Error()})
	18	+	const maxRetries = 3
	19	+	var err error
	20	+	for i := 0; i < maxRetries; i++ {
	21	+	err = validateSessionAndUser(c, a)
	22	+	if err == nil {
	23	+	return next(c)
	24	+	}
	25	+	if i < maxRetries-1 {
	26	+	time.Sleep(time.Duration(i+1) * 500 * time.Millisecond)
	27	+	}
20	28		}
21		-	return next(c)
	29	+	return c.JSON(401, map[string]string{"error": err.Error()})
22	30		}
23	31		}
24	32		}
25		-
26	33		func validateSessionAndUser(c echo.Context, a *server.App) error {
27	34		sessionExpiryTime := 30 * time.Minute
28	35

 func RequireLogin(a *server.App) echo.MiddlewareFunc {
 	return func(next echo.HandlerFunc) echo.HandlerFunc {
 		return func(c echo.Context) error {
-			if err := validateSessionAndUser(c, a); err != nil {
+			const maxRetries = 3
-				return c.JSON(401, map[string]string{"error": err.Error()})
+			var err error
+			for i := 0; i < maxRetries; i++ {
+				err = validateSessionAndUser(c, a)
+				if err == nil {
+					return next(c)
+				}
+				if i < maxRetries-1 {
+					time.Sleep(time.Duration(i+1) * 500 * time.Millisecond)
+				}
 			}
-			return next(c)
+			return c.JSON(401, map[string]string{"error": err.Error()})
 		}
 	}
 }
 func validateSessionAndUser(c echo.Context, a *server.App) error {
 	sessionExpiryTime := 30 * time.Minute