|
@@ -95,13 +95,7 @@ func (r *Role) List(ctx context.Context, req entity.RoleListReq) (*entity.RoleRe
|
|
// Add 新增角色
|
|
// Add 新增角色
|
|
func (r *Role) Add(ctx context.Context, req entity.RoleAddReq) *code.Result {
|
|
func (r *Role) Add(ctx context.Context, req entity.RoleAddReq) *code.Result {
|
|
// 验证角色名称是否已存在
|
|
// 验证角色名称是否已存在
|
|
- exist, err := r.CheckRoleNameExist(req.Name)
|
|
|
|
- if err != nil {
|
|
|
|
- mhayaLogger.Warnf("Add CheckRoleNameExist error:%v", ctx.Err())
|
|
|
|
- return common.NewResult(code.InternalError)
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if exist {
|
|
|
|
|
|
+ if !r.checkRoleNameExist(req.Name) {
|
|
return common.NewResult(code.RoleNameExistError)
|
|
return common.NewResult(code.RoleNameExistError)
|
|
}
|
|
}
|
|
|
|
|
|
@@ -161,8 +155,8 @@ func (r *Role) Update(ctx context.Context, req entity.RoleUpdateReq) *code.Resul
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
|
|
|
|
-// CheckRoleNameExist 检查角色名称是否已存在
|
|
|
|
-func (r *Role) CheckRoleNameExist(name string) (bool, error) {
|
|
|
|
|
|
+// checkRoleNameExist 检查角色名称是否已存在
|
|
|
|
+func (r *Role) checkRoleNameExist(name string) bool {
|
|
// 创建带超时的上下文
|
|
// 创建带超时的上下文
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
defer cancel()
|
|
@@ -173,17 +167,14 @@ func (r *Role) CheckRoleNameExist(name string) (bool, error) {
|
|
filter := bson.M{"name": name}
|
|
filter := bson.M{"name": name}
|
|
|
|
|
|
// 执行查询
|
|
// 执行查询
|
|
- err := collection.FindOne(ctx, filter).Err()
|
|
|
|
- if err != nil {
|
|
|
|
|
|
+ if err := collection.FindOne(ctx, filter).Err(); err != nil {
|
|
if errors.Is(err, mongo.ErrNoDocuments) {
|
|
if errors.Is(err, mongo.ErrNoDocuments) {
|
|
mhayaLogger.Warnf("No document found with role name: %s", name)
|
|
mhayaLogger.Warnf("No document found with role name: %s", name)
|
|
- return false, nil
|
|
|
|
|
|
+ return true
|
|
}
|
|
}
|
|
-
|
|
|
|
- return false, err
|
|
|
|
|
|
+ return false
|
|
}
|
|
}
|
|
-
|
|
|
|
- return true, nil
|
|
|
|
|
|
+ return false
|
|
}
|
|
}
|
|
|
|
|
|
// Del 删除角色
|
|
// Del 删除角色
|