diff --git a/dl-module-jx/src/main/java/cn/iocoder/yudao/module/base/service/impl/DlDriveSchoolCoachServiceImpl.java b/dl-module-jx/src/main/java/cn/iocoder/yudao/module/base/service/impl/DlDriveSchoolCoachServiceImpl.java index 3bae8c64..78ac273a 100644 --- a/dl-module-jx/src/main/java/cn/iocoder/yudao/module/base/service/impl/DlDriveSchoolCoachServiceImpl.java +++ b/dl-module-jx/src/main/java/cn/iocoder/yudao/module/base/service/impl/DlDriveSchoolCoachServiceImpl.java @@ -1,5 +1,6 @@ package cn.iocoder.yudao.module.base.service.impl; +import cn.iocoder.yudao.common.SchoolRoleEnum; import cn.iocoder.yudao.framework.common.util.object.BeanUtils; import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO; import cn.iocoder.yudao.module.base.entity.DlDriveSchoolCoach; @@ -117,11 +118,15 @@ public class DlDriveSchoolCoachServiceImpl extends ServiceImpl roleCodes = new HashSet<>(); - // TODO: 2025/1/16 枚举值问老马(保存角色方法不对) + if ("jl".equals(createReqVO.getType())) { + roleCodes.add(SchoolRoleEnum.COACH.getCode()); + } else if ("yg".equals(createReqVO.getType())){ + roleCodes.add(SchoolRoleEnum.STAFF.getCode()); + } permissionApi.assignUserRole(userId, roleCodes); } else { userId = userDTO.getId();