1

zj

2025-04-23 decfc2ff7244cec365d3f32a5090edf5faeae861

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

@@ -1,31 +1,36 @@
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.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.gson.Gson;
import com.nq.dao.*;
import com.nq.dao.StockMapper;
import com.nq.dao.UserPositionMapper;
import com.nq.enums.EStockType;
import com.nq.pojo.*;
import com.nq.pojo.DataStockBean;
import com.nq.pojo.ReponseBase;
import com.nq.pojo.Stock;
import com.nq.pojo.UserPosition;
import com.nq.service.IMandatoryLiquidationService;
import com.nq.service.IStockService;
import com.nq.service.IUserPositionService;
import com.nq.utils.http.HttpClientRequest;
import com.nq.utils.redis.RedisKeyConstant;
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;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;

import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

@@ -46,76 +51,160 @@
    private final Lock stockConstraintLock = new ReentrantLock();



    @Autowired
    IMandatoryLiquidationService mandatoryLiquidationService;



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



    private final AtomicBoolean syncINStockData = new AtomicBoolean(false);

    private final Lock syncINStockDataLock = new ReentrantLock();

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




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




    /**
     * 加载公司信息
     * */
    public  void loadAllCompanies(){
        List<Stock>  list =   stockMapper.findStockList();
        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=1&page=1&key="+eStockType.stockKey);
        if (syncINStockData.get()) { // 判断任务是否在处理中
            return;
        }
        if (syncINStockDataLock.tryLock()) {
            try {
                JSONObject jsonObject = JSONObject.parseObject(result);
                JSONObject companiesInfo = jsonObject.getJSONArray("data").getJSONObject(0);
                RedisKeyUtil.setCacheCompanies(stock,new Gson().toJson(companiesInfo));
            }catch (Exception e){
                log.info("");

                syncINStockData.set(true); // 设置处理中标识为true
                loadAllStock(EStockType.ST);
            } finally {
                syncINStockDataLock.unlock();
                syncINStockData.set(false); // 设置处理中标识为false
            }
        }

    }

    //黄金
    private static final String gold_API_URL = "http://139.196.211.109/exchange_curr.action?username=Qq112233&password=3ce25a66d5b3a8cd661024fea6c79388&id=AUUSDO";

    //原油
    private static final String crude_oil_API_URL = "http://47.112.169.122/fOption_curr.action?username=Qq112233&password=3ce25a66d5b3a8cd661024fea6c79388&id=@CL0W";

    @Scheduled(cron = "0/10 * * * * ?")  // 每6秒执行一次
    public void gold() {
        try {
            // 使用RestTemplate发起HTTP请求
            RestTemplate restTemplate = new RestTemplate();
            String response = restTemplate.getForObject(gold_API_URL, String.class);

            // 解析返回的CSV格式数据，去除可能存在的换行符
            if (response != null) {
                // 清除换行符并按逗号分割数据
                String[] parts = response.trim().split(",");
                String price = parts[2].trim();  // "3348.4"
                // 转换价格为Double类型
                // 保存价格到Redis
                RedisShardedPoolUtils.set(RedisKeyConstant.gold, String.valueOf(price));
                log.info("黄金定时任务------成功");
            } else {
                log.info("黄金定时任务------没有接收到数据");
            }
        } catch (Exception e) {
            e.printStackTrace();
            log.error("黄金定时任务------请求或数据解析失败：" + e.getMessage());
        }
    }

    @Scheduled(cron = "0/15 * * * * ?")  // 每6秒执行一次
    public void crudeOil() {
        try {
            // 使用RestTemplate发起HTTP请求
            RestTemplate restTemplate = new RestTemplate();
            String response = restTemplate.getForObject(crude_oil_API_URL, String.class);

            // 解析返回的CSV格式数据，去除可能存在的换行符
            if (response != null) {
                // 清除换行符并按逗号分割数据
                String[] parts = response.trim().split(",");
                String price = parts[2].trim();  // "3348.4"
                // 转换价格为Double类型
                // 保存价格到Redis
                RedisShardedPoolUtils.set(RedisKeyConstant.crude_oil, String.valueOf(price));
                log.info("原油定时任务------没有接收到数据");
            } else {
                log.info("原油定时任务------没有接收到数据");
            }
        } catch (Exception e) {
            e.printStackTrace();
            log.error("原油定时任务------请求或数据解析失败：" + e.getMessage());
        }
    }


    //黄金
    private static final String k_gold_API_URL = "http://139.196.211.109/ldMetal_k.action?username=Qq112233&password=3ce25a66d5b3a8cd661024fea6c79388&id=AUUSDO&num=-100&period=";

    //原油
    private static final String k_crude_oil_API_URL = "http://47.112.169.122/fOption_k.action?username=Qq112233&password=3ce25a66d5b3a8cd661024fea6c79388&id=@CL0W&num=-100&period=";

    @Scheduled(cron = "0/6 * * * * ?")  // 每6秒执行一次
    public void getKDate() throws InterruptedException {
        RestTemplate restTemplate = new RestTemplate();
        String[] arr = {"d", "w", "m", "1", "5", "30"};
        for (String str : arr) {
            String g = restTemplate.getForObject(k_gold_API_URL+str, String.class);
            RedisShardedPoolUtils.set("k_gold_"+str, g);
            Thread.sleep(6000);
            String c = restTemplate.getForObject(k_crude_oil_API_URL+str, String.class);
            RedisShardedPoolUtils.set("k_crude_oil_"+str, c);
            Thread.sleep(6000);
        }
    }
    /**
     * 同步美国股票
     */
