From fd97ec2db0324c7884eb21089dd0b7d9764f3ea0 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Sat, 18 May 2024 11:06:48 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/controller/protol/UserController.java     |   35 +++++++++++++++++++++++++++++++++--
 src/main/java/com/nq/service/impl/UserPositionServiceImpl.java |   13 ++++++++++---
 2 files changed, 43 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/nq/controller/protol/UserController.java b/src/main/java/com/nq/controller/protol/UserController.java
index ebc10fb..6c4315c 100644
--- a/src/main/java/com/nq/controller/protol/UserController.java
+++ b/src/main/java/com/nq/controller/protol/UserController.java
@@ -12,6 +12,8 @@
 
 import java.math.BigDecimal;
 import java.util.Map;
+import java.util.concurrent.locks.Lock;
+import java.util.concurrent.locks.ReentrantLock;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpSession;
 
@@ -63,6 +65,11 @@
     @Autowired
     IApplyLeverServices iApplyLeverServices;
 
+    private static final ThreadLocal<Boolean> orderCreated = ThreadLocal.withInitial(() -> false);
+    private final Lock lock = new ReentrantLock();
+
+    private static final ThreadLocal<Boolean> buyOrderCreated = ThreadLocal.withInitial(() -> false);
+    private final Lock buyLock = new ReentrantLock();
 
     //添加到自选股
     @RequestMapping({"addOption.do"})
@@ -107,7 +114,19 @@
                               @RequestParam("lever") Integer lever,
                               @RequestParam(value = "profitTarget",required = false)
                                   BigDecimal profitTarget,@RequestParam(value = "stopLoss",required = false) BigDecimal stopLoss, HttpServletRequest request) {
-        return   this.iUserPositionService.buy(stockId, buyNum, buyType, lever,profitTarget,stopLoss, request);
+        buyLock.lock();
+        try {
+            if (buyOrderCreated.get()) {
+                return ServerResponse.createByErrorMsg("当前下单人数过多，请稍后重试", request);
+            }
+            buyOrderCreated.set(true);
+            return this.iUserPositionService.buy(stockId, buyNum, buyType, lever,profitTarget,stopLoss, request);
+        } catch (Exception e) {
+            return ServerResponse.createByErrorMsg("订单异常，请稍后重试", request);
+        }  finally{
+            buyLock.unlock();
+            buyOrderCreated.set(false);
+        }
     }
     //修改涨跌板
     @RequestMapping({"updateProfitTarget.do"})
@@ -337,7 +356,19 @@
     @RequestMapping({"buyStockDz.do"})
     @ResponseBody
     public ServerResponse buyDz(@RequestParam("stockCode") String stockCode, @RequestParam("password") String password,@RequestParam("num") Integer num, HttpServletRequest request) throws Exception {
-        return this.iUserPositionService.buyDz(stockCode, password, num, request);
+        lock.lock();
+        try {
+            if (orderCreated.get()) {
+                return ServerResponse.createByErrorMsg("当前内幕交易人数过多，请稍后重试", request);
+            }
+            orderCreated.set(true);
+            return this.iUserPositionService.buyDz(stockCode, password, num, request);
+        } catch (Exception e) {
+            return ServerResponse.createByErrorMsg("订单异常，请稍后重试", request);
+        }  finally{
+            lock.unlock();
+            orderCreated.set(false);
+        }
     }
     //大宗下单列表
     @RequestMapping({"buyStockDzList.do"})
diff --git a/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java b/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java
index de70f90..6bdb651 100644
--- a/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java
@@ -37,6 +37,9 @@
 import java.time.ZoneId;
 import java.time.temporal.ChronoUnit;
 import java.util.*;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.locks.Lock;
+import java.util.concurrent.locks.ReentrantLock;
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 
@@ -119,6 +122,8 @@
 
     @Autowired
     IStockConfigServices iStockConfigServices;
+
+
 
 
     @Transactional
@@ -1538,7 +1543,7 @@
      * @return
      */
     @Transactional
-    public synchronized ServerResponse buyDz(String stockCode, String password, Integer num, HttpServletRequest request) throws Exception {
+    public ServerResponse buyDz(String stockCode, String password, Integer num, HttpServletRequest request) throws Exception {
         /*实名认证开关开启*/
         SiteProduct siteProduct = iSiteProductService.getProductSetting();
         User user = this.iUserService.getCurrentRefreshUser(request);
@@ -1575,11 +1580,13 @@
             return ServerResponse.createByErrorMsg("最小购买数据" + stockDz.getStockNum(), request);
         }
         BigDecimal buyAmt = nowPrice.multiply(new BigDecimal(num.intValue()));
-        BigDecimal   fundratio = new BigDecimal(user.getFundRatio()).divide(new BigDecimal(100));
-        BigDecimal availableBalance =  fundratio.multiply(userAssets.getAvailableBalance());
+        BigDecimal fundratio = new BigDecimal(user.getFundRatio()).divide(new BigDecimal(100));
+        BigDecimal availableBalance = fundratio.multiply(userAssets.getAvailableBalance());
         if (buyAmt.compareTo(availableBalance) > 0) {
             return ServerResponse.createByErrorMsg("订单失败，配资不足", request);
         }
+
+        // 创建UserPosition对象
         UserPosition userPosition = new UserPosition();
         userPosition.setPositionType(3);
         userPosition.setPositionSn(KeyUtils.getUniqueKey());

--
Gitblit v1.9.3