From 37c4b8d314a93ac866a7886b262346ff8810dc3e Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Wed, 16 Jul 2025 17:57:10 +0800
Subject: [PATCH] bug修改
---
src/main/java/com/nq/service/impl/StockAiServiceImpl.java | 232 ++++++++++++++++++++++++++++++++++++++++++++++++++-------
1 files changed, 203 insertions(+), 29 deletions(-)
diff --git a/src/main/java/com/nq/service/impl/StockAiServiceImpl.java b/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
index ae549fa..b8d869b 100644
--- a/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
@@ -4,13 +4,17 @@
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.nq.Repository.ExchangeRateRepository;
+import com.nq.common.ResponseCode;
import com.nq.common.ServerResponse;
import com.nq.dao.StockAiMapper;
import com.nq.dao.StockAiOrderMapper;
+import com.nq.dao.StockAiOrderPositionMapper;
import com.nq.enums.*;
import com.nq.pojo.*;
import com.nq.service.*;
-import com.nq.vo.stock.StockAiVO;
+import com.nq.vo.stock.ai.StockAiOrderTypeVO;
+import com.nq.vo.stock.ai.StockAiOrderVO;
+import com.nq.vo.stock.ai.StockAiVO;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
@@ -21,23 +25,26 @@
import java.math.BigDecimal;
import java.util.Date;
import java.util.List;
+import java.util.Objects;
+import java.util.concurrent.ConcurrentHashMap;
@Service("iStockAiService")
public class StockAiServiceImpl implements IStockAiService {
private static final Logger log = LoggerFactory.getLogger(StockAiServiceImpl.class);
+ private final ConcurrentHashMap<Long, Object> locks = new ConcurrentHashMap<>();
@Autowired
StockAiMapper stockAiMapper;
@Autowired
StockAiOrderMapper stockAiOrderMapper;
@Autowired
+ StockAiOrderPositionMapper stockAiOrderPositionMapper;
+ @Autowired
ISiteProductService iSiteProductService;
@Autowired
IUserService iUserService;
@Autowired
IUserAssetsServices iUserAssetsServices;
- @Autowired
- ExchangeRateRepository exchangeRateRepository;
/**
* 获取上架ai产品
@@ -72,32 +79,36 @@
try {
User user = iUserService.getCurrentUser(request);
if (user == null ){
- return ServerResponse.createBySuccessMsg("請先登錄");
+ return ServerResponse.createByErrorCodeMsg(ResponseCode.NEED_LOGIN.getCode(),"请先登录");
}
- synchronized (user.getId()){
+ Object lock = locks.computeIfAbsent(Long.valueOf(user.getId()), k -> new Object());
+ synchronized (lock){
SiteProduct siteProduct = iSiteProductService.getProductSetting();
if (siteProduct.getRealNameDisplay() && user.getIsActive() != 2) {
- return ServerResponse.createByErrorMsg("订单失败，请先实名认证", request);
+ return ServerResponse.createByErrorMsg("订单失败，请先实名认证");
}
if (siteProduct.getRealNameDisplay() && user.getIsLock().intValue() == 1) {
- return ServerResponse.createByErrorMsg("订单失败，帐户已被锁定", request);
+ return ServerResponse.createByErrorMsg("订单失败，帐户已被锁定");
}
StockAI stockAI = stockAiMapper.selectById(id);
if (stockAI == null) {
- return ServerResponse.createByErrorMsg("订单失败，AI股票不存在", request);
+ return ServerResponse.createByErrorMsg("订单失败，AI股票不存在");
}
-
+ if (!stockAI.getStatus().equals(EStockAIStatus.online.getValue())) {
+ return ServerResponse.createByErrorMsg("订单失败，AI股票已下架");
+ }
if(buyNum.compareTo(stockAI.getMinPrice()) < 0){
- return ServerResponse.createByErrorMsg("最低购买数量" + stockAI.getMinPrice(), request);
+ return ServerResponse.createByErrorMsg("最低购买数量" + stockAI.getMinPrice());
}
//获取用户账户
UserAssets userAssets = iUserAssetsServices.assetsByTypeAndUserId(EStockType.MX.getCode(), user.getId());
- BigDecimal finalBuyNum = buyNum; //购买金额
+ /*BigDecimal finalBuyNum = buyNum; //购买金额
//如果不是墨西哥币需要转换金额
- if (!stockAI.getStockType().equals(EStockType.US.getCode())) {
+ if (!stockAI.getStockType().equals(EStockType.MX.getCode())) {
EStockType stockType = EStockType.getEStockTypeByCode(stockAI.getStockType());
+
ExchangeRate exchangeRate = exchangeRateRepository.findExchangeRateByCurrencyAndConversionCurrency(stockType.getSymbol(), EStockType.MX.getSymbol())
.orElse(null);
if (exchangeRate == null) {
@@ -105,29 +116,32 @@
}
//转换为墨西哥币
buyNum = iUserAssetsServices.exchangeAmountByRate(buyNum, exchangeRate.getRata());
- }
+ }*/
if(buyNum.compareTo(userAssets.getAvailableBalance()) > 0){
- return ServerResponse.createByErrorMsg("可用余额不足" + userAssets.getAvailableBalance(), request);
+ return ServerResponse.createByErrorMsg("可用余额不足" + userAssets.getAvailableBalance());
}
if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
- return ServerResponse.createByErrorMsg("请先缴清待补资金", request);
+ return ServerResponse.createByErrorMsg("请先缴清待补资金");
}
+ Boolean flag = iUserAssetsServices.aiAvailableBalanceChange(stockAI.getStockType(), user.getId(), EUserAssets.BUY_AI, buyNum, null);
+ if (!flag) {
+ return ServerResponse.createByErrorMsg("扣款失败，无法转换货币");
+ }
StockAIOrder stockAIOrder = new StockAIOrder();
stockAIOrder.setUserId(user.getId());
stockAIOrder.setStockAiId(id);
stockAIOrder.setBuyDate(new Date());
- stockAIOrder.setBuyAmount(finalBuyNum);
- stockAIOrder.setRemainAmount(finalBuyNum);
+ stockAIOrder.setBuyAmount(buyNum);
+ stockAIOrder.setRemainAmount(buyNum);
stockAIOrder.setRealEarning(BigDecimal.valueOf(0));
stockAIOrder.setStatus(EStockAIOrderStatus.wait.getStatus()); //等待审核
stockAiOrderMapper.insert(stockAIOrder);
- iUserAssetsServices.aiAvailableBalanceChange(userAssets, EUserAssets.BUY_AI, buyNum);
- return ServerResponse.createBySuccessMsg("下单成功", request);
+ return ServerResponse.createBySuccessMsg("下单成功");
}
} catch (Exception e) {
- log.error("StockAiService buyStockAiList error", e);
+ log.error("StockAiService buyStockAiList {}", e.getMessage());
}
return ServerResponse.createByError();
}
@@ -145,15 +159,12 @@
try {
User user = iUserService.getCurrentUser(request);
if (user == null ){
- return ServerResponse.createBySuccessMsg("請先登錄");
+ return ServerResponse.createByErrorCodeMsg(ResponseCode.NEED_LOGIN.getCode(),"请先登录");
}
-
PageHelper.startPage(pageNum, pageSize);
- List<StockAIOrder> stockAIOrders = stockAiOrderMapper.selectList(new QueryWrapper<StockAIOrder>()
- .eq("user_id", user.getId())
- .eq(status != null && !status.isEmpty(), "status", status));
+ List<StockAiOrderTypeVO> stockAIOrders = stockAiOrderMapper.getStockAiOrderList(user.getId(), status);
// 获取分页信息
- PageInfo<StockAIOrder> pageInfo = new PageInfo<>(stockAIOrders);
+ PageInfo<StockAiOrderTypeVO> pageInfo = new PageInfo<>(stockAIOrders);
return ServerResponse.createBySuccess(pageInfo);
} catch (Exception e) {
log.error("StockAiService getStockAiOrderList error", e);
@@ -171,15 +182,13 @@
if (!pageInfo.getList().isEmpty()) {
List<StockAiVO> newStockAiList = pageInfo.getList();
- newStockAiList.forEach(stockAiVO -> {
+ pageInfo.getList().forEach(stockAiVO -> {
EStockType eStockType = EStockType.getEStockTypeByCode(stockAiVO.getStockType());
stockAiVO.setStockTypeName(eStockType.getSymbol1());
stockAiVO.setSymbol(stockType);
});
pageInfo.setList(newStockAiList);
}
-
-
return ServerResponse.createBySuccess(pageInfo);
} catch (Exception ex) {
log.error("StockAiService getAdminStockAiList error", ex);
@@ -223,4 +232,169 @@
}
return ServerResponse.createByError();
}
+
+ /**
+ * 后台查询ai交易订单
+ * @param pageNum
+ * @param pageSize
+ * @param stockType
+ * @param status
+ * @param userId
+ * @param phone
+ * @return
+ */
+ @Override
+ public ServerResponse getAdminStockAiOrderList(Integer pageNum, Integer pageSize, String stockType, String status, Integer userId, String phone) {
+ try {
+ PageHelper.startPage(pageNum, pageSize);
+ List<StockAiOrderVO> stockAIOrders = stockAiOrderMapper.getAdminStockAiOrderList(stockType, status, userId, phone);
+ // 获取分页信息
+ PageInfo<StockAiOrderVO> pageInfo = new PageInfo<>(stockAIOrders);
+
+ if (!pageInfo.getList().isEmpty()) {
+ List<StockAiOrderVO> newStockAiOrders = pageInfo.getList();
+ newStockAiOrders.forEach(stockAiOrderVO -> {
+ EStockType eStockType = EStockType.getEStockTypeByCode(stockAiOrderVO.getStockType());
+ stockAiOrderVO.setStockTypeName(eStockType.getSymbol1());
+ stockAiOrderVO.setSymbol(stockType);
+ });
+ pageInfo.setList(newStockAiOrders);
+ }
+ return ServerResponse.createBySuccess(pageInfo);
+ } catch (Exception e) {
+ log.error("StockAiService getAdminStockAiOrderList error", e);
+ }
+ return ServerResponse.createByError();
+ }
+
+ /**
+ * 建仓
+ * @param model
+ * @return
+ */
+ @Override
+ public ServerResponse openPosition(StockAIOrderPosition model) {
+ try {
+ Object lock = locks.computeIfAbsent(model.getStockAiOrderId(), k -> new Object());
+ synchronized (lock) {
+ StockAIOrder stockAIOrder = stockAiOrderMapper.selectById(model.getStockAiOrderId());
+ if (stockAIOrder == null) {
+ return ServerResponse.createByErrorMsg("AI订单不存在");
+ }
+
+ if (!stockAIOrder.getStatus().equals(EStockAIOrderStatus.passed.getStatus())) {
+ return ServerResponse.createByErrorMsg("申请通过状态才能建仓!");
+ }
+ //计算剩余金额
+ BigDecimal remainAmount = model.getStockPrice().multiply(BigDecimal.valueOf(model.getStockNum()));
+ if (stockAIOrder.getRemainAmount().compareTo(remainAmount) < 0) {
+ return ServerResponse.createByErrorMsg("剩余买入金额不足");
+ }
+ stockAIOrder.setRemainAmount(stockAIOrder.getRemainAmount().subtract(remainAmount));
+
+ //计算收益 (平仓-建仓)*数量
+ BigDecimal earnings = model.getCoverPrice().subtract(model.getStockPrice()).multiply(BigDecimal.valueOf(model.getStockNum()));
+ model.setCreatDate(new Date());
+ model.setEarnings(earnings);
+ stockAIOrder.setRealEarning(stockAIOrder.getRealEarning().add(earnings));
+
+ //保存
+ stockAiOrderMapper.updateById(stockAIOrder);
+ stockAiOrderPositionMapper.insert(model);
+ return ServerResponse.createBySuccess("建仓成功");
+ }
+ } catch (Exception e) {
+ log.error("StockAiService openPosition error", e);
+ }
+ return ServerResponse.createByError();
+ }
+
+ @Override
+ public ServerResponse orderOperation(Long id, String status) {
+ try {
+ Object lock = locks.computeIfAbsent(id, k -> new Object());
+ synchronized (lock) {
+ StockAIOrder stockAIOrder = stockAiOrderMapper.selectById(id);
+ if (stockAIOrder == null) {
+ return ServerResponse.createByErrorMsg("AI订单不存在");
+ }
+ if (status.equals(EStockAIOrderStatus.passed.getStatus())) {
+ //通过
+ if (!stockAIOrder.getStatus().equals(EStockAIOrderStatus.wait.getStatus())) {
+ return ServerResponse.createByErrorMsg("只能通过待审核订单");
+ }
+ } else if (status.equals(EStockAIOrderStatus.notPass.getStatus())) {
+ if (!stockAIOrder.getStatus().equals(EStockAIOrderStatus.wait.getStatus())) {
+ return ServerResponse.createByErrorMsg("只能拒绝待审核订单");
+ }
+ //拒绝TODO 归还买入金额
+ StockAI stockAI = stockAiMapper.selectById(stockAIOrder.getStockAiId());
+ //买入金额
+ BigDecimal buyNum = stockAIOrder.getBuyAmount();
+ /*//如果不是墨西哥币需要转换金额
+ if (!stockAI.getStockType().equals(EStockType.MX.getCode())) {
+ EStockType stockType = EStockType.getEStockTypeByCode(stockAI.getStockType());
+ ExchangeRate exchangeRate = exchangeRateRepository.findExchangeRateByCurrencyAndConversionCurrency(stockType.getSymbol(), EStockType.MX.getSymbol())
+ .orElse(null);
+ if (exchangeRate == null) {
+ return ServerResponse.createByErrorMsg("请先设置当前货币汇率");
+ }
+ //转换为墨西哥币
+ buyNum = iUserAssetsServices.exchangeAmountByRate(buyNum, exchangeRate.getRata());
+ }*/
+ Boolean flag = iUserAssetsServices.aiAvailableBalanceChange(stockAI.getStockType(), stockAIOrder.getUserId(), EUserAssets.BUY_AI_REJECT, buyNum, null);
+ if (!flag) {
+ return ServerResponse.createByErrorMsg("扣款失败，无法转换货币");
+ }
+ } else if (status.equals(EStockAIOrderStatus.finished.getStatus())) {
+ if (!stockAIOrder.getStatus().equals(EStockAIOrderStatus.passed.getStatus())) {
+ return ServerResponse.createByErrorMsg("只能结算申请通过订单");
+ }
+ //结算 TODO结算建仓 计算收益
+ /*//所有建仓
+ List<StockAIOrderPosition> stockAIOrderPositionList = stockAiOrderPositionMapper
+ .selectList(new QueryWrapper<StockAIOrderPosition>()
+ .eq("stock_ai_order_id", id));
+ //结算金额
+ BigDecimal amount = stockAIOrder.getBuyAmount();
+ if (!stockAIOrderPositionList.isEmpty()) {
+ //总收益
+ BigDecimal earningsSUM = stockAIOrderPositionList.stream()
+ .map(StockAIOrderPosition::getEarnings) // 获取 BigDecimal 字段
+ .filter(Objects::nonNull) // 过滤 null 值
+ .reduce(BigDecimal.ZERO, BigDecimal::add); // 累加
+ amount = amount.add(earningsSUM);
+ }*/
+ StockAI stockAI = stockAiMapper.selectById(stockAIOrder.getStockAiId());
+ //收益+本金
+ BigDecimal amount = stockAIOrder.getBuyAmount().add(stockAIOrder.getRealEarning());
+ //需要解冻金额
+ BigDecimal buyAmount = stockAIOrder.getBuyAmount();
+ /*//如果不是墨西哥币需要转换金额
+ if (!stockAI.getStockType().equals(EStockType.MX.getCode())) {
+ EStockType stockType = EStockType.getEStockTypeByCode(stockAI.getStockType());
+ ExchangeRate exchangeRate = exchangeRateRepository.findExchangeRateByCurrencyAndConversionCurrency(stockType.getSymbol(), EStockType.MX.getSymbol())
+ .orElse(null);
+ if (exchangeRate == null) {
+ return ServerResponse.createByErrorMsg("请先设置当前货币汇率");
+ }
+ //转换为墨西哥币
+ amount = iUserAssetsServices.exchangeAmountByRate(amount, exchangeRate.getRata());
+ buyAmount = iUserAssetsServices.exchangeAmountByRate(buyAmount, exchangeRate.getRata());
+ }*/
+ Boolean flag = iUserAssetsServices.aiAvailableBalanceChange(stockAI.getStockType(), stockAIOrder.getUserId(), EUserAssets.AI_SETTLEMENT, amount, buyAmount);
+ if (!flag) {
+ return ServerResponse.createByErrorMsg("扣款失败，无法转换货币");
+ }
+ }
+ stockAIOrder.setStatus(status);
+ stockAIOrder.setAuditDate(new Date());
+ stockAiOrderMapper.updateById(stockAIOrder);
+ return ServerResponse.createBySuccess("操作成功");
+ }
+ } catch (Exception e) {
+ log.error("StockAiService openPosition error", e);
+ }
+ return ServerResponse.createByError();
+ }
}
--
Gitblit v1.9.3