From 05c70aec6fb13c9df2c90f84d9411cafe63570a8 Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Thu, 07 Aug 2025 10:53:50 +0800
Subject: [PATCH] 8.6修改

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

diff --git a/src/main/java/com/nq/service/impl/StockAiServiceImpl.java b/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
index ef8d8d5..c51b3a0 100644
--- a/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
@@ -106,10 +106,10 @@
                     return ServerResponse.createByErrorMsg("最低购买数量" + stockAI.getMinPrice(), request);
                 }
                 //获取用户账户
-                UserAssets userAssets = iUserAssetsServices.assetsByTypeAndUserId(EStockType.MX.getCode(), user.getId());
+                UserAssets userAssets = iUserAssetsServices.assetsByTypeAndUserId(EStockType.getDefault().getCode(), user.getId());
                 BigDecimal finalBuyNum = buyNum;
-                //如果不是墨西哥币需要转换金额
-                if (!stockAI.getStockType().equals(EStockType.MX.getCode())) {
+                //如果不是默认货币需要转换金额
+                if (!stockAI.getStockType().equals(EStockType.getDefault().getCode())) {
                     buyNum = userAssetsServices.exchangeAmountByRate(stockAI.getStockType(), buyNum);
                 }
                 if(buyNum.compareTo(userAssets.getAvailableBalance()) > 0){
@@ -119,8 +119,8 @@
                 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(EStockType.getDefault().getCode(), user.getId(), EUserAssets.BUY_AI, buyNum);
                 StockAIOrder stockAIOrder = new StockAIOrder();
                 stockAIOrder.setUserId(user.getId());
                 stockAIOrder.setStockAiId(id.intValue());
@@ -203,8 +203,8 @@
                 || 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