拆分聊天 Handler 文件职责

This commit is contained in:
yml2213
2026-06-10 15:03:58 +08:00
parent 70a4d8e9ff
commit b8b7221df8
7 changed files with 440 additions and 403 deletions
-403
View File
@@ -1,16 +1,5 @@
package chat
import (
"errors"
"net/http"
"strconv"
"hfb_sys/backend/internal/middleware"
"hfb_sys/backend/pkg/response"
"github.com/gin-gonic/gin"
)
type Handler struct {
service *Service
}
@@ -18,395 +7,3 @@ type Handler struct {
func NewHandler(service *Service) *Handler {
return &Handler{service: service}
}
func (h *Handler) List(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
h.list(c, Principal{Type: "user", ID: userID})
}
func (h *Handler) AdminList(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
filter := c.DefaultQuery("filter", "all")
page, pageSize := parsePagination(c)
principal := Principal{Type: "admin", ID: adminID}
result, err := h.service.ListConversationsWithFilter(c.Request.Context(), principal, page, pageSize, filter)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, result)
}
func (h *Handler) Detail(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
h.detail(c, Principal{Type: "user", ID: userID})
}
func (h *Handler) AdminDetail(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
h.detail(c, Principal{Type: "admin", ID: adminID})
}
func (h *Handler) OrderConversation(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
orderID, ok := parseID(c, "id")
if !ok {
return
}
item, err := h.service.FindOrderConversation(c.Request.Context(), userID, orderID)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, item)
}
func (h *Handler) EnsureSupportConversation(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
item, err := h.service.EnsureSupportConversation(c.Request.Context(), userID)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, item)
}
func (h *Handler) Messages(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
h.messages(c, Principal{Type: "user", ID: userID})
}
func (h *Handler) AdminMessages(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
h.messages(c, Principal{Type: "admin", ID: adminID})
}
func (h *Handler) Send(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
h.send(c, Principal{Type: "user", ID: userID})
}
func (h *Handler) AdminSend(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
h.send(c, Principal{Type: "admin", ID: adminID})
}
func (h *Handler) MarkRead(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
h.markRead(c, Principal{Type: "user", ID: userID})
}
func (h *Handler) AdminMarkRead(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
h.markRead(c, Principal{Type: "admin", ID: adminID})
}
func (h *Handler) AdminTransfer(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
id, ok := parseID(c, "id")
if !ok {
return
}
var req TransferRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "目标客服 ID 不能为空")
return
}
principal := Principal{Type: "admin", ID: adminID}
if err := h.service.TransferConversation(c.Request.Context(), principal, id, req); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"transferred": true})
}
func (h *Handler) AdminSupportAdmins(c *gin.Context) {
admins, err := h.service.GetAvailableSupportAdmins(c.Request.Context())
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, admins)
}
func (h *Handler) AdminUpdateRemark(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
id, ok := parseID(c, "id")
if !ok {
return
}
var req UpdateRemarkRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "请求参数错误")
return
}
principal := Principal{Type: "admin", ID: adminID}
if err := h.service.UpdateRemark(c.Request.Context(), principal, id, req); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"updated": true})
}
func (h *Handler) AdminListQuickReplies(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
replies, err := h.service.ListQuickReplies(c.Request.Context(), adminID)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, replies)
}
func (h *Handler) AdminCreateQuickReply(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
var req CreateQuickReplyRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "标题和内容不能为空")
return
}
reply, err := h.service.CreateQuickReply(c.Request.Context(), adminID, req)
if err != nil {
writeChatError(c, err)
return
}
response.Created(c, reply)
}
func (h *Handler) AdminUpdateQuickReply(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
id, ok := parseID(c, "id")
if !ok {
return
}
var req UpdateQuickReplyRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "请求参数错误")
return
}
if err := h.service.UpdateQuickReply(c.Request.Context(), adminID, id, req); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"updated": true})
}
func (h *Handler) AdminDeleteQuickReply(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
id, ok := parseID(c, "id")
if !ok {
return
}
if err := h.service.DeleteQuickReply(c.Request.Context(), adminID, id); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"deleted": true})
}
func (h *Handler) AdminGetAutoWelcome(c *gin.Context) {
message := h.service.GetAutoWelcomeMessage(c.Request.Context())
response.OK(c, gin.H{"message": message})
}
func (h *Handler) AdminUpdateAutoWelcome(c *gin.Context) {
var req struct {
Message string `json:"message" binding:"required"`
}
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "话术内容不能为空")
return
}
if err := h.service.UpdateAutoWelcomeMessage(c.Request.Context(), req.Message); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"updated": true})
}
func (h *Handler) list(c *gin.Context, principal Principal) {
page, pageSize := parsePagination(c)
result, err := h.service.ListConversations(c.Request.Context(), principal, page, pageSize)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, result)
}
func (h *Handler) detail(c *gin.Context, principal Principal) {
id, ok := parseID(c, "id")
if !ok {
return
}
item, err := h.service.FindConversation(c.Request.Context(), principal, id)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, item)
}
func (h *Handler) messages(c *gin.Context, principal Principal) {
id, ok := parseID(c, "id")
if !ok {
return
}
page, pageSize := parsePagination(c)
result, err := h.service.Messages(c.Request.Context(), principal, id, page, pageSize)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, result)
}
func (h *Handler) send(c *gin.Context, principal Principal) {
id, ok := parseID(c, "id")
if !ok {
return
}
var req SendMessageRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "消息格式不正确")
return
}
message, err := h.service.SendMessage(c.Request.Context(), principal, id, req)
if err != nil {
writeChatError(c, err)
return
}
response.Created(c, message)
}
func (h *Handler) markRead(c *gin.Context, principal Principal) {
id, ok := parseID(c, "id")
if !ok {
return
}
if err := h.service.MarkRead(c.Request.Context(), principal, id); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"read": true})
}
func parsePagination(c *gin.Context) (int, int) {
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
return normalizePagination(page, pageSize)
}
func parseID(c *gin.Context, key string) (uint64, bool) {
id, err := strconv.ParseUint(c.Param(key), 10, 64)
if err != nil || id == 0 {
response.BadRequest(c, "ID 不正确")
return 0, false
}
return id, true
}
func currentUserID(c *gin.Context) (uint64, bool) {
value, ok := c.Get(middleware.ContextUserID)
if !ok {
return 0, false
}
userID, ok := value.(uint64)
return userID, ok
}
func currentAdminID(c *gin.Context) (uint64, bool) {
value, ok := c.Get(middleware.ContextAdminID)
if !ok {
return 0, false
}
adminID, ok := value.(uint64)
return adminID, ok
}
func writeChatError(c *gin.Context, err error) {
switch {
case errors.Is(err, ErrDependencyUnavailable):
response.ServiceUnavailable(c, "聊天服务暂时不可用")
case errors.Is(err, ErrConversationNotFound):
response.Error(c, http.StatusNotFound, "chat_not_found", "会话不存在")
case errors.Is(err, ErrPermissionDenied):
response.Error(c, http.StatusForbidden, "permission_denied", "无权访问该会话")
case errors.Is(err, ErrInvalidMessage):
response.BadRequest(c, "消息内容不符合规则")
default:
response.Error(c, http.StatusInternalServerError, "internal_error", "聊天服务暂时不可用")
}
}
@@ -0,0 +1,115 @@
package chat
import (
"hfb_sys/backend/pkg/response"
"github.com/gin-gonic/gin"
)
func (h *Handler) AdminList(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
filter := c.DefaultQuery("filter", "all")
page, pageSize := parsePagination(c)
principal := Principal{Type: "admin", ID: adminID}
result, err := h.service.ListConversationsWithFilter(c.Request.Context(), principal, page, pageSize, filter)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, result)
}
func (h *Handler) AdminDetail(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
h.detail(c, Principal{Type: "admin", ID: adminID})
}
func (h *Handler) AdminMessages(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
h.messages(c, Principal{Type: "admin", ID: adminID})
}
func (h *Handler) AdminSend(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
h.send(c, Principal{Type: "admin", ID: adminID})
}
func (h *Handler) AdminMarkRead(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
h.markRead(c, Principal{Type: "admin", ID: adminID})
}
func (h *Handler) AdminTransfer(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
id, ok := parseID(c, "id")
if !ok {
return
}
var req TransferRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "目标客服 ID 不能为空")
return
}
principal := Principal{Type: "admin", ID: adminID}
if err := h.service.TransferConversation(c.Request.Context(), principal, id, req); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"transferred": true})
}
func (h *Handler) AdminSupportAdmins(c *gin.Context) {
admins, err := h.service.GetAvailableSupportAdmins(c.Request.Context())
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, admins)
}
func (h *Handler) AdminUpdateRemark(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
id, ok := parseID(c, "id")
if !ok {
return
}
var req UpdateRemarkRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "请求参数错误")
return
}
principal := Principal{Type: "admin", ID: adminID}
if err := h.service.UpdateRemark(c.Request.Context(), principal, id, req); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"updated": true})
}
@@ -0,0 +1,74 @@
package chat
import (
"hfb_sys/backend/pkg/response"
"github.com/gin-gonic/gin"
)
func (h *Handler) list(c *gin.Context, principal Principal) {
page, pageSize := parsePagination(c)
result, err := h.service.ListConversations(c.Request.Context(), principal, page, pageSize)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, result)
}
func (h *Handler) detail(c *gin.Context, principal Principal) {
id, ok := parseID(c, "id")
if !ok {
return
}
item, err := h.service.FindConversation(c.Request.Context(), principal, id)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, item)
}
func (h *Handler) messages(c *gin.Context, principal Principal) {
id, ok := parseID(c, "id")
if !ok {
return
}
page, pageSize := parsePagination(c)
result, err := h.service.Messages(c.Request.Context(), principal, id, page, pageSize)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, result)
}
func (h *Handler) send(c *gin.Context, principal Principal) {
id, ok := parseID(c, "id")
if !ok {
return
}
var req SendMessageRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "消息格式不正确")
return
}
message, err := h.service.SendMessage(c.Request.Context(), principal, id, req)
if err != nil {
writeChatError(c, err)
return
}
response.Created(c, message)
}
func (h *Handler) markRead(c *gin.Context, principal Principal) {
id, ok := parseID(c, "id")
if !ok {
return
}
if err := h.service.MarkRead(c.Request.Context(), principal, id); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"read": true})
}
@@ -0,0 +1,25 @@
package chat
import (
"errors"
"net/http"
"hfb_sys/backend/pkg/response"
"github.com/gin-gonic/gin"
)
func writeChatError(c *gin.Context, err error) {
switch {
case errors.Is(err, ErrDependencyUnavailable):
response.ServiceUnavailable(c, "聊天服务暂时不可用")
case errors.Is(err, ErrConversationNotFound):
response.Error(c, http.StatusNotFound, "chat_not_found", "会话不存在")
case errors.Is(err, ErrPermissionDenied):
response.Error(c, http.StatusForbidden, "permission_denied", "无权访问该会话")
case errors.Is(err, ErrInvalidMessage):
response.BadRequest(c, "消息内容不符合规则")
default:
response.Error(c, http.StatusInternalServerError, "internal_error", "聊天服务暂时不可用")
}
}
@@ -0,0 +1,43 @@
package chat
import (
"strconv"
"hfb_sys/backend/internal/middleware"
"hfb_sys/backend/pkg/response"
"github.com/gin-gonic/gin"
)
func parsePagination(c *gin.Context) (int, int) {
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
return normalizePagination(page, pageSize)
}
func parseID(c *gin.Context, key string) (uint64, bool) {
id, err := strconv.ParseUint(c.Param(key), 10, 64)
if err != nil || id == 0 {
response.BadRequest(c, "ID 不正确")
return 0, false
}
return id, true
}
func currentUserID(c *gin.Context) (uint64, bool) {
value, ok := c.Get(middleware.ContextUserID)
if !ok {
return 0, false
}
userID, ok := value.(uint64)
return userID, ok
}
func currentAdminID(c *gin.Context) (uint64, bool) {
value, ok := c.Get(middleware.ContextAdminID)
if !ok {
return 0, false
}
adminID, ok := value.(uint64)
return adminID, ok
}
@@ -0,0 +1,99 @@
package chat
import (
"hfb_sys/backend/pkg/response"
"github.com/gin-gonic/gin"
)
func (h *Handler) AdminListQuickReplies(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
replies, err := h.service.ListQuickReplies(c.Request.Context(), adminID)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, replies)
}
func (h *Handler) AdminCreateQuickReply(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
var req CreateQuickReplyRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "标题和内容不能为空")
return
}
reply, err := h.service.CreateQuickReply(c.Request.Context(), adminID, req)
if err != nil {
writeChatError(c, err)
return
}
response.Created(c, reply)
}
func (h *Handler) AdminUpdateQuickReply(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
id, ok := parseID(c, "id")
if !ok {
return
}
var req UpdateQuickReplyRequest
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "请求参数错误")
return
}
if err := h.service.UpdateQuickReply(c.Request.Context(), adminID, id, req); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"updated": true})
}
func (h *Handler) AdminDeleteQuickReply(c *gin.Context) {
adminID, ok := currentAdminID(c)
if !ok {
response.Unauthorized(c, "缺少管理员上下文")
return
}
id, ok := parseID(c, "id")
if !ok {
return
}
if err := h.service.DeleteQuickReply(c.Request.Context(), adminID, id); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"deleted": true})
}
func (h *Handler) AdminGetAutoWelcome(c *gin.Context) {
message := h.service.GetAutoWelcomeMessage(c.Request.Context())
response.OK(c, gin.H{"message": message})
}
func (h *Handler) AdminUpdateAutoWelcome(c *gin.Context) {
var req struct {
Message string `json:"message" binding:"required"`
}
if err := c.ShouldBindJSON(&req); err != nil {
response.BadRequest(c, "话术内容不能为空")
return
}
if err := h.service.UpdateAutoWelcomeMessage(c.Request.Context(), req.Message); err != nil {
writeChatError(c, err)
return
}
response.OK(c, gin.H{"updated": true})
}
@@ -0,0 +1,84 @@
package chat
import (
"hfb_sys/backend/pkg/response"
"github.com/gin-gonic/gin"
)
func (h *Handler) List(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
h.list(c, Principal{Type: "user", ID: userID})
}
func (h *Handler) Detail(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
h.detail(c, Principal{Type: "user", ID: userID})
}
func (h *Handler) OrderConversation(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
orderID, ok := parseID(c, "id")
if !ok {
return
}
item, err := h.service.FindOrderConversation(c.Request.Context(), userID, orderID)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, item)
}
func (h *Handler) EnsureSupportConversation(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
item, err := h.service.EnsureSupportConversation(c.Request.Context(), userID)
if err != nil {
writeChatError(c, err)
return
}
response.OK(c, item)
}
func (h *Handler) Messages(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
h.messages(c, Principal{Type: "user", ID: userID})
}
func (h *Handler) Send(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
h.send(c, Principal{Type: "user", ID: userID})
}
func (h *Handler) MarkRead(c *gin.Context) {
userID, ok := currentUserID(c)
if !ok {
response.Unauthorized(c, "缺少用户上下文")
return
}
h.markRead(c, Principal{Type: "user", ID: userID})
}