huang
eaa70ac255
主要功能: - 实现完整的 RBAC 权限系统（账号、角色、权限的多对多关联） - 基于 owner_id + shop_id 的自动数据权限过滤 - 使用 PostgreSQL WITH RECURSIVE 查询下级账号 - Redis 缓存优化下级账号查询性能（30分钟过期） - 支持多租户数据隔离和层级权限管理 技术实现: - 新增 Account、Role、Permission 模型及关联关系表 - 实现 GORM Scopes 自动应用数据权限过滤 - 添加数据库迁移脚本（000002_rbac_data_permission、000003_add_owner_id_shop_id） - 完善错误码定义（1010-1027 为 RBAC 相关错误） - 重构 main.go 采用函数拆分提高可读性 测试覆盖: - 添加 Account、Role、Permission 的集成测试 - 添加数据权限过滤的单元测试和集成测试 - 添加下级账号查询和缓存的单元测试 - 添加 API 回归测试确保向后兼容 文档更新: - 更新 README.md 添加 RBAC 功能说明 - 更新 CLAUDE.md 添加技术栈和开发原则 - 添加 docs/004-rbac-data-permission/ 功能总结和使用指南 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
119 lines
3.0 KiB
Go
119 lines
3.0 KiB
Go
package handler
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/break/junhong_cmp_fiber/internal/model"
|
|
"github.com/gofiber/fiber/v2"
|
|
|
|
"github.com/break/junhong_cmp_fiber/pkg/errors"
|
|
"github.com/break/junhong_cmp_fiber/pkg/response"
|
|
|
|
permissionService "github.com/break/junhong_cmp_fiber/internal/service/permission"
|
|
)
|
|
|
|
// PermissionHandler 权限 Handler
|
|
type PermissionHandler struct {
|
|
service *permissionService.Service
|
|
}
|
|
|
|
// NewPermissionHandler 创建权限 Handler
|
|
func NewPermissionHandler(service *permissionService.Service) *PermissionHandler {
|
|
return &PermissionHandler{service: service}
|
|
}
|
|
|
|
// Create 创建权限
|
|
// POST /api/v1/permissions
|
|
func (h *PermissionHandler) Create(c *fiber.Ctx) error {
|
|
var req model.CreatePermissionRequest
|
|
if err := c.BodyParser(&req); err != nil {
|
|
return errors.New(errors.CodeInvalidParam, "请求参数解析失败")
|
|
}
|
|
|
|
permission, err := h.service.Create(c.UserContext(), &req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return response.Success(c, permission)
|
|
}
|
|
|
|
// Get 获取权限详情
|
|
// GET /api/v1/permissions/:id
|
|
func (h *PermissionHandler) Get(c *fiber.Ctx) error {
|
|
id, err := strconv.ParseUint(c.Params("id"), 10, 64)
|
|
if err != nil {
|
|
return errors.New(errors.CodeInvalidParam, "无效的权限 ID")
|
|
}
|
|
|
|
permission, err := h.service.Get(c.UserContext(), uint(id))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return response.Success(c, permission)
|
|
}
|
|
|
|
// Update 更新权限
|
|
// PUT /api/v1/permissions/:id
|
|
func (h *PermissionHandler) Update(c *fiber.Ctx) error {
|
|
id, err := strconv.ParseUint(c.Params("id"), 10, 64)
|
|
if err != nil {
|
|
return errors.New(errors.CodeInvalidParam, "无效的权限 ID")
|
|
}
|
|
|
|
var req model.UpdatePermissionRequest
|
|
if err := c.BodyParser(&req); err != nil {
|
|
return errors.New(errors.CodeInvalidParam, "请求参数解析失败")
|
|
}
|
|
|
|
permission, err := h.service.Update(c.UserContext(), uint(id), &req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return response.Success(c, permission)
|
|
}
|
|
|
|
// Delete 删除权限
|
|
// DELETE /api/v1/permissions/:id
|
|
func (h *PermissionHandler) Delete(c *fiber.Ctx) error {
|
|
id, err := strconv.ParseUint(c.Params("id"), 10, 64)
|
|
if err != nil {
|
|
return errors.New(errors.CodeInvalidParam, "无效的权限 ID")
|
|
}
|
|
|
|
if err := h.service.Delete(c.UserContext(), uint(id)); err != nil {
|
|
return err
|
|
}
|
|
|
|
return response.Success(c, nil)
|
|
}
|
|
|
|
// List 查询权限列表
|
|
// GET /api/v1/permissions
|
|
func (h *PermissionHandler) List(c *fiber.Ctx) error {
|
|
var req model.PermissionListRequest
|
|
if err := c.QueryParser(&req); err != nil {
|
|
return errors.New(errors.CodeInvalidParam, "请求参数解析失败")
|
|
}
|
|
|
|
permissions, total, err := h.service.List(c.UserContext(), &req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return response.SuccessWithPagination(c, permissions, total, req.Page, req.PageSize)
|
|
}
|
|
|
|
// GetTree 获取权限树
|
|
// GET /api/v1/permissions/tree
|
|
func (h *PermissionHandler) GetTree(c *fiber.Ctx) error {
|
|
tree, err := h.service.GetTree(c.UserContext())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return response.Success(c, tree)
|
|
}
|