From 2326c31e5457433b323fd6d648dfe561a9a510f8 Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Fri, 12 Sep 2025 15:06:23 +0800
Subject: [PATCH] 注册修复

---
 src/main/java/com/nq/service/impl/StockAiServiceImpl.java |   31 ++++++++++++++++---------------
 1 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/StockAiServiceImpl.java b/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
index b1b2304..07d4fa6 100644
--- a/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
@@ -15,6 +15,7 @@
 import com.nq.vo.stock.ai.StockAiOrderTypeVO;
 import com.nq.vo.stock.ai.StockAiOrderVO;
 import com.nq.vo.stock.ai.StockAiVO;
+import org.apache.commons.lang.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -54,11 +55,12 @@
      * @return
      */
     @Override
-    public ServerResponse getStockAiList(Integer pageNum, Integer pageSize) {
+    public ServerResponse getStockAiList(Integer pageNum, Integer pageSize, String stockName) {
         try {
             PageHelper.startPage(pageNum, pageSize);
             List<StockAI> stockAiList = stockAiMapper.selectList(new QueryWrapper<StockAI>()
-                    .eq("status", EStockAIStatus.online.getValue()));
+                    .eq("status", EStockAIStatus.online.getValue())
+                    .eq(StringUtils.isNotBlank(stockName),"stock_name", stockName));
             // 获取分页信息
             PageInfo<StockAI> pageInfo = new PageInfo<>(stockAiList);
             return ServerResponse.createBySuccess(pageInfo);
@@ -104,12 +106,12 @@
                     return ServerResponse.createByErrorMsg("最低购买数量" + stockAI.getMinPrice(), request);
                 }
                 //获取用户账户
-                UserAssets userAssets = iUserAssetsServices.assetsByTypeAndUserId(EStockType.MX.getCode(), user.getId());
-                BigDecimal finalBuyNum = buyNum;
-                //如果不是墨西哥币需要转换金额
-                if (!stockAI.getStockType().equals(EStockType.MX.getCode())) {
+                UserAssets userAssets = iUserAssetsServices.assetsByTypeAndUserId(stockAI.getStockType(), user.getId());
+                /*BigDecimal finalBuyNum = buyNum;
+                //如果不是默认货币需要转换金额
+                if (!stockAI.getStockType().equals(EStockType.getDefault().getCode())) {
                     buyNum = userAssetsServices.exchangeAmountByRate(stockAI.getStockType(), buyNum);
-                }
+                }*/
                 if(buyNum.compareTo(userAssets.getAvailableBalance()) > 0){
                     return ServerResponse.createByErrorMsg("可用余额不足" + userAssets.getAvailableBalance(), request);
                 }
@@ -117,14 +119,13 @@
                 if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
                     return ServerResponse.createByErrorMsg("请先缴清待补资金", request);
                 }
-                //已经转化 直接穿MEX类型
-                iUserAssetsServices.aiAvailableBalanceChange(EStockType.MX.getCode(), user.getId(), EUserAssets.BUY_AI, buyNum);
+                iUserAssetsServices.aiAvailableBalanceChange(stockAI.getStockType(), user.getId(), EUserAssets.BUY_AI, buyNum);
                 StockAIOrder stockAIOrder = new StockAIOrder();
                 stockAIOrder.setUserId(user.getId());
                 stockAIOrder.setStockAiId(id.intValue());
                 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);
@@ -197,12 +198,12 @@
             if (model == null) {
                 return ServerResponse.createByErrorMsg("editStockAi model is null");
             }
-            if (model.getStockType().isEmpty() || model.getStockName().isEmpty() || model.getMinPrice() == null
-                || model.getSuccessRate() == null || model.getExpectedEarning() == null || model.getStatus().isEmpty()) {
+            if (StringUtils.isBlank(model.getStockType()) || StringUtils.isBlank(model.getStockName()) || model.getMinPrice() == null
+                || model.getSuccessRate() == null || model.getExpectedEarning() == null || StringUtils.isBlank(model.getStatus())) {
                 return ServerResponse.createByErrorMsg("请完善必填信息");
             }
-            if (!model.getStockType().equals(EStockType.US.getCode()) && !model.getStockType().equals(EStockType.MX.getCode())) {
-                return ServerResponse.createByErrorMsg("只能选择美股或墨西哥股");
+            if (!EStockType.isExistByCode(model.getStockType())) {
+                return ServerResponse.createByErrorMsg("股票市场不存在");
             }
             //新增
             if (model.getId() == null) {

--
Gitblit v1.9.3