1

zj

2024-06-03 09206aedcfdf30050123e99f2af0a192ebad1de4

 src/main/java/com/nq/utils/task/stock/StockTask.java

@@ -1,15 +1,23 @@
package com.nq.utils.task.stock;

import cn.hutool.core.date.DateUtil;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson2.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.google.gson.Gson;
import com.nq.common.ServerResponse;
import com.nq.dao.*;
import com.nq.enums.EStockType;
import com.nq.pojo.*;
import com.nq.service.IMandatoryLiquidationService;
import com.nq.service.IStockService;
import com.nq.service.IUserAssetsServices;
import com.nq.service.IUserPositionService;
import com.nq.utils.http.HttpClientRequest;
import com.nq.utils.redis.RedisKeyUtil;
import com.nq.utils.redis.RedisShardedPoolUtils;
import com.nq.utils.stock.BuyAndSellUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@@ -21,6 +29,9 @@
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
import java.util.stream.Collectors;


@Component
@@ -30,34 +41,46 @@
    @Autowired
    StockMapper stockMapper;

    @Autowired
    IUserPositionService userPositionService;

    @Autowired
    UserPositionMapper userPositionMapper;

    @Autowired
    IMandatoryLiquidationService mandatoryLiquidationService;

    @Autowired
    UserStockSubscribeMapper userStockSubscribeMapper;


    @Autowired
    StockSubscribeMapper stockSubscribeMapper;

    @Autowired
    IUserPositionService iUserPositionService;


    @Autowired
    IUserAssetsServices userAssetsServices;

    private final Lock ballotLock = new ReentrantLock();

    private final Lock subscriptionLock = new ReentrantLock();

    private final Lock stockConstraintLock = new ReentrantLock();


    private static final Logger log = LoggerFactory.getLogger(StockTask.class);

    public void time(boolean am, boolean pm) {
        am = false;
        pm = false;
        try {
            am = BuyAndSellUtils.isTransTime("9:29", "11:31");
            pm = BuyAndSellUtils.isTransTime("12:59", "15:00");
        } catch (Exception e) {
            log.error("= {}", e);
        }
        log.info("am = {}  pm = {}", Boolean.valueOf(am), Boolean.valueOf(pm));
    }


    /**
     * 同步系统所需要的股票
     * */
    @Scheduled(cron = "0 0/3 *  * * ?")
    @Scheduled(cron = "0 0 0/2  * * ?")
    public void syncINStockData() {
        loadAllStock(EStockType.US);
        loadAllStock(EStockType.MAS);
        loadAllStock(EStockType.IN);
//        loadAllStock(EStockType.HK);
    }


@@ -66,7 +89,7 @@
    /**
     * 同步美国股票
     * */
    @Scheduled(cron = "0 0/30 * * * ?")
//    @Scheduled(cron = "0 0/30 * * * ?")
    public  void loadStockCompanies(){
        loadAllCompanies();
    }
@@ -128,16 +151,27 @@
                    }else{
                        stock.setStockGid(o.getType());
                    }
                    if(eStockType.getCode().equals("US")){
                        stock.setStockSpell(o.getSymbol());
                    }else{
                        stock.setStockSpell(o.getTicker());
                    }
                    stock.setStockSpell(o.getSymbol());
                    stock.setIsLock(0);
                    stock.setIsShow(0);
                    stock.setDataBase(0);
                    stock.setAddTime(new Date());
                    stockMapper.insert1(stock);
                }else{
                    stock.setStockCode(o.getId());
                    stock.setStockName(o.getName());
                    stock.setStockType(eStockType.getCode());
                    if(o.getType() == null){
                        stock.setStockGid(eStockType.getCode());
                    }else{
                        stock.setStockGid(o.getType());
                    }
                    stock.setStockSpell(o.getSymbol());
                    stock.setIsLock(0);
                    stock.setIsShow(0);
                    stock.setDataBase(0);
                    stock.setAddTime(new Date());
                    stockMapper.updateById(stock);
                }
                RedisKeyUtil.setCaCheKeyBaseStock(eStockType,o);
            }
