From d00e00aa93bce9d52e2e3a9fa2f8057184f4dd75 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Thu, 11 Jul 2024 14:40:09 +0800
Subject: [PATCH] 1
---
src/main/java/com/nq/controller/protol/UserController.java | 35 +++++++++++++++++++++++++++++++++--
1 files changed, 33 insertions(+), 2 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"})
--
Gitblit v1.9.3