Allow fetch multiple queues in a batch

gin/handlers/queue.go

@@ -3,6 +3,7 @@ package handlers
 import (
 	"github.com/gin-gonic/gin"
 	"net/http"
+	"strings"
 	"vatprc-queue/gin/errors"
 	"vatprc-queue/gin/services"
 )
@@ -94,3 +95,29 @@ func GetQueueHandler(c *gin.Context) (interface{}, error) {
 	}
 	return services.GetQueueResult(airport, true), nil
 }
+
+func GetMultipleQueuesHandler(c *gin.Context) (interface{}, error) {
+	param, ok := c.Params.Get("airports")
+	if !ok {
+		return nil, errors.ApiError{
+			Status:           http.StatusBadRequest,
+			Code:             http.StatusBadRequest,
+			ShowInProduction: true,
+			Message:          "airports parameter is required",
+		}
+	}
+	airports := strings.Split(param, ",")
+	if len(airports) < 1 || len(airports) > 10 {
+		return nil, errors.ApiError{
+			Status:           http.StatusBadRequest,
+			Code:             http.StatusBadRequest,
+			ShowInProduction: true,
+			Message:          "enter 1 to 10 airports",
+		}
+	}
+	result := make(map[string][]services.QueueResult)
+	for _, airport := range airports {
+		result[airport] = services.GetQueueResult(airport, true)
+	}
+	return result, nil
+}

gin/router/routes.go

@@ -13,6 +13,7 @@ func InitRouter() *gin.Engine {
 	v1 := router.Group("v1")
 	{
 		v1.GET("/hello", errors.ErrorWrapper(handlers.HelloWorld))
+		v1.GET("/queue", errors.ErrorWrapper(handlers.GetMultipleQueuesHandler))
 		airport := v1.Group("/:airport")
 		{
 			airport.PATCH("/status", errors.ErrorWrapper(handlers.UpdateStatus))