From ec72fc1827d3e46a99d302dd3c9ca61411c70943 Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Thu, 24 Jul 2025 14:41:25 +0800
Subject: [PATCH] 修改

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

diff --git a/src/main/java/com/nq/service/impl/StockAiServiceImpl.java b/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
index a17362f..b1b2304 100644
--- a/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockAiServiceImpl.java
@@ -3,7 +3,6 @@
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 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;
@@ -12,6 +11,7 @@
 import com.nq.enums.*;
 import com.nq.pojo.*;
 import com.nq.service.*;
+import com.nq.vo.stock.ai.StockAIOrderPositionVO;
 import com.nq.vo.stock.ai.StockAiOrderTypeVO;
 import com.nq.vo.stock.ai.StockAiOrderVO;
 import com.nq.vo.stock.ai.StockAiVO;
@@ -44,6 +44,8 @@
     IUserService iUserService;
     @Autowired
     IUserAssetsServices iUserAssetsServices;
+    @Autowired
+    private UserAssetsServices userAssetsServices;
 
     /**
      * 获取上架ai产品
@@ -103,19 +105,11 @@
                 }
                 //获取用户账户
                 UserAssets userAssets = iUserAssetsServices.assetsByTypeAndUserId(EStockType.MX.getCode(), user.getId());
-                /*BigDecimal finalBuyNum = buyNum;    //购买金额
+                BigDecimal finalBuyNum = buyNum;
                 //如果不是墨西哥币需要转换金额
                 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());
-                }*/
+                    buyNum = userAssetsServices.exchangeAmountByRate(stockAI.getStockType(), buyNum);
+                }
                 if(buyNum.compareTo(userAssets.getAvailableBalance()) > 0){
                     return ServerResponse.createByErrorMsg("可用余额不足" + userAssets.getAvailableBalance(), request);
                 }
@@ -123,13 +117,14 @@
                 if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
                     return ServerResponse.createByErrorMsg("请先缴清待补资金", request);
                 }
-                iUserAssetsServices.aiAvailableBalanceChange(stockAI.getStockType(), user.getId(), EUserAssets.BUY_AI, buyNum);
+                //已经转化 直接穿MEX类型
+                iUserAssetsServices.aiAvailableBalanceChange(EStockType.MX.getCode(), user.getId(), EUserAssets.BUY_AI, buyNum);
                 StockAIOrder stockAIOrder = new StockAIOrder();
                 stockAIOrder.setUserId(user.getId());
                 stockAIOrder.setStockAiId(id.intValue());
                 stockAIOrder.setBuyDate(new Date());
-                stockAIOrder.setBuyAmount(buyNum);
-                stockAIOrder.setRemainAmount(buyNum);
+                stockAIOrder.setBuyAmount(finalBuyNum);
+                stockAIOrder.setRemainAmount(finalBuyNum);
                 stockAIOrder.setRealEarning(BigDecimal.valueOf(0));
                 stockAIOrder.setStatus(EStockAIOrderStatus.wait.getStatus());   //等待审核
                 stockAiOrderMapper.insert(stockAIOrder);
@@ -364,7 +359,27 @@
                 return ServerResponse.createBySuccess("操作成功");
             }
         } catch (Exception e) {
-            log.error("StockAiService openPosition error", e);
+            log.error("StockAiService orderOperation error", e);
+        }
+        return ServerResponse.createByError();
+    }
+
+    /**
+     * 建仓列表
+     * @param pageNum
+     * @param pageSize
+     * @param stockAiOrderId
+     * @return
+     */
+    @Override
+    public ServerResponse getPositionList(Integer pageNum, Integer pageSize, Integer stockAiOrderId) {
+        try {
+            PageHelper.startPage(pageNum, pageSize);
+            List<StockAIOrderPositionVO> positionVOList = stockAiOrderPositionMapper.getPositionListByOrderId(stockAiOrderId);
+            PageInfo<StockAIOrderPositionVO> pageInfo = new PageInfo<>(positionVOList);
+            return ServerResponse.createBySuccess(pageInfo);
+        } catch (Exception e) {
+            log.error("StockAiService getPositionList error", e);
         }
         return ServerResponse.createByError();
     }

--
Gitblit v1.9.3