From 1e42d705a1f3f040c00e499e1bf42159c601aa40 Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Fri, 11 Jul 2025 16:21:36 +0800
Subject: [PATCH] AI产品交易

---
 src/main/java/com/nq/service/impl/EchoServices.java |   31 +++++++++++++++++++++++++++----
 1 files changed, 27 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/EchoServices.java b/src/main/java/com/nq/service/impl/EchoServices.java
index 8ef961d..569b4d8 100644
--- a/src/main/java/com/nq/service/impl/EchoServices.java
+++ b/src/main/java/com/nq/service/impl/EchoServices.java
@@ -5,12 +5,14 @@
 import com.nq.dao.EChoMapper;
 import com.nq.dao.OrderEchoMapper;
 import com.nq.dao.UserAssetsMapper;
+import com.nq.dao.UserPositionMapper;
 import com.nq.enums.EStockType;
 import com.nq.enums.EUserAssets;
 import com.nq.pojo.*;
 import com.nq.pojo.reponse.OrderEChoReponse;
 import com.nq.service.IEchoServices;
 import com.nq.service.IUserService;
+import com.nq.utils.PropertiesUtil;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -20,6 +22,8 @@
 import java.math.RoundingMode;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
 
 @Slf4j
 @Service
@@ -40,13 +44,16 @@
     @Autowired
     private UserAssetsMapper userAssetsMapper;
 
+    @Autowired
+    private UserPositionMapper userPositionMapper;
+
     @Override
     public List<EChoBean> queryList() {
         return eChoMapper.selectList(new LambdaQueryWrapper<EChoBean>());
     }
 
     @Override
-    public boolean buyECho(String eId, BigDecimal money, HttpServletRequest request) {
+    public boolean buyECho(String eId, BigDecimal money, String accectType, HttpServletRequest request) {
 
         User user = this.iUserService.getCurrentRefreshUser(request);
 
@@ -73,7 +80,7 @@
 
             UserAssets userAssets = userAssetsMapper.selectOne(new LambdaQueryWrapper<UserAssets>()
                     .eq(UserAssets::getUserId, orderEchoBean.getUserId())
-                    .eq(UserAssets::getAccectType, "IN")
+                    .eq(UserAssets::getAccectType, accectType)
             );
             userAssets.setAvailableBalance(userAssets.getAvailableBalance().subtract(money));
             userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(money));
@@ -98,6 +105,18 @@
 
     @Override
     public void sendMoney() {
+        // 创建固定大小的线程池,根据需求调整线程数量
+        ExecutorService executor = Executors.newFixedThreadPool(2);
+        try {
+            executor.submit(() -> sendMoney(EStockType.US.getCode()));
+            executor.submit(() -> sendMoney(EStockType.MX.getCode()));
+        } finally {
+            // 关闭线程池
+            executor.shutdown();
+        }
+    }
+
+    private void sendMoney(String stockType) {
         try {
             QueryWrapper<OrderEchoBean> queryWrapper = new QueryWrapper<>();
             queryWrapper.eq("state", 1);
@@ -116,7 +135,7 @@
                     BigDecimal aml = incomeAmount.add(money);
                     UserAssets userAssets = userAssetsMapper.selectOne(new LambdaQueryWrapper<UserAssets>()
                             .eq(UserAssets::getUserId, orderEchoBean.getUserId())
-                            .eq(UserAssets::getAccectType, "IN")
+                            .eq(UserAssets::getAccectType, stockType)
                     );
                     if (userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0) {
                         continue;
@@ -132,6 +151,11 @@
                     } else {
                         userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(aml));
                     }
+                    UserPosition userPosition = userPositionMapper.selectOne(new LambdaQueryWrapper<UserPosition>().gt(UserPosition::getAmountToBeCovered, BigDecimal.ZERO).eq(UserPosition::getUserId,userAssets.getUserId()));
+                    if(null != userPosition){
+                        userPosition.setAmountToBeCovered(userAssets.getAmountToBeCovered());
+                        userPositionMapper.updateById(userPosition);
+                    }
                     userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(money));
                     userAssetsMapper.updateById(userAssets);
                     orderEchoBean.setState("2");
@@ -145,7 +169,6 @@
             log.error("基金定时任务---失败: ", ex);
         }
     }
-
 
     @Override
     public int insertEcho(EChoBean eChoBean) {

--
Gitblit v1.9.3