From c3e4a21800b47d4b8a8ae0a76458a6892bee677c Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Mon, 05 Aug 2024 13:59:28 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/controller/protol/UserController.java |   61 ++++++++++++++++++++++++++++--
 1 files changed, 56 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 b2ac108..6c4315c 100644
--- a/src/main/java/com/nq/controller/protol/UserController.java
+++ b/src/main/java/com/nq/controller/protol/UserController.java
@@ -4,6 +4,7 @@
 import com.google.common.collect.Maps;
 import com.nq.common.ServerResponse;
 import com.nq.enums.EStockType;
+import com.nq.pojo.ApplyLever;
 import com.nq.pojo.StockSubscribe;
 import com.nq.pojo.UserStockSubscribe;
 import com.nq.service.*;
@@ -11,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;
 
@@ -57,6 +60,17 @@
     @Autowired
     IUserRechargeService iUserRechargeService;
 
+
+
+    @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"})
     @ResponseBody
@@ -99,8 +113,20 @@
                               @RequestParam("buyType") Integer buyType,
                               @RequestParam("lever") Integer lever,
                               @RequestParam(value = "profitTarget",required = false)
-                                  BigDecimal profitTarget,@RequestParam(value = "stopTarget",required = false) BigDecimal stopTarget, HttpServletRequest request) {
-        return   this.iUserPositionService.buy(stockId, buyNum, buyType, lever,profitTarget,stopTarget, request);
+                                  BigDecimal profitTarget,@RequestParam(value = "stopLoss",required = false) BigDecimal stopLoss, HttpServletRequest 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"})
@@ -267,8 +293,14 @@
      */
     @RequestMapping({"list.do"})
     @ResponseBody
-    public ServerResponse list(@RequestParam(value = "name", required = false) String name,@RequestParam(value = "code", required = false) String code,@RequestParam(value = "zt", required = false) Integer zt,@RequestParam(value = "isLock", required = false) Integer isLock,@RequestParam(value = "type", required = false) Integer type, HttpServletRequest request) {
-        return this.iStockSubscribeService.list(name,code,zt,isLock,type,request);
+    public ServerResponse list(@RequestParam(value = "name", required = false) String name,
+                               @RequestParam(value = "code", required = false) String code,
+                               @RequestParam(value = "zt", required = false) Integer zt,
+                               @RequestParam(value = "isLock", required = false) Integer isLock,
+                               @RequestParam(value = "type", required = false) Integer type,
+                               @RequestParam(value = "getType", required = false) Integer getType,
+                               HttpServletRequest request) {
+        return this.iStockSubscribeService.list(name,code,zt,getType,isLock,type,request);
     }
     /**
      * 新股申购 添加
@@ -324,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"})
@@ -350,4 +394,11 @@
     }
 
 
+    @RequestMapping({"applyLever.do"})
+    @ResponseBody
+    public ServerResponse applyLever(ApplyLever applyLever,HttpServletRequest request) {
+        return iApplyLeverServices.applyLever(applyLever,request);
+    }
+
+
 }

--
Gitblit v1.9.3