From 1774094398c5c6ca376ffb677d0d86fa16c79710 Mon Sep 17 00:00:00 2001
From: xiaoyong931011 <15274802129@163.com>
Date: Wed, 26 Oct 2022 15:52:43 +0800
Subject: [PATCH] 20221021

---
 src/main/java/cc/mrbird/febs/job/OnHookPlanJob.java                             |   13 ++++++++++++-
 src/main/java/cc/mrbird/febs/dapp/service/impl/ApiIgtOnHookPlanServiceImpl.java |   12 +++++++++++-
 src/test/java/cc/mrbird/febs/JunitTest.java                                     |   11 +++++++++++
 3 files changed, 34 insertions(+), 2 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/ApiIgtOnHookPlanServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/ApiIgtOnHookPlanServiceImpl.java
index d63c917..05705df 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/ApiIgtOnHookPlanServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/ApiIgtOnHookPlanServiceImpl.java
@@ -28,6 +28,7 @@
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.aspectj.bridge.MessageUtil;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -49,6 +50,9 @@
     private final RedisUtils redisUtils;
     private final DappOnHookAwardDao dappOnHookAwardDao;
     private final DappWalletService dappWalletService;
+
+    @Value("${spring.profiles.active}")
+    private String active;
     @Override
     @Transactional(rollbackFor = Exception.class)
     public FebsResponse start(ApiStartDto apiStartDto) {
@@ -163,7 +167,13 @@
             throw new FebsException(MessageSourceUtils.getString("login_err_003"));
         }
         //获取下一个开奖
-        String redisKey = "on_hook_award_id_test";
+
+        String redisKey = "";
+        if("prod".equals(active)){
+            redisKey = "on_hook_award_id";
+        }else{
+            redisKey = "on_hook_award_id_test";
+        }
         String onHookAwardRedisId = redisUtils.getString(redisKey);
         DappOnHookAward dappOnHookAward = dappOnHookAwardDao.selectById(Long.parseLong(onHookAwardRedisId));
         if(ObjectUtil.isEmpty(dappOnHookAward)){
diff --git a/src/main/java/cc/mrbird/febs/job/OnHookPlanJob.java b/src/main/java/cc/mrbird/febs/job/OnHookPlanJob.java
index 57d0b83..c88ecba 100644
--- a/src/main/java/cc/mrbird/febs/job/OnHookPlanJob.java
+++ b/src/main/java/cc/mrbird/febs/job/OnHookPlanJob.java
@@ -18,6 +18,7 @@
 import org.apache.commons.lang3.StringUtils;
 import org.jline.utils.Log;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Component;
 
@@ -252,6 +253,11 @@
         }
     }
 
+
+
+    @Value("${spring.profiles.active}")
+    private String active;
+
     @Scheduled(cron = "0 0/5 * * * ? ")
     public void updateAwardInfo(){
         Log.info("更新挂架开奖记录");
@@ -262,7 +268,12 @@
         DappOnHookAward dappOnHookAward = dappOnHookAwardDao.selectByByAwardTime(yyyyMMddHHMM);
         dappOnHookAward.setState(1);
         dappOnHookAwardDao.updateById(dappOnHookAward);
-        String redisKey = "on_hook_award_id_test";
+        String redisKey = "";
+        if("prod".equals(active)){
+            redisKey = "on_hook_award_id";
+        }else{
+            redisKey = "on_hook_award_id_test";
+        }
         String onHookAwardRedisId = redisUtils.getString(redisKey);
         if(StrUtil.isBlank(onHookAwardRedisId)){
             redisUtils.set(redisKey,dappOnHookAward.getId().toString(),0L);
diff --git a/src/test/java/cc/mrbird/febs/JunitTest.java b/src/test/java/cc/mrbird/febs/JunitTest.java
index 9e67254..a187498 100644
--- a/src/test/java/cc/mrbird/febs/JunitTest.java
+++ b/src/test/java/cc/mrbird/febs/JunitTest.java
@@ -29,6 +29,7 @@
 import org.jline.utils.Log;
 import org.junit.jupiter.api.Test;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.boot.test.context.SpringBootTest;
 
 import java.math.BigDecimal;
@@ -237,6 +238,16 @@
 
     @Autowired
     private DappWalletService dappWalletService;
+
+
+    @Value("${spring.profiles.active}")
+    private String active;
+
+    @Test
+    public void getActive(){
+        System.out.println(active);
+    }
+
     @Test
     public void testProfitSharing(){
         IgtOnHookPlanOrder igtOnHookPlanOrder = igtOnHookPlanOrderDao.selectById(129L);

--
Gitblit v1.9.1