111 lines
2.9 KiB
Go
111 lines
2.9 KiB
Go
package message
|
||
|
||
import (
|
||
"github.com/gin-gonic/gin"
|
||
"go-admin/app/admin/models"
|
||
orm "go-admin/common/global"
|
||
"go-admin/tools"
|
||
"go-admin/tools/app"
|
||
"net/http"
|
||
"time"
|
||
)
|
||
|
||
// UserMessageList 用户消息列表
|
||
// @Summary 用户消息列表
|
||
// @Description 获取当前用户的消息记录
|
||
// @Tags 消息中心，V1.5.0
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param data body models.UserMessageListReq true "消息查询参数"
|
||
// @Success 200 {object} models.UserMessageListResp
|
||
// @Router /api/v1/user_message/list [post]
|
||
func UserMessageList(c *gin.Context) {
|
||
var req models.UserMessageListReq
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
app.Error(c, http.StatusBadRequest, nil, "参数错误: "+err.Error())
|
||
return
|
||
}
|
||
|
||
if req.Page <= 0 {
|
||
req.Page = 1
|
||
}
|
||
if req.PageSize <= 0 {
|
||
req.PageSize = 10
|
||
}
|
||
|
||
userID := tools.GetUserId(c)
|
||
query := orm.Eloquent.Where("user_id = ? AND is_visible = ?", userID, true)
|
||
|
||
if req.IsRead != 0 && req.IsRead != 1 {
|
||
app.Error(c, http.StatusBadRequest, nil, "不支持的目标状态")
|
||
return
|
||
} else {
|
||
query = query.Where("is_read = ?", req.IsRead)
|
||
}
|
||
|
||
var total int64
|
||
if err := query.Model(&models.SystemUserMessage{}).Count(&total).Error; err != nil {
|
||
app.Error(c, http.StatusInternalServerError, nil, "查询总数失败: "+err.Error())
|
||
return
|
||
}
|
||
|
||
var messages []models.SystemUserMessage
|
||
if err := query.Order("created_at desc").
|
||
Offset((req.Page - 1) * req.PageSize).
|
||
Limit(req.PageSize).
|
||
Find(&messages).Error; err != nil {
|
||
app.Error(c, http.StatusInternalServerError, nil, "查询列表失败: "+err.Error())
|
||
return
|
||
}
|
||
|
||
app.OK(c, models.UserMessageListResp{
|
||
List: messages,
|
||
Total: total,
|
||
Page: req.Page,
|
||
PageSize: req.PageSize,
|
||
}, "查询成功")
|
||
}
|
||
|
||
// UserMessageSetStatus 设置用户消息为已读（支持单条或全部）
|
||
// @Summary 设置用户消息为已读
|
||
// @Description 设为单条或全部已读
|
||
// @Tags 消息中心，V1.5.0
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param data body models.UserMessageSetStatusReq true "设置参数"
|
||
// @Success 200 {object} app.Response
|
||
// @Router /api/v1/user_message/set_status [post]
|
||
func UserMessageSetStatus(c *gin.Context) {
|
||
var req models.UserMessageSetStatusReq
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
app.Error(c, http.StatusBadRequest, nil, "参数错误: "+err.Error())
|
||
return
|
||
}
|
||
|
||
userID := tools.GetUserId(c)
|
||
now := time.Now()
|
||
query := orm.Eloquent.Model(&models.SystemUserMessage{}).
|
||
Where("user_id = ? AND is_read = ?", userID, false)
|
||
|
||
if req.ReadAll {
|
||
// 全部设为已读
|
||
} else {
|
||
if req.ID == 0 {
|
||
app.Error(c, http.StatusBadRequest, nil, "参数错误: 未提供消息ID")
|
||
return
|
||
}
|
||
query = query.Where("id = ?", req.ID)
|
||
}
|
||
|
||
err := query.Updates(map[string]interface{}{
|
||
"is_read": true,
|
||
"read_time": &now,
|
||
}).Error
|
||
if err != nil {
|
||
app.Error(c, http.StatusInternalServerError, nil, "设置失败: "+err.Error())
|
||
return
|
||
}
|
||
|
||
app.OK(c, nil, "设置成功")
|
||
}
|