dkforest

commit 666fe2df4de727ca2e9aee07b4556c469eda8ca8
parent 59435ea5abccc88338d1f1732f93305fff5afde0
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Sun, 18 Dec 2022 13:52:36 -0800

simplify code

Diffstat:
M
pkg/web/handlers/handlers.go
 | 
11
++++++-----

1 file changed, 6 insertions(+), 5 deletions(-)
diff --git a/pkg/web/handlers/handlers.go b/pkg/web/handlers/handlers.go
@@ -773,7 +773,8 @@ type WaitPageCookiePayload struct {
 }
 
 func waitPageWrapper(c echo.Context, clb echo.HandlerFunc, cookieName string) error {
-	start := time.Now().UnixNano()
+	now := time.Now()
+	start := now.UnixNano()
 	var waitToken string
 
 	if cc, payload, err := hutils.EncCookie[WaitPageCookiePayload](c, cookieName); err != nil {
@@ -783,8 +784,8 @@ func waitPageWrapper(c echo.Context, clb echo.HandlerFunc, cookieName string) er
 		payload := WaitPageCookiePayload{
 			Token: waitToken,
 			Count: 1,
-			Now:   time.Now().UnixMilli(),
-			Unix:  time.Now().Unix() + waitTime - 1, // unix time at which the wait time is over
+			Now:   now.UnixMilli(),
+			Unix:  now.Unix() + waitTime - 1, // unix time at which the wait time is over
 		}
 		c.SetCookie(hutils.CreateEncCookie(cookieName, payload, utils.OneMinuteSecs*5))
 
@@ -801,7 +802,7 @@ func waitPageWrapper(c echo.Context, clb echo.HandlerFunc, cookieName string) er
 		start = payload.Now
 		if c.Request().Method == http.MethodGet {
 			// If you reload the page before the wait time is over, we kill the circuit.
-			if time.Now().Unix() < payload.Unix {
+			if now.Unix() < payload.Unix {
 				// Kill circuit
 				if conn, ok := c.Request().Context().Value("conn").(net.Conn); ok {
 					config.ConnMap.Close(conn)
@@ -815,7 +816,7 @@ func waitPageWrapper(c echo.Context, clb echo.HandlerFunc, cookieName string) er
 				return c.String(http.StatusFound, "You tried to reload the page too many times. Now you have to wait one minute.")
 			}
 			payload.Count++
-			payload.Now = time.Now().UnixMilli()
+			payload.Now = now.UnixMilli()
 			c.SetCookie(hutils.CreateEncCookie(cookieName, payload, utils.OneMinuteSecs*5))
 		}
 	}