feat: rbac迁移完成,并已部署至dev服务器
This commit is contained in:
@@ -3,10 +3,13 @@ package logic
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"github.com/zeromicro/go-zero/core/logx"
|
||||
|
||||
sysModel "sundynix-micro-go/app/system/model"
|
||||
"sundynix-micro-go/app/system/rpc/internal/svc"
|
||||
"sundynix-micro-go/app/system/rpc/system"
|
||||
|
||||
"github.com/zeromicro/go-zero/core/logx"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type UpdateRoleLogic struct {
|
||||
@@ -20,15 +23,26 @@ func NewUpdateRoleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Update
|
||||
}
|
||||
|
||||
func (l *UpdateRoleLogic) UpdateRole(in *system.RoleUpdateReq) (*system.CommonResp, error) {
|
||||
if err := l.svcCtx.DB.Model(&sysModel.SundynixRole{}).Where("id = ?", in.Id).Updates(map[string]interface{}{
|
||||
"name": in.Name, "code": in.Code, "sort": in.Sort,
|
||||
}).Error; err != nil {
|
||||
return nil, fmt.Errorf("更新角色失败")
|
||||
}
|
||||
// 更新菜单关联
|
||||
l.svcCtx.DB.Where("role_id = ?", in.Id).Delete(&sysModel.SundynixRoleMenu{})
|
||||
for _, mid := range in.MenuIds {
|
||||
l.svcCtx.DB.Create(&sysModel.SundynixRoleMenu{RoleID: in.Id, MenuID: mid})
|
||||
err := l.svcCtx.DB.Transaction(func(tx *gorm.DB) error {
|
||||
if err := tx.Model(&sysModel.SundynixRole{}).Where("id = ?", in.Id).Updates(map[string]interface{}{
|
||||
"name": in.Name, "code": in.Code, "sort": in.Sort,
|
||||
}).Error; err != nil {
|
||||
return fmt.Errorf("更新角色失败: %w", err)
|
||||
}
|
||||
// 先删除旧的菜单关联
|
||||
if err := tx.Where("role_id = ?", in.Id).Delete(&sysModel.SundynixRoleMenu{}).Error; err != nil {
|
||||
return fmt.Errorf("清除菜单关联失败: %w", err)
|
||||
}
|
||||
// 重新关联菜单
|
||||
for _, mid := range in.MenuIds {
|
||||
if err := tx.Create(&sysModel.SundynixRoleMenu{RoleID: in.Id, MenuID: mid}).Error; err != nil {
|
||||
return fmt.Errorf("关联菜单失败: %w", err)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &system.CommonResp{Code: 200, Msg: "success"}, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user