update: fix stock reverse proxy

Makefile

@@ -1,6 +1,7 @@
 ENDPOINT=http://192.168.137.1:7860
 
-start: go run main.go
+start:
+	go run main.go
 
 haproxy-test:
 	haproxy -f ./haproxy/haproxy.cfg

proxy/middlewares.go

@@ -9,7 +9,10 @@ import (
 	"github.com/gofiber/fiber/v2/middleware/limiter"
 )
 
-// Logging Middleware
+
+/*
+ *  --- Logging Middleware ---
+*/
 func LoggingMiddleware(c *fiber.Ctx) error {
 	start := time.Now()
 	err   := c.Next()
@@ -28,7 +31,7 @@ func LoggingMiddleware(c *fiber.Ctx) error {
 func RateLimiterMiddleware() func(*fiber.Ctx) error {
 	return limiter.New(limiter.Config{
 		Max:        200,
-		Expiration: 1 * time.Minute,
+		Expiration: (1 * time.Minute),
 
 		KeyGenerator: func(c *fiber.Ctx) string {
 			return c.IP()
@@ -41,3 +44,4 @@ func RateLimiterMiddleware() func(*fiber.Ctx) error {
 		},
 	})
 }
+

proxy/stock/list_service.go

@@ -5,6 +5,7 @@ import (
 	"context"
 	"net/http"
 	"encoding/json"
+
 	proxy "tebakaja_lb_proxy/proxy"
 )
 
@@ -29,6 +30,7 @@ func (s *StockServiceImpl) StockListsService(ctx context.Context) (ApiResponse,
 			StatusCode: http.StatusInternalServerError,
 		}, err
 	}
+
 	defer resp.Body.Close()
 
 	if resp.StatusCode != http.StatusOK {

proxy/stock/prediction_handler.go

@@ -17,7 +17,7 @@ import (
  */
 func StockPredictionHandler(service StockService) fiber.Handler {
 	return func(c *fiber.Ctx) error {
-		ctx, cancel := context.WithTimeout(c.Context(), 120*time.Second)
+		ctx, cancel := context.WithTimeout(c.Context(), 120 * time.Second)
 		defer cancel()
 
 		ch := make(chan ApiResponse, 1)
@@ -55,19 +55,19 @@ func StockPredictionHandler(service StockService) fiber.Handler {
 		}()
 
 		select {
-		case apiResponse, ok := <-ch:
-			if !ok {
-				return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
-					"error": "Failed to get a response from the server",
-				})
-			}
-			return c.Status(apiResponse.StatusCode).JSON(apiResponse)
+			case apiResponse, ok := <-ch:
+				if !ok {
+					return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
+						"error": "Failed to get a response from the server",
+					})
+				}
+				return c.Status(apiResponse.StatusCode).JSON(apiResponse)
 
-		case <-ctx.Done():
-			log.Printf("[%s] Timeout: %v", time.Now().Format("2006-01-02 15:04:05"), ctx.Err())
-			return c.Status(http.StatusRequestTimeout).JSON(fiber.Map{
-				"error": "Request timeout",
-			})
+			case <-ctx.Done():
+				log.Printf("[%s] Timeout: %v", time.Now().Format("2006-01-02 15:04:05"), ctx.Err())
+				return c.Status(http.StatusRequestTimeout).JSON(fiber.Map{
+					"error": "Request timeout",
+				})
 		}
 	}
 }

proxy/stock/prediction_service.go

@@ -6,6 +6,7 @@ import (
 	"context"
 	"net/http"
 	"encoding/json"
+
 	proxy "tebakaja_lb_proxy/proxy"
 )