88 lines
2.4 KiB
Go
88 lines
2.4 KiB
Go
package v1
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"message-nest/pkg/app"
|
|
"message-nest/service/statistic_service"
|
|
"net/http"
|
|
)
|
|
|
|
// GetStatisticData 获取发送统计数据
|
|
func GetStatisticData(c *gin.Context) {
|
|
var (
|
|
appG = app.Gin{C: c}
|
|
)
|
|
|
|
// 获取类型参数
|
|
statisticType := c.Query("type")
|
|
|
|
msgService := statistic_service.StatisticService{}
|
|
|
|
switch statisticType {
|
|
case "basic":
|
|
data, err := msgService.GetBasicStatisticData()
|
|
if err != nil {
|
|
appG.CResponse(http.StatusInternalServerError, fmt.Sprintf("获取基础统计失败！原因：%s", err), nil)
|
|
return
|
|
}
|
|
appG.CResponse(http.StatusOK, "获取基础统计成功", data)
|
|
case "trend":
|
|
data, err := msgService.GetTrendStatisticData()
|
|
if err != nil {
|
|
appG.CResponse(http.StatusInternalServerError, fmt.Sprintf("获取趋势统计失败！原因：%s", err), nil)
|
|
return
|
|
}
|
|
appG.CResponse(http.StatusOK, "获取趋势统计成功", data)
|
|
case "channels":
|
|
data, err := msgService.GetChannelStatisticData()
|
|
if err != nil {
|
|
appG.CResponse(http.StatusInternalServerError, fmt.Sprintf("获取渠道统计失败！原因：%s", err), nil)
|
|
return
|
|
}
|
|
appG.CResponse(http.StatusOK, "获取渠道统计成功", data)
|
|
case "send_stats":
|
|
// 新增：基于 send_stats 表的统计数据
|
|
data, err := msgService.GetSendStatsData()
|
|
if err != nil {
|
|
appG.CResponse(http.StatusInternalServerError, fmt.Sprintf("获取发送统计失败！原因：%s", err), nil)
|
|
return
|
|
}
|
|
appG.CResponse(http.StatusOK, "获取发送统计成功", data)
|
|
default:
|
|
// 默认返回完整统计数据（保持向后兼容）
|
|
data, err := msgService.GetStatisticData()
|
|
if err != nil {
|
|
appG.CResponse(http.StatusInternalServerError, fmt.Sprintf("获取失败！原因：%s", err), nil)
|
|
return
|
|
}
|
|
appG.CResponse(http.StatusOK, "获取成功", data)
|
|
}
|
|
}
|
|
|
|
// GetSendStatsByTask 获取指定任务的发送统计数据
|
|
func GetSendStatsByTask(c *gin.Context) {
|
|
var (
|
|
appG = app.Gin{C: c}
|
|
)
|
|
|
|
taskID := c.Query("task_id")
|
|
if taskID == "" {
|
|
appG.CResponse(http.StatusBadRequest, "任务ID不能为空", nil)
|
|
return
|
|
}
|
|
|
|
msgService := statistic_service.StatisticService{
|
|
TaskID: taskID,
|
|
}
|
|
|
|
data, err := msgService.GetSendStatsByTask()
|
|
if err != nil {
|
|
appG.CResponse(http.StatusInternalServerError, fmt.Sprintf("获取任务统计失败！原因：%s", err), nil)
|
|
return
|
|
}
|
|
|
|
appG.CResponse(http.StatusOK, "获取任务统计成功", data)
|
|
}
|
|
|