//    @Scheduled(cron = "0 0/30 * * * ?")
//    public void loadStockCompanies() {
//        loadAllCompanies();
//    }
//
//
//    /**
//     * 加载公司信息
//     */
//    public void loadAllCompanies() {
//        List<Stock> list = stockMapper.findStockList();
//        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=1&page=1&key=" + eStockType.stockKey);
//            try {
//                JSONObject jsonObject = JSONObject.parseObject(result);
//                JSONObject companiesInfo = jsonObject.getJSONArray("data").getJSONObject(0);
//                RedisKeyUtil.setCacheCompanies(stock, new Gson().toJson(companiesInfo));
//            } catch (Exception e) {
//                log.info("");
//
//            }
//        }
//
//    }

    /**
     * 加载所有股票数据
     * */
    public  void  loadAllStock(EStockType eStockType){
        log.info("同步股票 数据 {}",eStockType.getCode());
     */
    public void loadAllStock(EStockType eStockType) {
        log.info("同步股票 数据 {}", eStockType.getCode());
        List<DataStockBean> list = new ArrayList<>();
        int totleStock = 1;
        int page =0;
        int page = 0;
        try {
            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);
            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++;
                    totleStock = reponseBase.getTotal();
                }catch (Exception e){
                } catch (Exception e) {
                    e.printStackTrace();
                    break;
                }
@@ -127,9 +216,9 @@
                    stock.setStockCode(o.getId());
                    stock.setStockName(o.getName());
                    stock.setStockType(eStockType.getCode());
                    if(o.getType() == null){
                    if (o.getType() == null) {
                        stock.setStockGid(eStockType.getCode());
                    }else{
                    } else {
                        stock.setStockGid(o.getType());
                    }
                    stock.setStockSpell(o.getSymbol());
@@ -138,13 +227,13 @@
                    stock.setDataBase(0);
                    stock.setAddTime(new Date());
                    stockMapper.insert1(stock);
                }else{
                } else {
                    stock.setStockCode(o.getId());
                    stock.setStockName(o.getName());
                    stock.setStockType(eStockType.getCode());
                    if(o.getType() == null){
                    if (o.getType() == null) {
                        stock.setStockGid(eStockType.getCode());
                    }else{
                    } else {
                        stock.setStockGid(o.getType());
                    }
                    stock.setStockSpell(o.getSymbol());
@@ -154,36 +243,41 @@
                    stock.setAddTime(new Date());
                    stockMapper.updateById(stock);
                }
                RedisKeyUtil.setCaCheKeyBaseStock(eStockType,o);
                RedisKeyUtil.setCaCheKeyBaseStock(eStockType, o);
            }
            log.info("同步股票 数据 成功 {}  总共同步数据 {}",eStockType.getCode(),list.size());
            log.info("同步股票 数据 成功 {}  总共同步数据 {}", eStockType.getCode(), list.size());
        } catch (
                Exception e) {
            log.error("同步出错", e);
        }
    }

    private final AtomicBoolean stockConstraint = new AtomicBoolean(false);

    /**
     * 强制平仓
     */
    @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("强制平仓任务--------->上次任务还未执行完成，本次任务忽略");
        }
    }
//    @Scheduled(cron = "0/1 * * * * ?")
//    public void stockConstraint() {
//        if (stockConstraint.get()) { // 判断任务是否在处理中
//            return;
//        }
//        if (stockConstraintLock.tryLock()) {
//            try {
//                stockConstraint.set(true); // 设置处理中标识为true
//                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();
//                stockConstraint.set(false); // 设置处理中标识为false
//            }
//        } else {
//            log.info("强制平仓任务--------->上次任务还未执行完成，本次任务忽略");
//        }
//    }
}