From bd21cffed808eff59e330de7142c59af6b48e169 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Tue, 01 Oct 2024 19:17:52 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/utils/task/stock/CarryPositionTask.java |   29 +++++++++++++++++------------
 1 files changed, 17 insertions(+), 12 deletions(-)

diff --git a/src/main/java/com/nq/utils/task/stock/CarryPositionTask.java b/src/main/java/com/nq/utils/task/stock/CarryPositionTask.java
index 1178088..8fd088b 100644
--- a/src/main/java/com/nq/utils/task/stock/CarryPositionTask.java
+++ b/src/main/java/com/nq/utils/task/stock/CarryPositionTask.java
@@ -23,6 +23,7 @@
 
 import java.math.BigDecimal;
 import java.util.List;
+import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.locks.Lock;
 import java.util.concurrent.locks.ReentrantLock;
 import java.util.stream.Collectors;
@@ -56,15 +57,17 @@
 
     private final Lock subscriptionLock = new ReentrantLock();
 
-    boolean ballot = true;
+    private final AtomicBoolean ballot = new AtomicBoolean(false);
 
-    @Scheduled(cron = "0 0/10 * * * ?")
+
+//    @Scheduled(cron = "0 0/10 * * * ?")
     public void ballot() {
-        if (ballot)
-            ballot = false;
-        else return;
+        if (ballot.get()) { // 判断任务是否在处理中
+            return;
+        }
         if (ballotLock.tryLock()) {
             try {
+                ballot.set(true); // 设置处理中标识为true
                 log.info("新股上市转持仓定时任务--------->开始");
                 List<StockSubscribe> stockSubscribes = stockSubscribeMapper.selectList(new LambdaQueryWrapper<StockSubscribe>()
                         .lt(StockSubscribe::getListDate, DateUtil.date()));
@@ -74,7 +77,7 @@
                             .eq(UserStockSubscribe::getStatus, 4).in(UserStockSubscribe::getNewCode, codeList));
                     //订单转持仓
                     userStockSubscribes.forEach(f -> {
-                        ServerResponse serverResponse = iUserPositionService.newStockToPosition(f.getId());//转持仓
+                        ServerResponse serverResponse = iUserPositionService.newStockToPosition(f.getId(),BigDecimal.ZERO);//转持仓
                         if(serverResponse.isSuccess()){
                             f.setStatus(5);
                             userStockSubscribeMapper.updateById(f);
@@ -88,21 +91,23 @@
                 log.error("新股上市转持仓定时任务发生异常", e);
             } finally {
                 ballotLock.unlock();
-                ballot = true;
+                ballot.set(false); // 设置处理中标识为false
             }
         } else {
             log.info("新股上市转持仓定时任务--------->上次任务还未执行完成,本次任务忽略");
         }
     }
 
-    boolean subscription = true;
+    private final AtomicBoolean subscription = new AtomicBoolean(false);
+
     @Scheduled(cron = "0 0/1 * * * ?")
     public void subscription() {
-        if (subscription)
-            subscription = false;
-        else return;
+        if (subscription.get()) { // 判断任务是否在处理中
+            return;
+        }
         if (subscriptionLock.tryLock()) {
             try {
+                subscription.set(true); // 设置处理中标识为true
                 log.info("自动转已认缴--------->开始");
                 List<UserStockSubscribe> userStockSubscribes = userStockSubscribeMapper.selectList(new LambdaQueryWrapper<UserStockSubscribe>()
                         .eq(UserStockSubscribe::getStatus, 3));
@@ -118,7 +123,7 @@
                 log.error("自动转已认缴定时任务发生异常", e);
             } finally {
                 subscriptionLock.unlock();
-                subscription = true;
+                subscription.set(false); // 设置处理中标识为false
             }
         } else {
             log.info("自动转已认缴定时任务--------->上次任务还未执行完成,本次任务忽略");

--
Gitblit v1.9.3