From 37a07118ef1b4c2189b0098c49a8dd63a082b3df Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Thu, 10 Oct 2024 15:05:02 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserPositionServiceImpl.java |  113 ++++++++++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 92 insertions(+), 21 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java b/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java
index de70f90..fc2eea5 100644
--- a/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java
@@ -1,5 +1,6 @@
 package com.nq.service.impl;
 
+import cn.hutool.core.convert.Convert;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.nq.dao.*;
@@ -37,6 +38,9 @@
 import java.time.ZoneId;
 import java.time.temporal.ChronoUnit;
 import java.util.*;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.locks.Lock;
+import java.util.concurrent.locks.ReentrantLock;
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 
@@ -119,6 +123,10 @@
 
     @Autowired
     IStockConfigServices iStockConfigServices;
+
+    @Autowired
+    UserPositionCheckDzService userPositionCheckDzService;
+
 
 
     @Transactional
@@ -328,7 +336,8 @@
                 .eq(UserAssets::getUserId, userPosition.getUserId())
                 .eq(UserAssets::getAccectType, "IN")
         );
-        if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
+        Long count = userStockSubscribeMapper.selectCount(new LambdaQueryWrapper<>(UserStockSubscribe.class).eq(UserStockSubscribe::getNewCode, userPosition.getStockCode()));
+        if(count > 0 && userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
             return ServerResponse.createByErrorMsg("请先缴清待补资金", request);
         }
         Boolean b = tradingHourService.timeCheck(userPosition.getStockCode());
@@ -336,7 +345,7 @@
             return ServerResponse.createByErrorMsg("订单失败,不在交易时间之内", request);
         }
         if(userPosition.getPositionType() == 3){
-            StockDz stockDz = stockDzMapper.selectOne(new LambdaQueryWrapper<StockDz>().eq(StockDz::getStockCode, userPosition.getStockCode()));
+            StockDz stockDz = stockDzMapper.selectOne(new LambdaQueryWrapper<StockDz>().eq(StockDz::getId, userPosition.getDzId()));
             LocalDateTime buyOrderLocalDateTime = LocalDateTime.ofInstant(userPosition.getBuyOrderTime().toInstant(), ZoneId.systemDefault());
             // 计算天数差
             long daysBetween = ChronoUnit.DAYS.between(buyOrderLocalDateTime, LocalDateTime.now());
@@ -675,11 +684,17 @@
             end_time = DateTimeUtil.searchStrToTimestamp(endTime);
         }
 
+
+
+
+        List<Integer> ids = new ArrayList<>();
+        if(null != searchId){
+            ids = getSubordinates(searchId);
+            ids.add(searchId);
+        }
         PageHelper.startPage(pageNum, pageSize);
-
-
         List<UserPosition> userPositions = this.userPositionMapper.listByAgent(positionType, state,
-                userId, searchId, positionSn, begin_time, end_time,null);
+                userId, ids, positionSn, begin_time, end_time,null);
 
         List<AgentPositionVO> agentPositionVOS = Lists.newArrayList();
         for (UserPosition position : userPositions) {
@@ -708,8 +723,14 @@
         }
 
 
+        List<Integer> ids = new ArrayList<>();
+        if(null != agentId){
+            ids = getSubordinates(agentId);
+            ids.add(agentId);
+        }
+
         List<UserPosition> userPositions = this.userPositionMapper.listByAgent(positionType, Integer.valueOf(1),
-                null, agentId, null, begin_time, end_time,null);
+                null, ids, null, begin_time, end_time,null);
 
 
         BigDecimal order_fee_amt = new BigDecimal("0");
@@ -741,15 +762,38 @@
         if (StringUtils.isNotBlank(endTime)) {
             end_time = DateTimeUtil.searchStrToTimestamp(endTime);
         }