@@ -148,38 +182,86 @@
        }
    }




    /***
     * 每3分钟同步一次数据源到数据库
     *
     */
    @Scheduled(cron = "0 0/30 * * * ?")
    public  void  sysnInstockData(){
        QueryWrapper<Stock>  queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("stock_type",EStockType.SZHB.getCode());
        List<Stock> stockList = stockMapper.selectList(queryWrapper);
        for (int i = 0; i < stockList.size(); i++) {
    @Scheduled(cron = "0 0/10 * * * ?")
    public void ballot() {
        if (ballotLock.tryLock()) {
            try {
                String hdataString =  HttpClientRequest.doGet(EStockType.SZHB.stockUrl+"market/detail?symbol="+stockList.get(i).getStockCode());
                HBData hbData = new Gson().fromJson(hdataString,HBData.class);
                StockRealTimeBean stockRealTimeBean = new StockRealTimeBean();
                stockRealTimeBean.setLast(hbData.getTick().getClose()+"");
                stockRealTimeBean.setHigh(hbData.getTick().getHigh()+"");
                stockRealTimeBean.setPid(stockList.get(i).getStockCode());
                stockRealTimeBean.setLow(hbData.getTick().getLow()+"");
                stockRealTimeBean.setBid(hbData.getTick().getOpen()+"");
                stockRealTimeBean.setLast_close("0.00");
                 BigDecimal sss =  (new BigDecimal(hbData.getTick().getClose()).subtract(new BigDecimal(hbData.getTick().getOpen())));
                       sss = sss.divide(new BigDecimal(hbData.getTick().getOpen()),BigDecimal.ROUND_CEILING);
                stockRealTimeBean.setPcp(sss.setScale(2,BigDecimal.ROUND_UP).toString());
                mandatoryLiquidationService.RealTimeDataProcess(EStockType.SZHB,stockRealTimeBean);
            }catch (Exception e){

                log.info("新股上市转持仓定时任务--------->开始");
                List<StockSubscribe> stockSubscribes = stockSubscribeMapper.selectList(new LambdaQueryWrapper<StockSubscribe>()
                        .lt(StockSubscribe::getListDate, DateUtil.date()));
                if (CollectionUtils.isNotEmpty(stockSubscribes)) {
                    List<String> codeList = stockSubscribes.stream().map(StockSubscribe::getCode).collect(Collectors.toList());
                    List<UserStockSubscribe> userStockSubscribes = userStockSubscribeMapper.selectList(new LambdaQueryWrapper<UserStockSubscribe>()
                            .eq(UserStockSubscribe::getStatus, 4).in(UserStockSubscribe::getNewCode, codeList));
                    //订单转持仓
                    userStockSubscribes.forEach(f -> {
                        ServerResponse serverResponse = iUserPositionService.newStockToPosition(f.getId());//转持仓
                        if(serverResponse.isSuccess()){
                            f.setStatus(5);
                            userStockSubscribeMapper.updateById(f);
                        }else{
                            log.info("新股上市转持仓失败申购订单id:"+f.getId()+",失败原因："+serverResponse.getMsg());
                        }
                    });
                }
                log.info("新股上市转持仓定时任务--------->结束");
            } catch (Exception e) {
                log.error("新股上市转持仓定时任务发生异常", e);
            } finally {
                ballotLock.unlock();
            }

        } else {
            log.info("新股上市转持仓定时任务--------->上次任务还未执行完成，本次任务忽略");
        }
    }

//    @Scheduled(cron = "0/5 * * * * ?")
//    public void subscription() {
//        if (subscriptionLock.tryLock()) {
//            try {
//                log.info("自动转已认缴--------->开始");
//                List<UserStockSubscribe> userStockSubscribes = userStockSubscribeMapper.selectList(new LambdaQueryWrapper<UserStockSubscribe>()
//                        .eq(UserStockSubscribe::getStatus, 3));
//                userStockSubscribes.forEach(f->{
//                    Stock stock = stockMapper.selectOne(new LambdaQueryWrapper<Stock>().eq(Stock::getStockCode, f.getNewCode()));
//                    UserAssets userAssets = userAssetsServices.assetsByTypeAndUserId(stock.getStockType(),f.getUserId());
//                    if(null != userAssets && userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) == 0){
//                        f.setStatus(4);
//                        userStockSubscribeMapper.updateById(f);
//                    }
//                });
//                log.info("自动转已认缴--------->结束");
//            } catch (Exception e) {
//                log.error("自动转已认缴定时任务发生异常", e);
//            } finally {
//                subscriptionLock.unlock();
//            }
//        } else {
//            log.info("自动转已认缴定时任务--------->上次任务还未执行完成，本次任务忽略");
//        }
//    }

    /**
     * 强制平仓
     */
//    @Scheduled(cron = "0/1 * * * * ?")
    public void stockConstraint(){
        if (stockConstraintLock.tryLock()) {
            log.info("强制平仓任务：--------->开始");
            try {
                List<UserPosition> userPositions = userPositionMapper.selectList(new LambdaQueryWrapper<UserPosition>().isNull(UserPosition::getSellOrderId));
                if(CollectionUtils.isNotEmpty(userPositions)){
                    userPositionService.stockConstraint(userPositions);
                }
            } catch (Exception e) {
                e.printStackTrace();
                log.error("强制平仓任务错误：" + e.getMessage());
            } finally {
                stockConstraintLock.unlock();
                log.info("强制平仓任务：--------->结束");
            }
        } else {
            log.info("强制平仓任务--------->上次任务还未执行完成，本次任务忽略");
        }
    }
}