feat: 实现RBAC系统 - 完成Tenant/Role/Resource/User模型、路由配置及MySQL连接配置

This commit is contained in:
OpenClaw
2026-03-01 11:55:04 +08:00
parent 1de4524b5e
commit 3af5009ba8
6 changed files with 870 additions and 4 deletions

View File

@@ -0,0 +1,204 @@
package handlers
import (
"github.com/gin-gonic/gin"
"smart-customer-service/config"
"smart-customer-service/internal/database"
"smart-customer-service/internal/models"
"smart-customer-service/internal/middleware"
"strconv"
"time"
)
type TicketHandler struct {
cfg *config.Config
}
func (h *TicketHandler) List(c *gin.Context) {
tenantID, _ := c.Get("tenant_id")
db := database.DB
// 查询参数
status := c.Query("status")
priority := c.Query("priority")
category := c.Query("category")
pagesize := c.DefaultQuery("page_size", "10")
page := c.DefaultQuery("page", "1")
limit, _ := strconv.Atoi(pagesize)
offset, _ := strconv.Atoi(page)
q := db.Where("tenant_id = ?", tenantID)
if status != "" {
q = q.Where("status = ?", status)
}
if priority != "" {
q = q.Where("priority = ?", priority)
}
if category != "" {
q = q.Where("category = ?", category)
}
var tickets []models.Ticket
var total int64
q.Count(&total)
q.Preload("User").Order("priority DESC, created_at DESC").Limit(limit).Offset((offset-1)*limit).Find(\&tickets)
c.JSON(200, gin.H{
"total": total,
"page": page,
"page_size": limit,
"tickets": tickets,
})
}
func (h *TicketHandler) Create(c *gin.Context) {
userID, _ := c.Get("user_id").(uint)
tenantID, _ := c.Get("tenant_id").(uint)
var ticket models.Ticket
if err := c.ShouldBindJSON(\&ticket); err != nil {
c.JSON(400, gin.H{"error": "参数错误", "message": err.Error()})
return
}
// 生成工单编号
now := time.Now()
ticket.TicketNumber = h.generateTicketNumber(now)
ticket.UserID = \&userID
ticket.TenantID = tenantID
if ticket.Priority == "" {
ticket.Priority = models.TicketPriorityMedium
}
if ticket.Status == "" {
ticket.Status = models.TicketStatusOpen
}
if ticket.Category == nil {
category := models.TicketCategorySupport
ticket.Category = \&category
}
if err := database.DB.Create(\&ticket).Error; err != nil {
c.JSON(500, gin.H{"error": "创建失败", "message": err.Error()})
return
}
c.JSON(201, gin.H{
"message": "工单创建成功",
"ticket_id": ticket.ID,
"ticket_number": ticket.TicketNumber,
})
}
func (h *TicketHandler) Get(c *gin.Context) {
idStr := c.Param("id")
id, _ := strconv.ParseUint(idStr, 10, 32)
var ticket models.Ticket
if err := database.DB.Preload("User").Preload("Assignee").First(\&ticket, uint(id)); err != nil {
c.JSON(404, gin.H{"error": "工单不存在"})
return
}
c.JSON(200, ticket)
}
func (h *TicketHandler) Update(c *gin.Context) {
idStr := c.Param("id")
id, _ := strconv.ParseUint(idStr, 10, 32)
var ticket models.Ticket
if err := database.DB.First(\&ticket, uint(id)).Error; err != nil {
c.JSON(404, gin.H{"error": "工单不存在"})
return
}
if err := c.ShouldBindJSON(\&ticket); err != nil {
c.JSON(400, gin.H{"error": "参数错误", "message": err.Error()})
return
}
if err := database.DB.Save(\&ticket).Error; err != nil {
c.JSON(500, gin.H{"error": "更新失败", "message": err.Error()})
return
}
c.JSON(200, gin.H{"message": "工单更新成功"})
}
func (h *TicketHandler) Delete(c *gin.Context) {
idStr := c.Param("id")
id, _ := strconv.ParseUint(idStr, 10, 32)
if err := database.DB.Delete(\&models.Ticket{}, uint(id)).Error; err != nil {
c.JSON(500, gin.H{"error": "删除失败", "message": err.Error()})
return
}
c.JSON(200, gin.H{"message": "工单删除成功"})
}
func (h *TicketHandler) Assign(c *gin.Context) {
idStr := c.Param("id"),
id, _ := strconv.ParseUint(idStr, 10, 32)
var ticket models.Ticket
if err := database.DB.First(\&ticket, uint(id)).Error; err != nil {
c.JSON(404, gin.H{"error": "工单不存在"})
return
}
var assignRequest struct {
AssignedTo uint `json:"assigned_to"`
}
if err := c.ShouldBindJSON(\&assignRequest); err != nil {
c.JSON(400, gin.H{"error": "参数错误", "message": err.Error()})
return
}
ticket.AssignedTo = \&assignRequest.AssignedTo
ticket.Status = models.TicketStatusInProgress
if err := database.DB.Save(\&ticket).Error; err != nil {
c.JSON(500, gin.H{"error": "指配失败", "message": err.Error()})
return
}
c.JSON(200, gin.H{"message": "工单指配成功"})
}
func (h *TicketHandler) AddComment(c *gin.Context) {
idStr := c.Param("id"),
id, _ := strconv.ParseUint(idStr, 10, 32)
userID, _ := c.Get("user_id").(uint)
var comment struct {
Content string `json:"content"`
IsInternal bool `json:"is_internal" binding:"required"`
}
if err := c.ShouldBindJSON(\&comment); err != nil {
c.JSON(400, gin.H{"error": "参数错误", "message": err.Error()})
return
}
msg := models.TicketMessage{
TicketID: uint(id),
UserID: userID,
Content: comment.Content,
IsInternal: comment.IsInternal,
}
if err := database.DB.Create(\&msg).Error; err != nil {
c.JSON(500, gin.H{"error": "添加备注失败", "message": err.Error()})
return
}
c.JSON(201, gin.H{"message": "备注添加成功", "comment_id": msg.ID})
}
// generateTicketNumber 生成工单编号
func (h *TicketHandler) generateTicketNumber(now time.Time) string {
return now.Format("20060102") + "-" + "TKT" + "00001"
}