-        List<UserPosition> userPositions = this.userPositionMapper.listByAgent(positionType, state, userId, agentId, positionSn, begin_time, end_time,phone);
+        List<Integer> ids = new ArrayList<>();
+        if(null != agentId){
+            ids = getSubordinates(agentId);
+            ids.add(agentId);
+        }
+
+
+        List<UserPosition> userPositions = this.userPositionMapper.listByAgent(positionType, state, userId, ids, positionSn, begin_time, end_time,phone);
         List<AdminPositionVO> adminPositionVOS = Lists.newArrayList();
         for (UserPosition position : userPositions) {
             AdminPositionVO adminPositionVO = assembleAdminPositionVO(position);
+            AgentUser agentUser = agentUserMapper.selectById(adminPositionVO.getAgentId());
+            adminPositionVO.setAgentName(agentUser.getAgentName());
+            User user = userMapper.selectById(adminPositionVO.getUserId());
+            adminPositionVO.setPhone(user.getPhone());
             adminPositionVOS.add(adminPositionVO);
         }
         PageInfo pageInfo = new PageInfo(userPositions);
         pageInfo.setList(adminPositionVOS);
         return ServerResponse.createBySuccess(pageInfo);
+    }
+
+    public List<Integer> getSubordinates(Integer id) {
+        List<AgentUser> agentUsers = agentUserMapper.selectList(new LambdaQueryWrapper<AgentUser>());
+        List<Integer> subordinates = new ArrayList<>();
+        for (AgentUser user : agentUsers) {
+            if (id.equals(user.getParentId())) {
+                subordinates.add(user.getId());
+                subordinates.addAll(getSubordinates(user.getId()));
+            }
+        }
+        return subordinates;
     }
 
     public int CountPositionNum(Integer state, Integer accountType) {
@@ -1091,7 +1135,14 @@
 
     private AgentPositionVO assembleAgentPositionVO(UserPosition position) {
         AgentPositionVO agentPositionVO = new AgentPositionVO();
-
+        User user = userMapper.selectById(position.getUserId());
+        if(null != user){
+            AgentUser agentUser = agentUserMapper.selectById(user.getAgentId());
+            agentPositionVO.setPhone(user.getPhone());
+            if(null != agentUser){
+                agentPositionVO.setAgentName(agentUser.getAgentName());
+            }
+        }
         agentPositionVO.setId(position.getId());
         agentPositionVO.setPositionSn(position.getPositionSn());
         agentPositionVO.setPositionType(position.getPositionType());
@@ -1178,7 +1229,7 @@
         if (userStockSubscribe == null) {
             return ServerResponse.createByErrorMsg("无该申购记录");
         }
-        StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<StockSubscribe>().eq("code", userStockSubscribe.getNewCode()).eq("type",userStockSubscribe.getType()));
+        StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<StockSubscribe>().eq("newlist_id", userStockSubscribe.getNewStockId()));
         if (userStockSubscribe == null) {
             return ServerResponse.createByErrorMsg("该新股不存在");
         }
@@ -1196,8 +1247,10 @@
             userPosition.setStockCode(stock.getStockCode());
             userPosition.setStockSpell(stock.getStockSpell());
             userPosition.setStockName(userStockSubscribe.getNewName());
-            userPosition.setStockGid(stockSubscribe.getStockType() + userStockSubscribe.getNewCode());
-
+            StringBuffer gid = new StringBuffer();
+            gid.append(stockSubscribe.getStockType()!=null?stockSubscribe.getStockType():"");
+            gid.append(userStockSubscribe.getNewCode()!=null?userStockSubscribe.getNewCode():"stock code invaild");
+            userPosition.setStockGid(gid.toString());
             userPosition.setBuyOrderId(GeneratePosition.getPositionId());
             userPosition.setBuyOrderTime(new Date());
             userPosition.setBuyOrderPrice(userStockSubscribe.getBuyPrice());
@@ -1253,6 +1306,7 @@
                 return ServerResponse.createByErrorMsg("新股转持仓失败");
             }
             userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(userPosition.getOrderTotalPrice()));
