modelRT/handler/real_time_data_query.go

// Package handler provides HTTP handlers for various endpoints.
package handler

import (
	"fmt"
	"net/http"
	"strconv"

	"modelRT/alert"
	"modelRT/constants"
	"modelRT/logger"
	"modelRT/network"

	"github.com/gin-gonic/gin"
)

// QueryRealTimeDataHandler define query real time data process API
func QueryRealTimeDataHandler(c *gin.Context) {
	token := c.Query("token")
	beginStr := c.Query("begin")
	begin, err := strconv.Atoi(beginStr)
	if err != nil {
		logger.Error(c, "convert begin param from string to int failed", "error", err)

		resp := network.FailureResponse{
			Code: http.StatusBadRequest,
			Msg:  err.Error(),
		}
		c.JSON(http.StatusOK, resp)
	}
	endStr := c.Query("end")
	end, err := strconv.Atoi(endStr)
	if err != nil {
		logger.Error(c, "convert end param from string to int failed", "error", err)

		resp := network.FailureResponse{
			Code: http.StatusBadRequest,
			Msg:  err.Error(),
		}
		c.JSON(http.StatusOK, resp)
	}

	fmt.Println(token, begin, end)
	// TODO parse token to dataRT query params
	var level int
	var targetLevel constants.AlertLevel
	alertManger := alert.GetAlertMangerInstance()
	targetLevel = constants.AlertLevel(level)
	events := alertManger.GetRangeEventsByLevel(targetLevel)

	resp := network.SuccessResponse{
		Code: http.StatusOK,
		Msg:  "success",
		PayLoad: map[string]interface{}{
			"events": events,
		},
	}
	c.JSON(http.StatusOK, resp)
}