Переглянути джерело

演练推优评优状态判断: 删除魔法值,使用枚举值代替

jingyuanchao 1 рік тому
батько
коміт
4e2a2f5ac3

+ 7 - 5
soc-modules/soc-modules-core/src/main/java/com/xunmei/core/drill/service/impl/CoreDrillTaskServiceImpl.java

@@ -1068,15 +1068,17 @@ public class CoreDrillTaskServiceImpl extends ServiceImpl<CoreDrillTaskMapper, C
         }
         //前端传null,说明页面两个复选框都没勾选
         //如果当前是正常案例/行社优秀案例/已推荐地区优秀案例 取消推荐后状态为正常案例
-        if (drillTask.getRecStatus() == 0 || drillTask.getRecStatus() == 1 || drillTask.getRecStatus() == 2) {
-            return 0;
+        if (drillTask.getRecStatus().equals(DrillRecStatus.NORMAL.getCode()) ||
+                drillTask.getRecStatus().equals(DrillRecStatus.HS_EXCELLENT_CASE.getCode()) ||
+                drillTask.getRecStatus().equals(DrillRecStatus.REC_REGION_EXCELLENT_CASE.getCode())) {
+            return DrillRecStatus.NORMAL.getCode();
         }
         //如果当前是地区优秀案例/已推荐省级优秀案例  取消推荐后状态为行社优秀案例
-        if (drillTask.getRecStatus() == 3 || drillTask.getRecStatus() == 4) {
-            return 1;
+        if (drillTask.getRecStatus().equals(DrillRecStatus.REGION_EXCELLENT_CASE.getCode()) || drillTask.getRecStatus().equals(DrillRecStatus.REC_PROVINCE_EXCELLENT_CASE.getCode())) {
+            return DrillRecStatus.HS_EXCELLENT_CASE.getCode();
         }
         //走到这说明当前推荐状态为省级优秀案例,取消推荐后状态为地区优秀案例
-        return 3;
+        return DrillRecStatus.REGION_EXCELLENT_CASE.getCode();
     }
 
     private Boolean checkRecNum(SysOrg sysOrg) {