搭建Go后端框架：数据模型、JWT鉴权、租户隔离、WebSocket、REST API路由

server/internal/handler/session.go

@@ -0,0 +1,180 @@
+package handler
+
+import (
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+	"kefu-sys/server/internal/middleware"
+	"kefu-sys/server/internal/model"
+)
+
+type SessionHandler struct{}
+
+func NewSessionHandler() *SessionHandler { return &SessionHandler{} }
+
+type CreateSessionReq struct {
+	ChannelID  uint   `json:"channel_id"`
+	CustomerID uint   `json:"customer_id"`
+	Priority   string `json:"priority"`
+}
+
+type AssignSessionReq struct {
+	AgentID uint `json:"agent_id" binding:"required"`
+}
+
+func (h *SessionHandler) List(c *gin.Context) {
+	tenantID := middleware.GetTenantID(c)
+	page, pageSize := middleware.GetPageParams(c)
+	status := c.Query("status")
+	priority := c.Query("priority")
+
+	var sessions []model.Session
+	var total int64
+
+	query := model.DB.Where("tenant_id = ?", tenantID)
+	if status != "" {
+		query = query.Where("status = ?", status)
+	}
+	if priority != "" {
+		query = query.Where("priority = ?", priority)
+	}
+
+	query.Model(&model.Session{}).Count(&total)
+	query.Order("created_at desc").Offset((page - 1) * pageSize).Limit(pageSize).Find(&sessions)
+
+	middleware.JSONList(c, sessions, total, page, pageSize)
+}
+
+func (h *SessionHandler) Get(c *gin.Context) {
+	tenantID := middleware.GetTenantID(c)
+	id := c.Param("id")
+
+	var session model.Session
+	if err := model.DB.Where("id = ? AND tenant_id = ?", id, tenantID).First(&session).Error; err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"code": 404, "message": "会话不存在"})
+		return
+	}
+
+	var messages []model.Message
+	model.DB.Where("session_id = ?", session.ID).Order("seq asc").Find(&messages)
+
+	middleware.JSON(c, gin.H{"session": session, "messages": messages})
+}
+
+func (h *SessionHandler) Create(c *gin.Context) {
+	var req CreateSessionReq
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误"})
+		return
+	}
+
+	session := model.Session{
+		TenantID:   middleware.GetTenantID(c),
+		ChannelID:  req.ChannelID,
+		CustomerID: req.CustomerID,
+		Priority:   req.Priority,
+		Status:     "waiting",
+	}
+
+	if session.Priority == "" {
+		session.Priority = "normal"
+	}
+
+	if err := model.DB.Create(&session).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "message": "创建会话失败"})
+		return
+	}
+
+	middleware.JSON(c, session)
+}
+
+func (h *SessionHandler) Assign(c *gin.Context) {
+	tenantID := middleware.GetTenantID(c)
+	id := c.Param("id")
+	var req AssignSessionReq
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误"})
+		return
+	}
+
+	result := model.DB.Model(&model.Session{}).
+		Where("id = ? AND tenant_id = ? AND status = ?", id, tenantID, "waiting").
+		Updates(map[string]interface{}{"agent_id": req.AgentID, "status": "active"})
+
+	if result.RowsAffected == 0 {
+		c.JSON(http.StatusNotFound, gin.H{"code": 404, "message": "会话不存在或已被分配"})
+		return
+	}
+
+	middleware.JSON(c, gin.H{"message": "分配成功"})
+}
+
+func (h *SessionHandler) Transfer(c *gin.Context) {
+	tenantID := middleware.GetTenantID(c)
+	id := c.Param("id")
+	var req AssignSessionReq
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误"})
+		return
+	}
+
+	model.DB.Model(&model.Session{}).
+		Where("id = ? AND tenant_id = ?", id, tenantID).
+		Update("agent_id", req.AgentID)
+
+	model.DB.Create(&model.SessionEvent{
+		SessionID:  parseID(id),
+		OperatorID: middleware.GetUserID(c),
+		Action:     "transfer",
+		Detail:     "会话转接",
+	})
+
+	middleware.JSON(c, gin.H{"message": "转接成功"})
+}
+
+func (h *SessionHandler) End(c *gin.Context) {
+	tenantID := middleware.GetTenantID(c)
+	id := c.Param("id")
+	reason := c.Query("reason")
+
+	result := model.DB.Model(&model.Session{}).
+		Where("id = ? AND tenant_id = ?", id, tenantID).
+		Updates(map[string]interface{}{"status": "ended", "end_reason": reason})
+
+	if result.RowsAffected == 0 {
+		c.JSON(http.StatusNotFound, gin.H{"code": 404, "message": "会话不存在"})
+		return
+	}
+
+	model.DB.Create(&model.SessionEvent{
+		SessionID:  parseID(id),
+		OperatorID: middleware.GetUserID(c),
+		Action:     "end",
+		Detail:     "结束会话: " + reason,
+	})
+
+	middleware.JSON(c, gin.H{"message": "已结束"})
+}
+
+func (h *SessionHandler) UpdatePriority(c *gin.Context) {
+	tenantID := middleware.GetTenantID(c)
+	id := c.Param("id")
+	priority := c.Query("priority")
+
+	model.DB.Model(&model.Session{}).
+		Where("id = ? AND tenant_id = ?", id, tenantID).
+		Update("priority", priority)
+
+	middleware.JSON(c, gin.H{"message": "已更新"})
+}
+
+func parseID(s string) uint {
+	var id uint
+	// Simple atoi for uint
+	for _, c := range s {
+		if c >= '0' && c <= '9' {
+			id = id*10 + uint(c-'0')
+		}
+	}
+	return id
+}