1

zj

2024-04-26 9112da7b09e001e465313b163f7b72f84d8c05d3

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

@@ -1,13 +1,20 @@
package com.nq.utils.task.stock;

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.dao.*;
import com.nq.enums.EStockType;
import com.nq.pojo.*;
import com.nq.service.IMandatoryLiquidationService;
import com.nq.service.IStockService;
import com.nq.utils.HttpClientRequest;
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;
@@ -19,6 +26,8 @@
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;


@Component
@@ -28,47 +37,44 @@
    @Autowired
    StockMapper stockMapper;

    @Autowired
    IUserPositionService userPositionService;

    @Autowired
    UserPositionMapper userPositionMapper;

    private final Lock stockConstraintLock = new ReentrantLock();



    @Autowired
    IMandatoryLiquidationService mandatoryLiquidationService;



    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 0/1 * * ?")
    /**
     * 同步系统所需要的股票
     * */
    @Scheduled(cron = "0 0 0/2  * * ?")
    public void syncINStockData() {
        loadAllStock(EStockType.IN);
//        loadAllStock(EStockType.HK);
    }




    /**
     * 同步美国股票
     * */
    @Scheduled(cron = "0 0 0/1 * * ?")
    public  void synsUSStockData(){
        loadAllStock(EStockType.US);
    }


    /**
     * 同步美国股票
     * */
    @Scheduled(cron = "0 0/5 * * * ?")
//    @Scheduled(cron = "0 0/30 * * * ?")
    public  void loadStockCompanies(){
        loadAllCompanies();
    }




@@ -80,8 +86,7 @@
        for (int i = 0; i <list.size() ; i++) {
            Stock stock = list.get(i);
            EStockType eStockType = EStockType.getEStockTypeByCode(stock.getStockType());
            String result = HttpClientRequest.doGet(eStockType.stockUrl+"companies?pid=+"+stock.getStockCode()+"+country_id="+eStockType.getContryId()+"&size=1000&page=1&key="+eStockType.stockKey);
            log.info(" 公司信息返回{}",result);
            String result = HttpClientRequest.doGet(eStockType.stockUrl+"companies?pid=+"+stock.getStockCode()+"+country_id="+eStockType.getContryId()+"&size=1&page=1&key="+eStockType.stockKey);
            try {
                JSONObject jsonObject = JSONObject.parseObject(result);
                JSONObject companiesInfo = jsonObject.getJSONArray("data").getJSONObject(0);
@@ -93,9 +98,6 @@
        }

    }




    /**
     * 加载所有股票数据
@@ -109,7 +111,6 @@
            while (totleStock>list.size()){
                try{
                    String result = HttpClientRequest.doGet(eStockType.stockUrl+"list?country_id="+eStockType.getContryId()+"&size=1000&page="+page+"&key="+eStockType.stockKey);

                    ReponseBase reponseBase  = new Gson().fromJson(result,ReponseBase.class);
                    list.addAll(reponseBase.getData());
                    page++;
@@ -131,17 +132,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);
            }
@@ -152,27 +163,27 @@
        }
    }




    /***
     * 每3分钟同步一次数据源到数据库
     *
    /**
     * 强制平仓
     */
    @Scheduled(cron = "0 0/1 * * * ?")
    public  void  sysnInstockData(){
        List<Stock>  list =   stockMapper.findStockList();
        if(list.size()>0){
            for (int i = 0; i < list.size(); i++) {
                try {
                    Stock stock = list.get(i);
                    StockRealTimeBean stockDetailBean = RedisKeyUtil.getCacheRealTimeStock(stock);
                    stock.setIncreaseRatio(new BigDecimal(stockDetailBean.getPcp().replace("%","")));
                    stockMapper.updateById(stock);
                }catch (Exception e){

    @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("强制平仓任务--------->上次任务还未执行完成，本次任务忽略");
        }
    }
}