From fa0bfaabaecaabde7f7f097af055c9f299665fdf Mon Sep 17 00:00:00 2001
From: ck <851316342@qq.com>
Date: Mon, 21 Dec 2020 20:14:02 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=AF=8F=E6=97=A5=E6=B1=A0?=
 =?UTF-8?q?=E5=AD=907=E4=B8=AA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../jingcheng/template/mapper/AwardRuleMapper.java  |  1 +
 .../jingcheng/template/quartz/ScheduledTask.java    |  5 +++++
 .../template/service/AwardRecordService.java        |  2 ++
 .../service/impl/AwardRecordServiceImpl.java        | 13 ++++++++++++-
 src/main/resources/mapper/AwardRuleMapper.xml       |  7 +++++++
 5 files changed, 27 insertions(+), 1 deletion(-)
 create mode 100644 src/main/resources/mapper/AwardRuleMapper.xml

diff --git a/src/main/java/com/jingcheng/template/mapper/AwardRuleMapper.java b/src/main/java/com/jingcheng/template/mapper/AwardRuleMapper.java
index 2bce8e6..0397321 100644
--- a/src/main/java/com/jingcheng/template/mapper/AwardRuleMapper.java
+++ b/src/main/java/com/jingcheng/template/mapper/AwardRuleMapper.java
@@ -6,4 +6,5 @@ import org.springframework.stereotype.Repository;
 
 @Repository
 public interface AwardRuleMapper extends CommonMapper<AwardRule> {
+    int updateDayAwardTask();
 }
diff --git a/src/main/java/com/jingcheng/template/quartz/ScheduledTask.java b/src/main/java/com/jingcheng/template/quartz/ScheduledTask.java
index ca59706..1189f64 100644
--- a/src/main/java/com/jingcheng/template/quartz/ScheduledTask.java
+++ b/src/main/java/com/jingcheng/template/quartz/ScheduledTask.java
@@ -1,5 +1,7 @@
 package com.jingcheng.template.quartz;
 
+import com.jingcheng.template.mapper.AwardRuleMapper;
+import com.jingcheng.template.service.AwardRecordService;
 import com.jingcheng.template.service.UsersService;
 import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Component;
@@ -15,6 +17,8 @@ public class ScheduledTask {
 
     @Resource
     UsersService usersService;
+    @Resource
+    AwardRecordService awardRecordService;
 
     private static final String CHANCE_TASK = "0 0 0 * * ?";
     /**
@@ -24,5 +28,6 @@ public class ScheduledTask {
     @Scheduled(cron = CHANCE_TASK)
     public void downloadWxEmployee() {
         usersService.chancesTask();
+        awardRecordService.refreshDayAwards();
     }
 }
diff --git a/src/main/java/com/jingcheng/template/service/AwardRecordService.java b/src/main/java/com/jingcheng/template/service/AwardRecordService.java
index 178a02f..4ae1130 100644
--- a/src/main/java/com/jingcheng/template/service/AwardRecordService.java
+++ b/src/main/java/com/jingcheng/template/service/AwardRecordService.java
@@ -54,4 +54,6 @@ public interface AwardRecordService<AwardRecord> extends BaseService<AwardRecord
      * @return
      */
     AjaxResult isHaveDayShare(String userId);
+
+    void refreshDayAwards();
 }
diff --git a/src/main/java/com/jingcheng/template/service/impl/AwardRecordServiceImpl.java b/src/main/java/com/jingcheng/template/service/impl/AwardRecordServiceImpl.java
index af5043a..e748515 100644
--- a/src/main/java/com/jingcheng/template/service/impl/AwardRecordServiceImpl.java
+++ b/src/main/java/com/jingcheng/template/service/impl/AwardRecordServiceImpl.java
@@ -106,10 +106,12 @@ public class AwardRecordServiceImpl extends BaseServiceImpl<AwardRecordMapper, A
         //今日抽总次数
         Integer todayCount = awardRecordMapper.selectCountByExample(awardRecordExample2);
         Integer awardNum = awardRule.getAwardNum();
+        AwardRule todayAward = awardRuleMapper.selectByPrimaryKey(2L);
+        Integer todayAwardNum = todayAward.getAwardNum();
         if (todayAwardCount >= 1){
             logger.info("此用户已抽中过奖品");
             isAward = false;
-        }else {
+        }else if (todayAwardNum > 0){
             int range =  (int) (1 / awardRate);
             int num = 1 + (int)(Math.random()*(range - 1 + 1));
             if ( 0 == isHaveBigAward){
@@ -150,6 +152,8 @@ public class AwardRecordServiceImpl extends BaseServiceImpl<AwardRecordMapper, A
                 awardRule.setUpdateDateTime(new Date());
             }else {
                 awardRule.setAwardNum(awardNum - 1);
+                todayAward.setAwardNum(todayAwardNum -1);
+                awardRuleMapper.updateByPrimaryKeySelective(todayAward);
             }
             awardRuleMapper.updateByPrimaryKeySelective(awardRule);
         }else {
@@ -396,4 +400,11 @@ public class AwardRecordServiceImpl extends BaseServiceImpl<AwardRecordMapper, A
         }
         return ajaxResult;
     }
+
+    @Override
+    public void refreshDayAwards() {
+        SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd");
+        logger.info(simpleDateFormat.format(new Date())+ "刷新每日中奖池");
+        awardRuleMapper.updateDayAwardTask();
+    }
 }
diff --git a/src/main/resources/mapper/AwardRuleMapper.xml b/src/main/resources/mapper/AwardRuleMapper.xml
new file mode 100644
index 0000000..57db020
--- /dev/null
+++ b/src/main/resources/mapper/AwardRuleMapper.xml
@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="com.jingcheng.template.mapper.AwardRuleMapper">
+    <update id="updateDayAwardTask">
+        update award_rule set award_num = 7 where id = 2
+    </update>
+</mapper>