feat: RBAC 基本完成

service/system/sys_role.go

@@ -2,11 +2,12 @@ package system
 
 import (
 	"errors"
-	"gorm.io/gorm"
 	"sundynix-go/global"
 	common "sundynix-go/model/commom/request"
 	"sundynix-go/model/system"
 	systemReq "sundynix-go/model/system/request"
+
+	"gorm.io/gorm"
 )
 
 type RoleService struct {
@@ -48,7 +49,41 @@ func (s *RoleService) DeleteRoleByIds(ids common.IdsReq) error {
 	return global.DB.Where("id in ?", ids.Ids).Delete(&system.Role{}).Error
 }
 
-func (s *RoleService) GetRoleById(id int) (role system.Role, err error) {
-	err = global.DB.Where("id = ?", id).First(&role).Error
-	return role, err
+func (s *RoleService) GetRoleById(id string) (role *system.Role, err error) {
+	var r system.Role
+	err = global.DB.Where("id = ?", id).First(&r).Error
+	return &r, err
+}
+
+func (s *RoleService) GrantRole(userId string, roleIds []string) error {
+	//1. 检查是否存在userid的授权记录 存在就删除 不存在就插入
+	//2. 插入新的数据
+	return global.DB.Transaction(func(tx *gorm.DB) error {
+		if err := tx.Where("user_id = ?", userId).Delete(&system.UserRole{}).Error; err != nil {
+			return err
+		}
+		for _, roleId := range roleIds {
+			if err := tx.Create(&system.UserRole{UserId: userId, RoleId: roleId}).Error; err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+
+}
+
+func (s *RoleService) GrantMenu(roleId string, menuIds []string) error {
+	//1. 检查是否存在userid的授权记录 存在就删除 不存在就插入
+	//2. 插入新的数据
+	return global.DB.Transaction(func(tx *gorm.DB) error {
+		if err := tx.Where("role_id = ?", roleId).Delete(&system.RoleMenu{}).Error; err != nil {
+			return err
+		}
+		for _, menuId := range menuIds {
+			if err := tx.Create(&system.RoleMenu{RoleId: roleId, MenuId: menuId}).Error; err != nil {
+				return err
+			}
+		}
+		return nil
+	})
 }