From 6edc45c5e58354f74158c330597f6f347b21c1cc Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Tue, 08 Oct 2024 21:51:03 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java |  241 ++++++++++++++++++++++++++++++-----------------
 1 files changed, 154 insertions(+), 87 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java b/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java
index 725d646..e6128f1 100644
--- a/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java
@@ -1,23 +1,29 @@
 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.github.pagehelper.PageHelper;
 import com.github.pagehelper.PageInfo;
 import com.nq.common.ServerResponse;
 import com.nq.dao.StockSubscribeMapper;
+import com.nq.dao.UserAssetsMapper;
 import com.nq.dao.UserMapper;
 import com.nq.dao.UserStockSubscribeMapper;
 import com.nq.enums.EStockType;
 import com.nq.enums.EUserAssets;
 import com.nq.pojo.*;
 import com.nq.service.*;
+import com.nq.utils.redis.CookieUtils;
 import com.nq.utils.timeutil.DateTimeUtil;
 import com.nq.utils.KeyUtils;
 import com.nq.utils.PropertiesUtil;
 import com.nq.utils.redis.JsonUtil;
 import com.nq.utils.redis.RedisShardedPoolUtils;
 import com.nq.utils.stock.BuyAndSellUtils;
+import com.nq.vo.stock.UserStockSubscribeAddIn;
+import com.sun.org.apache.bcel.internal.generic.RETURN;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -27,6 +33,8 @@
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 import java.math.BigDecimal;
+import java.text.ParseException;
+import java.text.SimpleDateFormat;
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
@@ -64,6 +72,9 @@
 
     @Resource
     IUserAssetsServices iUserAssetsServices;
+
+    @Autowired
+    UserAssetsMapper userAssetsMapper;
     /**
      * 用户新股申购
      * @param model
@@ -71,86 +82,116 @@
      */
     @Override
     @Transactional
