diff --git a/src/main/java/com/budwk/app/zhgh/activity/trainSignUp/controller/mobile/TrainScanCodeController.java b/src/main/java/com/budwk/app/zhgh/activity/trainSignUp/controller/mobile/TrainScanCodeController.java index 404a1e3..d5abd3c 100644 --- a/src/main/java/com/budwk/app/zhgh/activity/trainSignUp/controller/mobile/TrainScanCodeController.java +++ b/src/main/java/com/budwk/app/zhgh/activity/trainSignUp/controller/mobile/TrainScanCodeController.java @@ -2,8 +2,10 @@ package com.budwk.app.zhgh.activity.trainSignUp.controller.mobile; import cn.dev33.satoken.annotation.SaCheckPermission; import cn.hutool.core.date.DateUtil; +import cn.hutool.core.util.ObjectUtil; import com.aspose.slides.internal.og.add; import com.budwk.app.base.result.Result; +import com.budwk.app.sys.models.Sys_user; import com.budwk.app.web.commons.auth.utils.SecurityUtil; import com.budwk.app.zhgh.activity.trainSignUp.models.TrainSignUpCourse; import com.budwk.app.zhgh.activity.trainSignUp.models.TrainSignUpUser; @@ -45,6 +47,12 @@ public class TrainScanCodeController { @At @SaCheckPermission("h5.trainSignUp.sign") public Result getSignData(@Param("courseId") String courseId) { + String userId = SecurityUtil.getUserId(); + Sys_user user = dao.fetch(Sys_user.class, userId); + if (!(user.getMember() || List.of("年薪制", "境外聘用", "劳务派遣一类", "劳务派遣二类").contains(ObjectUtil.defaultIfBlank(user.getPreparedBy(), "")))) { + return Result.error(99, "您没有权限参与此次活动"); + } + TrainSignUpCourse course = dao.fetch(TrainSignUpCourse.class, courseId); int count = dao.count(TrainSignUpUserCourse.class, Cnd.where("activityId", "=", course.getActivityId()) diff --git a/src/main/java/com/budwk/app/zhgh/activity/trainSignUp/scanHandle/TrainScanHandle.java b/src/main/java/com/budwk/app/zhgh/activity/trainSignUp/scanHandle/TrainScanHandle.java index 35b5d04..ac4abc9 100644 --- a/src/main/java/com/budwk/app/zhgh/activity/trainSignUp/scanHandle/TrainScanHandle.java +++ b/src/main/java/com/budwk/app/zhgh/activity/trainSignUp/scanHandle/TrainScanHandle.java @@ -1,6 +1,7 @@ package com.budwk.app.zhgh.activity.trainSignUp.scanHandle; import cn.hutool.core.date.DateUtil; +import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.StrUtil; import com.budwk.app.base.result.Result; import com.budwk.app.sys.models.Sys_user; @@ -43,11 +44,6 @@ public class TrainScanHandle implements ScanHandler { @Override public Object handle(Map params) { try { - String userId = SecurityUtil.getUserId(); - Sys_user user = dao.fetch(Sys_user.class, userId); - if (!(user.getMember() && List.of("年薪制", "境外聘用", "劳务派遣一类", "劳务派遣二类").contains(user.getPreparedBy()))) { - return Result.error("您没有权限参与此次活动"); - } // 获取参数 String courseId = params.get("courseId").toString();