proxy/stock/list_handler.go

package stock

import (
	"log"
	"sync"
	"time"
	"context"
	"net/http"

	"github.com/gofiber/fiber/v2"
)



// StockListsHandler   godoc
// @Summary            Stock List
// @Description        Stock List
// @Tags               Stock
// @Produce            json
// @Success            200  {object} ApiResponse
// @Router             /stock/lists [get]
func StockListsHandler(service StockService) fiber.Handler {
	return func(c *fiber.Ctx) error {
		ctx, cancel := context.WithTimeout(c.Context(), 120*time.Second)
		defer cancel()

		ch := make(chan ApiResponse, 1)
		var wg sync.WaitGroup
		wg.Add(1)

		go func() {
			defer wg.Done()

			apiResponse, err := service.StockListsService(ctx)
			if err != nil {
				log.Printf("[%s] %v", time.Now().Format("2006-01-02 15:04:05"), err)
				ch <- apiResponse
				return
			}

			ch <- apiResponse
		}()

		go func() {
			wg.Wait()
			close(ch)
		}()

		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 <-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",
				})
		}
	}
}