-    public ServerResponse insert(UserStockSubscribe model, HttpServletRequest request) throws Exception {
+    public ServerResponse insert(UserStockSubscribeAddIn model, HttpServletRequest request) throws Exception {
         int ret = 0;
         if (model == null) {
             return ServerResponse.createByErrorMsg("参数错误",request);
         }
-            User user = iUserService.getCurrentRefreshUser(request);
-            if (model.getNewCode() != null) {
-                StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<StockSubscribe>().eq("code", model.getNewCode()));
-                //实名认证开关
-                SiteProduct siteProduct = iSiteProductService.getProductSetting();
-                if (siteProduct.getRealNameDisplay() && (StringUtils.isBlank(user.getRealName()) || StringUtils.isBlank(user.getIdCard()))) {
-                    return ServerResponse.createByErrorMsg("订单失败,请先实名认证",request);
-                }
+        User user = iUserService.getCurrentRefreshUser(request);
+        UserAssets userAssets =   iUserAssetsServices.assetsByTypeAndUserId("IN",user.getId());
+        if (model.getNewCode() != null) {
+            StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<StockSubscribe>()
+                    .eq("newlist_id", model.getNewlistId()));
+            //判断购买新股需不需要密钥
+            if(StringUtils.isNotEmpty(stockSubscribe.getPassword()) && !stockSubscribe.getPassword().equals(model.getPassword())){
+                return ServerResponse.createByErrorMsg("密钥输入错误",request);
+            }
+            //实名认证开关
+            SiteProduct siteProduct = iSiteProductService.getProductSetting();
+            if (siteProduct.getRealNameDisplay() && user.getIsActive() != 2) {
+                return ServerResponse.createByErrorMsg("订单失败,请先实名认证",request);
+            }
+            if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
+                return ServerResponse.createByErrorMsg("请先缴清待补资金", request);
+            }
+            Date currentDate = new Date();
+            if(currentDate.before(stockSubscribe.getSubscriptionTime())){
+                return ServerResponse.createByErrorMsg("未开售",request);
+            }
+            if(currentDate.after(stockSubscribe.getSubscribeTime())){
+                return ServerResponse.createByErrorMsg("已结束",request);
+            }
+//            //重复申购限制
+//            UserStockSubscribe userStockSubscribe = userStockSubscribeMapper.selectOne(new QueryWrapper<UserStockSubscribe>().eq(
+//                    "new_code", model.getNewCode()).eq("user_id", user.getId()).eq("type",model.getType()));
+//            if (userStockSubscribe != null) {
+//                return ServerResponse.createByErrorMsg("请勿重复申请",request);
+//            }
+            if (siteProduct.getRealNameDisplay() && user.getIsLock().intValue() == 1) {
+                return ServerResponse.createByErrorMsg("订单失败,帐户已被锁定",request);
+            }
+            if (stockSubscribe == null) {
+                return ServerResponse.createByErrorMsg("新的股票代码不存在",request);
+            }
 
-                if(new Date().before(stockSubscribe.getSubscribeTime()) && new Date().after(stockSubscribe.getSubscribeTime()) ){
-                    return ServerResponse.createByErrorMsg("不在认缴或配售时间之内",request);
-                }
+//            if ( model.getApplyNums() == null){
+//                // 申请数量为空时返回错误信息
+//                return ServerResponse.createByErrorMsg("申请数量不能为空", request);
+//            }
+//
+//            if (model.getApplyNums() > stockSubscribe.getOrderNumber()){
+//                // 申请数量大于订单数量时返回错误信息
+//                return ServerResponse.createByErrorMsg("最多可申购:" + stockSubscribe.getOrderNumber(), request);
+//            }
 
-                //重复申购限制
-                UserStockSubscribe userStockSubscribe = userStockSubscribeMapper.selectOne(new QueryWrapper<UserStockSubscribe>().eq(
-                        "new_code", model.getNewCode()).eq("user_id", user.getId()).eq("type",model.getType()));
-                if (userStockSubscribe != null) {
-                    return ServerResponse.createByErrorMsg("请勿重复申购",request);
-                }
-                if (siteProduct.getRealNameDisplay() && user.getIsLock().intValue() == 1) {
-                    return ServerResponse.createByErrorMsg("订单失败,帐户已被锁定",request);
-                }
-                if (stockSubscribe == null) {
-                    return ServerResponse.createByErrorMsg("新的股票代码不存在",request);
-                }
-                if (model.getApplyNums() == null || model.getApplyNums() > stockSubscribe.getOrderNumber()|| model.getApplyNums() < 500) {
-                    return ServerResponse.createByErrorMsg("最小起订量是500,最大值"
-                            + stockSubscribe.getOrderNumber(),request);
-                }
-                if(stockSubscribe.getType() == 1){
-                    model.setUserId(user.getId());
-                    model.setNewName(stockSubscribe.getName());
-                    model.setAgentId(user.getAgentId());
-                    model.setAgentName(user.getAgentName());
-                    model.setPhone(user.getPhone());
-                    model.setBuyPrice(stockSubscribe.getPrice());
-                    model.setBond(new BigDecimal(model.getApplyNums()).multiply(stockSubscribe.getPrice()));
-                    model.setRealName(Objects.equals(user.getRealName(), "")||user.getRealName()==null ?"模拟用户无实名":user.getRealName());
-                    model.setAddTime(new Date());
-                    model.setOrderNo(KeyUtils.getUniqueKey());
-                    model.setType(stockSubscribe.getType());
-                    ret = userStockSubscribeMapper.insert(model);
-                    if (ret > 0) {
-                        return ServerResponse.createBySuccessMsg("申购成功",request);
-                    } else {
-                        return ServerResponse.createByErrorMsg("申购失败",request);
-                    }
-                }else{
-                   BigDecimal bound =  new BigDecimal(model.getApplyNums()).multiply(stockSubscribe.getPrice());
-                   BigDecimal useEnaAmount = iUserAssetsServices.getAvailableBalance(EStockType.IN.getCode(), user.getId());
-                   if(useEnaAmount.compareTo(bound)<0){
-                       ServerResponse.createByErrorMsg("余额不足,配售失败",request);
-                   }
-                    model.setUserId(user.getId());
-                    model.setNewName(stockSubscribe.getName());
-                    model.setAgentId(user.getAgentId());
-                    model.setAgentName(user.getAgentName());
-                    model.setPhone(user.getPhone());
-                    model.setBuyPrice(stockSubscribe.getPrice());
-                    model.setBond(new BigDecimal(model.getApplyNums()).multiply(stockSubscribe.getPrice()));
-                    model.setRealName(Objects.equals(user.getRealName(), "")||user.getRealName()==null ?"模拟用户无实名":user.getRealName());
-                    model.setAddTime(new Date());
-                    model.setOrderNo(KeyUtils.getUniqueKey());
-                    model.setType(stockSubscribe.getType());
-                    model.setDbMoney(BigDecimal.ZERO);
-                    ret = userStockSubscribeMapper.insert(model);
-                    iUserAssetsServices.availablebalanceChange(EStockType.IN.getCode(), user.getId(),EUserAssets.BUY,bound.negate(),"","");
-                    if (ret > 0) {
-                        return ServerResponse.createBySuccessMsg("配售成功",request);
-                    } else {
-                        return ServerResponse.createByErrorMsg("配售失败",request);
-                    }
-                }
+            //默认数量最大
+            model.setApplyNums(stockSubscribe.getOrderNumber().intValue());
 
+            if(stockSubscribe.getType() == 1){
+                model.setUserId(user.getId());
+                model.setNewName(stockSubscribe.getName());
+                model.setAgentId(user.getAgentId());
+                model.setAgentName(user.getAgentName());
+                model.setPhone(user.getPhone());
+                model.setBuyPrice(stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice());
+                model.setBond(new BigDecimal(model.getApplyNums()).multiply((stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice())));
+                model.setRealName(Objects.equals(user.getRealName(), "")||user.getRealName()==null ?"模拟用户无实名":user.getRealName());
+                model.setAddTime(new Date());
+                model.setOrderNo(KeyUtils.getUniqueKey());
+                model.setType(stockSubscribe.getType());
+                UserStockSubscribe userStockSubscribe = Convert.convert(UserStockSubscribe.class, model);
+                userStockSubscribe.setNewStockId(stockSubscribe.getNewlistId());
+                ret = userStockSubscribeMapper.insert(userStockSubscribe);
+                if (ret > 0) {
+                    return ServerResponse.createBySuccessMsg("申购成功",request);
+                } else {
+                    return ServerResponse.createByErrorMsg("申购失败",request);
+                }
+            }else{
 
+               BigDecimal bound;
+               if(stockSubscribe.getMinPrice() != null || stockSubscribe.getMinPrice().compareTo(BigDecimal.ZERO)<0){
+                   bound =  new BigDecimal(model.getApplyNums()).multiply(stockSubscribe.getMinPrice());
+               }else{
+                   bound =  new BigDecimal(model.getApplyNums()).multiply(stockSubscribe.getPrice());
+               }
+
+               BigDecimal useEnaAmount = iUserAssetsServices.getAvailableBalance(EStockType.IN.getCode(), user.getId());
+               if(useEnaAmount.compareTo(bound)<0){
+                   return ServerResponse.createByErrorMsg("余额不足,配售失败",request);
+               }
+                model.setUserId(user.getId());
+                model.setNewName(stockSubscribe.getName());
+                model.setAgentId(user.getAgentId());
+                model.setAgentName(user.getAgentName());
+                model.setPhone(user.getPhone());
+                model.setBuyPrice(stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice());
+                model.setBond(new BigDecimal(model.getApplyNums()).multiply((stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice())));
+                model.setRealName(Objects.equals(user.getRealName(), "")||user.getRealName()==null ?"模拟用户无实名":user.getRealName());
+                model.setAddTime(new Date());
+                model.setOrderNo(KeyUtils.getUniqueKey());
+                model.setType(stockSubscribe.getType());
+                model.setDbMoney(BigDecimal.ZERO);
+                UserStockSubscribe userStockSubscribe = Convert.convert(UserStockSubscribe.class, model);
+                userStockSubscribe.setNewStockId(stockSubscribe.getNewlistId());
+                ret = userStockSubscribeMapper.insert(userStockSubscribe);
+                iUserAssetsServices.availablebalanceChange(EStockType.IN.getCode(), user.getId(),EUserAssets.BUY,bound.negate(),"","");
+                if (ret > 0) {
+                    return ServerResponse.createBySuccessMsg("配售成功",request);
+                } else {
+                    return ServerResponse.createByErrorMsg("配售失败",request);
+                }
+            }
         }
         return ServerResponse.createByErrorMsg("申购失败",request);
     }
