77 lines
2.1 KiB
Go
77 lines
2.1 KiB
Go
package getnotifications
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/gin-gonic/gin"
|
||
"net/http"
|
||
"strconv"
|
||
"time"
|
||
"toutoukan/init/databaseInit"
|
||
)
|
||
|
||
type Notification struct {
|
||
// ⭐️ 新增: 主键 id
|
||
ID uint `gorm:"primaryKey" json:"id"` // GORM 会自动识别为自增主键
|
||
SenderID string `json:"sender_id"`
|
||
ReceiverID string `json:"receiver_id"`
|
||
Status int8 `json:"status"`
|
||
Sequence string `json:"sequence"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
Content string `json:"content"`
|
||
IsRead int8 `json:"is_read"`
|
||
MsgType string `json:"msg_type"`
|
||
Target string `json:"target"`
|
||
}
|
||
|
||
// TableName 指定 GORM 使用的表名
|
||
func (Notification) TableName() string {
|
||
return "user_msg"
|
||
}
|
||
|
||
type UidRequest struct {
|
||
Uid string `json:"uid" binding:"required"` // binding:"required" 确保 uid 字段必须存在
|
||
}
|
||
|
||
// --- Gin Handler ---
|
||
|
||
// GetNotifications 使用 GORM 进行数据库查询
|
||
func GetNotifications(c *gin.Context) {
|
||
// 1. 从请求 Body 中获取 uid
|
||
var requestBody UidRequest
|
||
if err := c.ShouldBindJSON(&requestBody); err != nil {
|
||
// 如果 JSON 格式不正确或缺少 uid 字段，ShouldBindJSON 会返回错误
|
||
c.JSON(http.StatusBadRequest, gin.H{"message": "请求参数错误: " + err.Error()})
|
||
return
|
||
}
|
||
userID := requestBody.Uid
|
||
|
||
// 2. 获取分页参数 (这部分逻辑不变)
|
||
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
||
limit, _ := strconv.Atoi(c.DefaultQuery("limit", "20"))
|
||
offset := (page - 1) * limit
|
||
|
||
// 3. 准备用于接收结果的切片
|
||
var notifications []Notification
|
||
|
||
// 4. 使用 GORM 进行链式查询 (这部分逻辑不变)
|
||
result := databaseInit.UserDB.
|
||
Where("receiver_id = ?", userID).
|
||
Order("created_at DESC").
|
||
Limit(limit).
|
||
Offset(offset).
|
||
Find(¬ifications)
|
||
|
||
// 5. GORM 的错误处理 (这部分逻辑不变)
|
||
if result.Error != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"message": fmt.Sprintf("查询数据库失败: %v", result.Error)})
|
||
return
|
||
}
|
||
|
||
// 6. 返回 JSON 响应 (这部分逻辑不变)
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"data": notifications,
|
||
"message": "获取消息列表成功",
|
||
"success": true,
|
||
})
|
||
}
|