bugfix-209: 积分规则列表添加编号

yudao-module-member/yudao-module-member-impl/src/main/java/cn/iocoder/yudao/module/member/controller/admin/job/ScoreRuleCodeFlushTask.java

+package cn.iocoder.yudao.module.member.controller.admin.job;
+
+import cn.iocoder.yudao.framework.quartz.core.handler.JobHandler;
+import cn.iocoder.yudao.module.member.util.ScoreRuleGenCodeUtils;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.Resource;
+
+/**
+ * 积分规则编号刷新任务
+ * @author zhaobiyan
+ */
+@Component
+@Slf4j
+public class ScoreRuleCodeFlushTask implements JobHandler {
+
+    @Resource
+    private ScoreRuleGenCodeUtils scoreRuleGenCodeUtils;
+
+    @Override
+    public String execute(String param) throws Exception {
+        scoreRuleGenCodeUtils.historyScoreRuleCodeFlush();
+        return "success";
+    }
+}
\ No newline at end of file

yudao-module-member/yudao-module-member-impl/src/main/java/cn/iocoder/yudao/module/member/dal/dataobject/scoreRule/ScoreRuleDO.java

@@ -24,6 +24,10 @@ public class ScoreRuleDO extends BaseDO {
      */
     @TableId
     private Long id;
+    /**
+     * 规则编号
+     */
+    private String code;
     /**
      * 指标类型
      */

yudao-module-member/yudao-module-member-impl/src/main/java/cn/iocoder/yudao/module/member/dal/mysql/scoreRule/ScoreRuleMapper.java

@@ -11,6 +11,8 @@ import cn.iocoder.yudao.framework.mybatis.core.vo.PageVO;
 import cn.iocoder.yudao.module.member.enums.RelationSymbolEnum;
 import cn.iocoder.yudao.module.member.vo.scoreRule.ScoreRuleQueryVO;
 import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.ResultType;
+import org.apache.ibatis.annotations.Select;
 
 /**
  * 积分规则 Mapper
@@ -69,4 +71,7 @@ public interface ScoreRuleMapper extends AbstractMapper<ScoreRuleDO> {
         }
         return null;
     }
+    @ResultType(String.class)
+    @Select({"select code from score_rule order by code desc limit 1"})
+    String getCurrentMaxScoreRuleCode();
 }

yudao-module-member/yudao-module-member-impl/src/main/java/cn/iocoder/yudao/module/member/service/scoreRule/ScoreRuleServiceImpl.java

@@ -21,6 +21,7 @@ import cn.iocoder.yudao.module.member.enums.ScoreRuleStatusEnum;
 import cn.iocoder.yudao.module.member.enums.ScoreRuleTypeEnum;
 import cn.iocoder.yudao.module.member.enums.TransportTypeEnum;
 import cn.iocoder.yudao.module.member.enums.YesOrNoTypeEnum;
+import cn.iocoder.yudao.module.member.util.ScoreRuleGenCodeUtils;
 import cn.iocoder.yudao.module.member.vo.scoreRule.*;
 import cn.iocoder.yudao.module.member.vo.scoreRule.extra.ScoreRuleOrderVExtraVO;
 import cn.iocoder.yudao.module.member.vo.scoreRule.extra.ScoreRuleRegisterExtraVO;
@@ -60,6 +61,8 @@ public class ScoreRuleServiceImpl extends AbstractService<ScoreRuleMapper, Score
     private WarehouseService warehouseService;
     @Resource
     private AdminUserApi adminUserApi;
+    @Resource
+    private ScoreRuleGenCodeUtils scoreRuleGenCodeUtils;
 
     @Override
     public Long createScoreRule(ScoreRuleCreateReqVO createReqVO) {
@@ -70,6 +73,7 @@ public class ScoreRuleServiceImpl extends AbstractService<ScoreRuleMapper, Score
         verifyCommon(createReqVO);
         Integer scoreRuleType = scoreRule.getType();
         verifyAndSetExtraDO(createReqVO, scoreRuleType, scoreRule);
+        scoreRule.setCode(scoreRuleGenCodeUtils.generateScoreRuleCode());
         scoreRuleMapper.insert(scoreRule);
         scoreRuleRedisDao.deleteEnableScoreRule(scoreRule.getType());
         // 返回
@@ -367,6 +371,7 @@ public class ScoreRuleServiceImpl extends AbstractService<ScoreRuleMapper, Score
         scoreRuleDO.setCreateTime(null);
         scoreRuleDO.setUpdateTime(null);
         scoreRuleDO.setUpdater(null);
+        scoreRuleDO.setCode(scoreRuleGenCodeUtils.generateScoreRuleCode());
         scoreRuleMapper.insert(scoreRuleDO);
         return scoreRuleDO.getId();
     }
@@ -449,9 +454,10 @@ public class ScoreRuleServiceImpl extends AbstractService<ScoreRuleMapper, Score
             case RECOMMEND:
             case REGISTER:
                 return getEnableScoreRuleByType(scoreRuleType);
-        }
+            default:
                 return null;
         }
+    }
 
     @Override
     public List<WarehouseTreeRegionSimpleVO> getWarehouseTreeRegionListSimple() {

yudao-module-member/yudao-module-member-impl/src/main/java/cn/iocoder/yudao/module/member/util/ScoreRuleGenCodeUtils.java

+package cn.iocoder.yudao.module.member.util;
+
+import cn.iocoder.yudao.framework.redis.helper.RedisHelper;
+import cn.iocoder.yudao.module.member.dal.dataobject.scoreRule.ScoreRuleDO;
+import cn.iocoder.yudao.module.member.dal.mysql.scoreRule.ScoreRuleMapper;
+import com.alibaba.excel.util.DateUtils;
+import org.redisson.api.RLock;
+import org.redisson.api.RedissonClient;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.Resource;
+import java.util.Date;
+import java.util.List;
+import java.util.concurrent.TimeUnit;
+
+import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
+import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.GET_LOCK_FAILED;
+
+/**
+ * 根据规则生成各种code
+ */
+@Component
+public class ScoreRuleGenCodeUtils {
+
+    @Resource
+    private RedisHelper redisHelper;
+    @Resource
+    private RedissonClient redissonClient;
+    @Resource
+    private ScoreRuleMapper scoreRuleMapper;
+
+    /**
+     * 生成积分规则编号
+     */
+    public String generateScoreRuleCode() {
+        // 编号规则为PR+年份+5位数，例如PR2400001
+        String key = "scoreRule:max:number";
+        StringBuilder code = new StringBuilder();
+        code.append("PR");
+        code.append(DateUtils.format(new Date(), "yy"));
+        Long codeNum;
+        if (redisHelper.hasKey(key)) {
+            codeNum = redisHelper.incrBy(key, 1);
+        } else {
+            RLock lock = redissonClient.getLock("next:scoreRule:code:lock");
+            try {
+                boolean lockSuccess = lock.tryLock(2, 2, TimeUnit.SECONDS);
+                if (!lockSuccess) {
+                    throw exception(GET_LOCK_FAILED);
+                }
+                String currentMaxScoreRuleCode = scoreRuleMapper.getCurrentMaxScoreRuleCode();
+                if (currentMaxScoreRuleCode == null){
+                    codeNum = 1L;
+                } else {
+                    codeNum = Long.parseLong(currentMaxScoreRuleCode.substring(4)) + 1;
+                }
+                redisHelper.set(key, String.valueOf(codeNum),10, TimeUnit.SECONDS);
+            } catch (InterruptedException e) {
+                throw exception(GET_LOCK_FAILED);
+            } finally {
+                lock.unlock();
+            }
+        }
+        // 获得5位序列号，不足位前面补0
+        code.append(String.format("%05d", codeNum));
+        return code.toString();
+    }
+    public void historyScoreRuleCodeFlush() {
+        long start = 1L;
+        redisHelper.delete("scoreRule:max:number");
+        List<ScoreRuleDO> scoreRuleDOS = scoreRuleMapper.selectList();
+        for (ScoreRuleDO scoreRuleDO : scoreRuleDOS) {
+            StringBuilder code = new StringBuilder();
+            code.append("PR");
+            code.append(DateUtils.format(new Date(), "yy"));
+            code.append(String.format("%05d", start));
+            scoreRuleDO.setCode(code.toString());
+            scoreRuleMapper.updateById(scoreRuleDO);
+            start += 1;
+        }
+    }
+}

yudao-module-member/yudao-module-member-impl/src/main/java/cn/iocoder/yudao/module/member/vo/scoreRule/ScoreRuleBackVO.java

@@ -22,6 +22,10 @@ public class ScoreRuleBackVO {
     @ApiModelProperty(value = "主键", required = true)
     private Long id;
 
+    @ExcelProperty("编号")
+    @ApiModelProperty(value = "编号", required = true)
+    private String code;
+
     @ExcelProperty("指标类型")
     @ApiModelProperty(value = "指标类型", required = true)
     private Integer type;