feat(im): Task 5 - IM REST Controller + Router + Wire 集成

IMController (REST API):
- GET /api/v1/im/conversations: 会话列表
- GET /api/v1/im/messages: 历史消息(游标分页)
- PUT /api/v1/im/conversations/:id/pin: 置顶/取消
- DELETE /api/v1/im/conversations/🆔 删除会话
- DELETE /api/v1/im/conversations/:id/messages: 清空记录
- GET /api/v1/im/messages/search: 全局搜索
- GET /api/v1/im/unread: 全局未读数

Wire 集成:
- IMSet Provider Set + wire.Bind 接口注入
- App struct 新增 IM 字段
- wire_gen.go 手动更新
- OfflinePusher 通过 SetOfflinePusher 注入 WS Handler

Made-with: Cursor
This commit is contained in:
bujinyuan
2026-03-03 10:51:56 +08:00
parent 1e0c061e9d
commit 2c7be494f2
7 changed files with 344 additions and 13 deletions

View File

@@ -0,0 +1,213 @@
// Package controller 提供 IM 模块的 HTTP 接口处理
package controller
import (
"strconv"
"github.com/echochat/backend/app/dto"
"github.com/echochat/backend/app/im/service"
"github.com/echochat/backend/pkg/middleware"
"github.com/echochat/backend/pkg/utils"
"github.com/gin-gonic/gin"
)
// IMController 即时通讯控制器REST API
type IMController struct {
imService *service.IMService
}
// NewIMController 创建 IMController 实例
func NewIMController(imService *service.IMService) *IMController {
return &IMController{imService: imService}
}
// GetConversations 获取会话列表
// GET /api/v1/im/conversations
func (ctl *IMController) GetConversations(c *gin.Context) {
ctx := c.Request.Context()
userID, ok := middleware.GetCurrentUserID(c)
if !ok {
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
return
}
result, err := ctl.imService.GetConversationList(ctx, userID)
if err != nil {
ctl.handleError(c, err, "获取会话列表失败")
return
}
utils.ResponseOK(c, result)
}
// GetHistoryMessages 获取历史消息
// GET /api/v1/im/messages?conversation_id=xx&before_id=xx&limit=30
func (ctl *IMController) GetHistoryMessages(c *gin.Context) {
ctx := c.Request.Context()
userID, ok := middleware.GetCurrentUserID(c)
if !ok {
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
return
}
var req dto.HistoryMessageRequest
if err := c.ShouldBindQuery(&req); err != nil {
utils.ResponseBadRequest(c, "请求参数错误: "+err.Error())
return
}
result, err := ctl.imService.GetHistoryMessages(ctx, userID, &req)
if err != nil {
ctl.handleError(c, err, "获取历史消息失败")
return
}
utils.ResponseOK(c, result)
}
// PinConversation 置顶/取消置顶会话
// PUT /api/v1/im/conversations/:id/pin
func (ctl *IMController) PinConversation(c *gin.Context) {
ctx := c.Request.Context()
userID, ok := middleware.GetCurrentUserID(c)
if !ok {
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
return
}
convID, err := strconv.ParseInt(c.Param("id"), 10, 64)
if err != nil {
utils.ResponseBadRequest(c, "会话 ID 格式错误")
return
}
var req dto.PinConversationRequest
if err := c.ShouldBindJSON(&req); err != nil {
utils.ResponseBadRequest(c, "请求参数错误: "+err.Error())
return
}
if err := ctl.imService.PinConversation(ctx, userID, convID, req.IsPinned); err != nil {
ctl.handleError(c, err, "更新置顶状态失败")
return
}
utils.ResponseOK(c, nil)
}
// DeleteConversation 删除会话
// DELETE /api/v1/im/conversations/:id
func (ctl *IMController) DeleteConversation(c *gin.Context) {
ctx := c.Request.Context()
userID, ok := middleware.GetCurrentUserID(c)
if !ok {
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
return
}
convID, err := strconv.ParseInt(c.Param("id"), 10, 64)
if err != nil {
utils.ResponseBadRequest(c, "会话 ID 格式错误")
return
}
if err := ctl.imService.DeleteConversation(ctx, userID, convID); err != nil {
ctl.handleError(c, err, "删除会话失败")
return
}
utils.ResponseOK(c, nil)
}
// ClearHistory 清空聊天记录
// DELETE /api/v1/im/conversations/:id/messages
func (ctl *IMController) ClearHistory(c *gin.Context) {
ctx := c.Request.Context()
userID, ok := middleware.GetCurrentUserID(c)
if !ok {
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
return
}
convID, err := strconv.ParseInt(c.Param("id"), 10, 64)
if err != nil {
utils.ResponseBadRequest(c, "会话 ID 格式错误")
return
}
if err := ctl.imService.ClearHistory(ctx, userID, convID); err != nil {
ctl.handleError(c, err, "清空聊天记录失败")
return
}
utils.ResponseOK(c, nil)
}
// SearchMessages 全局消息搜索
// GET /api/v1/im/messages/search?keyword=xx&limit=50
func (ctl *IMController) SearchMessages(c *gin.Context) {
ctx := c.Request.Context()
userID, ok := middleware.GetCurrentUserID(c)
if !ok {
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
return
}
var req dto.SearchMessageRequest
if err := c.ShouldBindQuery(&req); err != nil {
utils.ResponseBadRequest(c, "请求参数错误: "+err.Error())
return
}
result, err := ctl.imService.SearchMessages(ctx, userID, &req)
if err != nil {
ctl.handleError(c, err, "消息搜索失败")
return
}
utils.ResponseOK(c, result)
}
// GetTotalUnread 获取全局未读消息总数
// GET /api/v1/im/unread
func (ctl *IMController) GetTotalUnread(c *gin.Context) {
ctx := c.Request.Context()
userID, ok := middleware.GetCurrentUserID(c)
if !ok {
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
return
}
count, err := ctl.imService.GetTotalUnread(ctx, userID)
if err != nil {
ctl.handleError(c, err, "获取未读消息数失败")
return
}
utils.ResponseOK(c, gin.H{"total_unread": count})
}
// handleError 统一业务错误映射
// 已知业务错误 → 返回 Service 层定义的具体提示
// 未知错误 → 返回 fallbackMsg未传则默认"服务器内部错误"
func (ctl *IMController) handleError(c *gin.Context, err error, fallbackMsg ...string) {
switch err {
case service.ErrNotFriend:
utils.ResponseForbidden(c, err.Error())
case service.ErrEmptyContent:
utils.ResponseBadRequest(c, err.Error())
case service.ErrInvalidMsgType:
utils.ResponseBadRequest(c, err.Error())
case service.ErrConvNotFound:
utils.ResponseNotFound(c, err.Error())
case service.ErrMsgNotFound:
utils.ResponseNotFound(c, err.Error())
case service.ErrNotSender:
utils.ResponseForbidden(c, err.Error())
case service.ErrRecallTimeout:
utils.ResponseBadRequest(c, err.Error())
case service.ErrNotMember:
utils.ResponseForbidden(c, err.Error())
case service.ErrDuplicateMsg:
utils.ResponseBadRequest(c, err.Error())
default:
msg := "服务器内部错误"
if len(fallbackMsg) > 0 && fallbackMsg[0] != "" {
msg = fallbackMsg[0]
}
utils.ResponseError(c, msg)
}
}