first commit

service/system/sys_role.go

@@ -0,0 +1,89 @@
+package system
+
+import (
+	"errors"
+	"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 {
+}
+
+var RoleServiceApp = new(RoleService)
+
+func (s *RoleService) SaveRole(role system.Role) error {
+	if !errors.Is(global.DB.Where("code = ?", role.Code).First(&system.Role{}).Error, gorm.ErrRecordNotFound) {
+		return errors.New("存在重复角色")
+	}
+	return global.DB.Create(&role).Error
+}
+
+func (s *RoleService) UpdateRole(role system.Role) error {
+	return global.DB.Model(&role).Where("id = ?", role.Id).Updates(&role).Error
+}
+
+func (s *RoleService) GetRoleList(info systemReq.GetRoleList) (list interface{}, total int64, err error) {
+	limit := info.PageSize
+	offset := info.PageSize * (info.Current - 1)
+	db := global.DB.Model(&system.Role{})
+	var roleList []system.Role
+	if info.Code != "" {
+		db = db.Where("code = ?", info.Code)
+	}
+	if info.Name != "" {
+		db = db.Where("name LIKE ?", "%"+info.Name+"%")
+	}
+	err = db.Count(&total).Error
+	if err != nil {
+		return
+	}
+	err = db.Limit(limit).Offset(offset).Find(&roleList).Error
+	return roleList, total, err
+}
+
+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 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
+	})
+}