+            userAssets.setFreezeMoney(userAssets.getFreezeMoney()!=null?userAssets.getFreezeMoney().setScale(2,BigDecimal.ROUND_UP):null);
             userAssetsMapper.updateById(userAssets);
             if (ret > 0) {
                 userStockSubscribe.setStatus(5);
@@ -1538,7 +1592,7 @@
      * @return
      */
     @Transactional
-    public synchronized ServerResponse buyDz(String stockCode, String password, Integer num, HttpServletRequest request) throws Exception {
+    public ServerResponse buyDz(Integer dzId, String password, Integer num, HttpServletRequest request) throws Exception {
         /*实名认证开关开启*/
         SiteProduct siteProduct = iSiteProductService.getProductSetting();
         User user = this.iUserService.getCurrentRefreshUser(request);
@@ -1553,8 +1607,8 @@
         if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
             return ServerResponse.createByErrorMsg("请先缴清待补资金", request);
         }
-        StockDz stockDz = this.stockDzMapper.selectOne(new QueryWrapper<StockDz>().eq("stock_code", stockCode));
-        if (!Objects.equals(stockDz.getPassword(), password)) {
+        StockDz stockDz = this.stockDzMapper.selectOne(new QueryWrapper<StockDz>().eq("id", dzId));
+        if (StringUtils.isNotEmpty(stockDz.getPassword()) && !Objects.equals(stockDz.getPassword(), password)) {
             return ServerResponse.createByErrorMsg("密码错误", request);
         }
         if (stockDz.getIsLock() != 0) {
@@ -1566,7 +1620,8 @@
         if(stockDz.getStartTime().getTime() > new Date().getTime() || stockDz.getEndTime().getTime() < new Date().getTime()){
             return ServerResponse.createByErrorMsg("不在内幕交易时间之内", request);
         }
-        BigDecimal nowPrice = priceServices.getNowPrice(stockCode).multiply(stockDz.getDiscount());
+//        BigDecimal nowPrice = priceServices.getNowPrice(stockDz.getStockCode()).multiply(stockDz.getDiscount());
+        BigDecimal nowPrice = stockDz.getNowPrice();
 
         if (nowPrice.compareTo(new BigDecimal("0")) == 0) {
             return ServerResponse.createByErrorMsg("股票价格0,请重试", request);
@@ -1575,11 +1630,29 @@
             return ServerResponse.createByErrorMsg("最小购买数据" + stockDz.getStockNum(), request);
         }
         BigDecimal buyAmt = nowPrice.multiply(new BigDecimal(num.intValue()));
-        BigDecimal   fundratio = new BigDecimal(user.getFundRatio()).divide(new BigDecimal(100));
-        BigDecimal availableBalance =  fundratio.multiply(userAssets.getAvailableBalance());
+        BigDecimal fundratio = new BigDecimal(user.getFundRatio()).divide(new BigDecimal(100));
+        BigDecimal availableBalance = fundratio.multiply(userAssets.getAvailableBalance());
         if (buyAmt.compareTo(availableBalance) > 0) {
             return ServerResponse.createByErrorMsg("订单失败,配资不足", request);
         }
+
+        //判断审核开关
+        if(stockDz.getSwitchType() == 1){
+            UserPosition userPosition = getUserPosition(dzId,num, user, stockDz, nowPrice, stock, buyAmt);
+            UserPositionCheckDz userPositionCheckDz = Convert.convert(UserPositionCheckDz.class, userPosition);
+            userPositionCheckDz.setDzId(dzId);
+            userPositionCheckDzService.save(userPositionCheckDz);
+            return ServerResponse.createBySuccess("购买成功,等待审核", request);
+        }
+
+        // 创建UserPosition对象
+        UserPosition userPosition = getUserPosition(dzId,num, user, stockDz, nowPrice, stock, buyAmt);
+        userPositionMapper.insert(userPosition);
+        userAssetsServices.availablebalanceChange(EStockType.IN.getCode(), user.getId(), EUserAssets.BUY, buyAmt.negate(),"","");
+        return ServerResponse.createBySuccess("购买成功", request);
+    }
+
+    private UserPosition getUserPosition(Integer dzId,Integer num, User user, StockDz stockDz, BigDecimal nowPrice, Stock stock, BigDecimal buyAmt) {
         UserPosition userPosition = new UserPosition();
         userPosition.setPositionType(3);
         userPosition.setPositionSn(KeyUtils.getUniqueKey());
@@ -1608,11 +1681,9 @@
         userPosition.setAllProfitAndLose(all_profit_and_lose);
         userPosition.setOrderStayDays(Integer.valueOf(0));
         userPosition.setOrderStayFee(new BigDecimal("0"));
-
         userPosition.setOrderSpread(BigDecimal.ZERO);
-        userPositionMapper.insert(userPosition);
-        userAssetsServices.availablebalanceChange(EStockType.IN.getCode(), user.getId(), EUserAssets.BUY, buyAmt.negate(),"","");
-        return ServerResponse.createBySuccess("购买成功", request);
+        userPosition.setDzId(dzId);
+        return userPosition;
     }
 
     @Override

--
Gitblit v1.9.3