@@ -164,9 +205,9 @@
      * admin 新股申购-添加和修改
      */
     @Override
+    @Transactional
     public ServerResponse save(UserStockSubscribe model, HttpServletRequest request) {
         int ret = 0;
-
         if( model.getId() != null  ){
             if (model.getStatus() == 3||model.getStatus() == 2) {
                 model.setEndTime(DateTimeUtil.getCurrentDate());
@@ -175,21 +216,43 @@
             if (userStockSubscribe.getStatus() == 5) {
                 return ServerResponse.createByErrorMsg("It's been converted");
             }
-            StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<>(new StockSubscribe()).eq("code", userStockSubscribe.getNewCode()));
+            if(userStockSubscribe.getStatus() == 3){
+                return ServerResponse.createByErrorMsg("不能更改申购状态");
+            }
+            StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<>(new StockSubscribe())
+                    .eq("newlist_id", userStockSubscribe.getNewStockId()));
+            if((model.getStatus() == 3 && model.getApplyNumber() == null) || (model.getStatus() == 3 && model.getApplyNumber() == 0) ){
+                return ServerResponse.createByErrorMsg("中签数量不能小于0");
+            }
             if (model.getStatus() == 3 && model.getApplyNumber() != null){
                 if(stockSubscribe.getType() == 1){
-                    model.setBond(stockSubscribe.getPrice().multiply(BigDecimal.valueOf(model.getApplyNumber())));
+                    model.setBond((stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice()).multiply(BigDecimal.valueOf(model.getApplyNumber())));
                     model.setDbMoney(model.getDbMoney());
+                    //客户中签直接扣除客户账户可用资金
+                    UserAssets userAssets = iUserAssetsServices.assetsByTypeAndUserId("IN", userStockSubscribe.getUserId());
+                    if(null == userAssets){
+                        return ServerResponse.createByErrorMsg("客户资金账户不存在");
+                    }
+                    BigDecimal decimal = userStockSubscribe.getBuyPrice().multiply(new BigDecimal(model.getApplyNumber()));//中签总价
+                    BigDecimal availableBalance = userAssets.getAvailableBalance().subtract(decimal);//剩余可用余额
+                    if(decimal.compareTo(userAssets.getAvailableBalance()) <= 0){
+                        userAssets.setAvailableBalance(availableBalance);
+                    }else{
+                        BigDecimal amountToBeCovered = decimal.subtract(userAssets.getAvailableBalance());//待补金额
+                        userAssets.setAvailableBalance(BigDecimal.ZERO);
+                        userAssets.setAmountToBeCovered(userAssets.getAmountToBeCovered().add(amountToBeCovered));
+                    }
+                    userAssetsMapper.updateById(userAssets);
                     ret = userStockSubscribeMapper.update1(model);
                 }else{
                     if(model.getApplyNumber()>model.getApplyNums()){
                         return  ServerResponse.createByErrorMsg("配置中签数量不能超过申请数量",request);
                     }
                     BigDecimal cCount = new BigDecimal(model.getApplyNums()-model.getApplyNumber());
-                    BigDecimal tMoney = stockSubscribe.getPrice().multiply(cCount);
+                    BigDecimal tMoney = ((stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice())).multiply(cCount);
                     iUserAssetsServices.availablebalanceChange(EStockType.IN.getCode(),userStockSubscribe.getUserId(),
                             EUserAssets.TOP_UP,tMoney,"","");
-                    model.setBond(stockSubscribe.getPrice().multiply(BigDecimal.valueOf(model.getApplyNumber())));
+                    model.setBond((stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice()).multiply(BigDecimal.valueOf(model.getApplyNumber())));
                     model.setDbMoney(BigDecimal.ZERO);
                     model.setStatus(4);
                     ret = userStockSubscribeMapper.update1(model);
@@ -255,13 +318,13 @@
                 model.setUserId(user.getId());
                 model.setAgentId(user.getAgentId());
                 model.setAgentName(user.getAgentName());
-                StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<>(new StockSubscribe()).eq("code", model.getNewCode()));
+                StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new LambdaQueryWrapper<>(StockSubscribe.class).eq(StockSubscribe::getCode,model.getNewCode()));
                 if (stockSubscribe == null) {
                     return ServerResponse.createByErrorMsg("Failed. Ipo information does not exist");
                 }
 
                 model.setNewName(stockSubscribe.getName());
-                model.setBuyPrice(stockSubscribe.getPrice());
+                model.setBuyPrice(stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice());
                 if (model.getApplyNums() > stockSubscribe.getOrderNumber() || model.getApplyNumber() > stockSubscribe.getOrderNumber()) {
                     return ServerResponse.createByErrorMsg("The number of applications or the number of wins is the maximum" + stockSubscribe.getOrderNumber());
                 }
@@ -278,7 +341,6 @@
 
         return ServerResponse.createByErrorMsg("Operation failure");
     }
