87 lines
2.5 KiB
Go
87 lines
2.5 KiB
Go
// Package handler provides HTTP handlers for various endpoints.
|
||
package handler
|
||
|
||
import (
|
||
"net/http"
|
||
|
||
"modelRT/database"
|
||
"modelRT/diagram"
|
||
"modelRT/logger"
|
||
"modelRT/network"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// MeasurementGetHandler retrieves the value of measurement data
|
||
func MeasurementGetHandler(c *gin.Context) {
|
||
var request network.MeasurementGetRequest
|
||
var clientToken string
|
||
|
||
// TODO 优化client_token获取
|
||
token, ok := c.Get("client_token")
|
||
if ok {
|
||
// 将 token 断言为 ClientToken 类型
|
||
clientToken, isClientToken := token.(string)
|
||
if !isClientToken {
|
||
// TODO 优化相关报错输出
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"message": "Request processed successfully",
|
||
"token": clientToken,
|
||
})
|
||
return
|
||
}
|
||
clientToken = token.(string)
|
||
}
|
||
|
||
if err := c.ShouldBindJSON(&request); err != nil {
|
||
logger.Error(c, "Failed to unmarshal measurement get request", "error", err)
|
||
c.JSON(http.StatusOK, network.FailureResponse{
|
||
Code: http.StatusBadRequest,
|
||
Msg: err.Error(),
|
||
})
|
||
return
|
||
}
|
||
// token 当前客户端的唯一标识（token）,用于区分不同的客户端。
|
||
zset := diagram.NewRedisZSet(c, request.MeasurementToken, clientToken, 0, false)
|
||
points, err := zset.ZRANGE(request.MeasurementToken, 0, -1)
|
||
if err != nil {
|
||
logger.Error(c, "failed to get measurement data from redis", "measurement_token", request.MeasurementToken, "error", err)
|
||
c.JSON(http.StatusOK, network.FailureResponse{
|
||
Code: http.StatusInternalServerError,
|
||
Msg: err.Error(),
|
||
PayLoad: map[string]interface{}{
|
||
"measurement_id": request.MeasurementID,
|
||
"measurement_token": request.MeasurementToken,
|
||
},
|
||
})
|
||
return
|
||
}
|
||
|
||
pgClient := database.GetPostgresDBClient()
|
||
measurementInfo, err := database.QueryMeasurementByID(c, pgClient, request.MeasurementID)
|
||
if err != nil {
|
||
logger.Error(c, "failed to query measurement by id", "measurement_id", request.MeasurementID, "error", err)
|
||
c.JSON(http.StatusOK, network.FailureResponse{
|
||
Code: http.StatusBadRequest,
|
||
Msg: err.Error(),
|
||
PayLoad: map[string]interface{}{
|
||
"measurement_id": request.MeasurementID,
|
||
"measurement_token": request.MeasurementToken,
|
||
"measurement_value": points,
|
||
},
|
||
})
|
||
return
|
||
}
|
||
|
||
c.JSON(http.StatusOK, network.SuccessResponse{
|
||
Code: http.StatusOK,
|
||
Msg: "success",
|
||
PayLoad: map[string]interface{}{
|
||
"measurement_id": request.MeasurementID,
|
||
"measurement_token": request.MeasurementToken,
|
||
"measurement_info": measurementInfo,
|
||
"measurement_value": points,
|
||
},
|
||
})
|
||
}
|