From 910a9849f706261d823c3fbd6bbc3862aeb827c6 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Fri, 26 Jan 2024 20:42:03 +0800
Subject: [PATCH] :bug: Fix timeout issue

---
 pkg/navi/responder.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/pkg/navi/responder.go b/pkg/navi/responder.go
index b6b34a6..a4870f0 100644
--- a/pkg/navi/responder.go
+++ b/pkg/navi/responder.go
@@ -31,7 +31,14 @@ func makeUnifiedResponse(c *fiber.Ctx, dest *Destination) error {
 }
 
 func makeHypertextResponse(c *fiber.Ctx, dest *Destination) error {
-	limit := 60 * time.Millisecond
+	_, queries := dest.GetRawUri()
+	raw := lo.Ternary(len(queries.Get("timeout")) > 0, queries.Get("timeout"), "5000")
+	num, err := strconv.Atoi(raw)
+	if err != nil {
+		num = 5000
+	}
+
+	limit := time.Duration(num) * time.Millisecond
 	uri := dest.MakeUri(c)
 	return proxy.Do(c, uri, &fasthttp.Client{
 		ReadTimeout:  limit,