-
     /**
      * 发送站内信
      */
@@ -289,7 +351,7 @@
         if(model!=null){
             //所有人发站内信
             if(model.getUserId() == 0){
-                List<User> users = this.userMapper.listByAdmin(null, null, null, null);
+                List<User> users = this.userMapper.listByAdmin(null, null, null, null,null,null,null,null);
                 for(int k=0;k<users.size();k++){
                     User user = users.get(k);
                     SiteMessage siteMessage = new SiteMessage();
@@ -323,9 +385,15 @@
 
     /*新股申购-查询列表*/
     @Override
-    public ServerResponse<PageInfo> getList(int pageNum, int pageSize, String keyword, HttpServletRequest request){
+    public ServerResponse<PageInfo> getList(int pageNum, int pageSize, String keyword, String agentId,HttpServletRequest request){
+        String loginToken = CookieUtils.readLoginToken(request, PropertiesUtil.getProperty("agent.cookie.name"));
+        String agentJson = RedisShardedPoolUtils.get(loginToken);
+        AgentUser agentUser = (AgentUser) JsonUtil.string2Obj(agentJson, AgentUser.class);
+        if (agentUser ==null){
+            return   ServerResponse.createByError("請先登錄",null);
+        }
         PageHelper.startPage(pageNum, pageSize);
-        List<UserStockSubscribe> listData = this.userStockSubscribeMapper.pageList(pageNum, pageSize, keyword);
+        List<UserStockSubscribe> listData = this.userStockSubscribeMapper.pageList(pageNum, pageSize, keyword,agentUser.getId().toString());
         PageInfo pageInfo = new PageInfo(listData);
         pageInfo.setList(listData);
         return ServerResponse.createBySuccess(pageInfo);
@@ -353,7 +421,8 @@
 
             List<UserStockSubscribe> list = new ArrayList<>();
             for (UserStockSubscribe userStockSubscribe1 : userStockSubscribe) {
-                StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<>(new StockSubscribe()).eq("code", userStockSubscribe1.getNewCode()));
+                StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<>(new StockSubscribe())
+                        .eq("newlist_id", userStockSubscribe1.getNewStockId()));
                 if (stockSubscribe != null) {
                     list.add(userStockSubscribe1);
                 }
@@ -378,8 +447,6 @@
             }
             UserStockSubscribe userStockSubscribe = userStockSubscribeMapper.load(id);
             if (userStockSubscribe != null && userStockSubscribe.getUserId().equals(user.getId())) {
-                StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<>(new StockSubscribe())
-                        .eq("code", userStockSubscribe.getNewCode()));
                 if(userStockSubscribe.getType()== 2 ){
                     return ServerResponse.createByErrorMsg("未中签,无需支付",request);
                 }
@@ -395,7 +462,7 @@
                         return ServerResponse.createByErrorMsg("余额不足",request);
                     }
                     iUserAssetsServices.availablebalanceChange(EStockType.IN.getCode(),userStockSubscribe.getUserId(),
-                            EUserAssets.BUY,multiply,"","");
+                            EUserAssets.BUY,multiply.negate(),"","");
                     userStockSubscribe.setDbMoney(BigDecimal.ZERO);
                     userStockSubscribeMapper.update1(userStockSubscribe);
                     return  ServerResponse.createBySuccess();

--
Gitblit v1.9.3