1

zj

2025-04-27 5ce072de78ae263b52dc0e2be5c7391b23c093cf

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

@@ -1,25 +1,33 @@
package com.nq.utils.task.stock;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.json.JSONArray;
import com.alibaba.fastjson2.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
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.common.ServerResponse;
import com.nq.dao.StockMapper;
import com.nq.dao.UserMapper;
import com.nq.dao.UserPendingorderMapper;
import com.nq.dao.UserPositionMapper;
import com.nq.enums.EConfigKey;
import com.nq.enums.EStockType;
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.enums.EUserAssets;
import com.nq.pojo.*;
import com.nq.pojo.reponse.kResponse;
import com.nq.service.*;
import com.nq.service.impl.StockServiceImpl;
import com.nq.service.impl.UserServiceImpl;
import com.nq.utils.ConverterUtil;
import com.nq.utils.KeyUtils;
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.GeneratePosition;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
@@ -27,9 +35,16 @@
import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLEncoder;
import java.text.DecimalFormat;
import java.util.*;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
@@ -48,11 +63,20 @@
    @Autowired
    UserPositionMapper userPositionMapper;

    private final Lock stockConstraintLock = new ReentrantLock();


    @Autowired
    private UserPendingorderService userPendingorderService;
    @Autowired
    IMandatoryLiquidationService mandatoryLiquidationService;
    @Autowired
    IUserAssetsServices iUserAssetsServices;
    @Autowired
    IStockConfigServices iStockConfigServices;

    @Autowired
    IPriceServices priceServices;
    @Autowired
    UserMapper userMapper;



    private static final Logger log = LoggerFactory.getLogger(StockTask.class);
@@ -82,115 +106,318 @@
        }
    }

    //黄金
    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 * * * * ?")
    public void pendingOrder() {
        ReentrantLock lock = new ReentrantLock();
        if (lock.tryLock()) { // 尝试获取锁
            try {
                pending();
            }catch (Exception e){
                e.printStackTrace();
                log.error("挂单定时任务报错：",e.getMessage());
            }
            finally {
                lock.unlock();  // 释放锁
            }
        } else {
            // 如果锁不可用，可以选择不执行或打印日志等
            System.out.println("挂单定时任务  任务正在执行，跳过本次执行。");
        }
    }

    @Scheduled(cron = "0/6 * * * * ?")  // 每6秒执行一次
    public void pending(){
        List<UserPendingorder> list = userPendingorderService.list(new LambdaQueryWrapper<>(UserPendingorder.class).eq(UserPendingorder::getPositionType, 1));
        if(CollectionUtil.isNotEmpty(list)){
            list.forEach(f->{
                if(f.getStockGid().equals("ST")){
                    //获取当前价格
                    //股票类型 现价 数据源的处理
                    BigDecimal nowPrice = priceServices.getNowPrice(f.getStockCode());
                    if(f.getBuyOrderPrice().compareTo(nowPrice) <= 0){
                        stockTransferPositions(f);
                    }
                }else{
                    String price = RedisShardedPoolUtils.get(RedisKeyConstant.getRedisKey(f.getStockName()));
                    if(f.getBuyOrderPrice().compareTo(new BigDecimal(price)) <= 0){
                        hjTransferPositions(f);
                    }
                }
            });
        }
    }
    public void hjTransferPositions(UserPendingorder userPendingorder){
        User user = userMapper.selectById(userPendingorder.getUserId());
        BigDecimal siteSettingBuyFee = new BigDecimal(iStockConfigServices.queryByKey(EConfigKey.BUY_HANDLING_CHARGE.getCode()).getCValue()) ;
        BigDecimal orderFree = siteSettingBuyFee.multiply(userPendingorder.getOrderTotalPrice());
        UserPosition userPosition = ConverterUtil.convert(userPendingorder, UserPosition.class);
        userPosition.setPositionType(user.getAccountType());
        userPosition.setId(null);
        userPendingorder.setPositionType(0);
        userPendingorderService.updateById(userPendingorder);
        userPositionMapper.insert(userPosition);
        iUserAssetsServices.availablebalanceChange("USDT", user.getId(), EUserAssets.HANDLING_CHARGE, orderFree, "", "");
    }

    public void stockTransferPositions(UserPendingorder userPendingorder){
        // 手续费率
        BigDecimal siteSettingBuyFee = new BigDecimal(iStockConfigServices.queryByKey(EConfigKey.BUY_HANDLING_CHARGE.getCode()).getCValue()) ;
        BigDecimal orderFree = siteSettingBuyFee.multiply(userPendingorder.getOrderTotalPrice());
        User user = userMapper.selectById(userPendingorder.getUserId());
        UserPosition userPosition = ConverterUtil.convert(userPendingorder, UserPosition.class);
        userPosition.setPositionType(user.getAccountType());
        userPosition.setId(null);
        userPendingorder.setPositionType(0);
        userPendingorderService.updateById(userPendingorder);
        userPositionMapper.insert(userPosition);
        //挂单成功扣除手续费
        iUserAssetsServices.availablebalanceChange("ST", user.getId(), EUserAssets.HANDLING_CHARGE, orderFree, "", "");
    }

    //最新价格url
    private static final String PRICE_URL = "https://quote.alltick.io/quote-b-api/trade-tick?token=794ea65dc03c5af582cddec476fbb0d7-c-app&query=";
    //k线url
    private static final String K_URL = "https://quote.alltick.io/quote-b-api/kline?token=794ea65dc03c5af582cddec476fbb0d7-c-app&query=";


    public String getPriceData() {
        // 创建查询数据
        JSONObject data = new JSONObject();

        // 用于存储符号的列表
        JSONArray symbolList = new JSONArray();

        // 需要查询的符号列表（按你的要求，这里符号从 ["857.HK", "UNH.US"] 开始）
        String[] symbols = {"GOLD", "USOIL"};

        // 遍历符号列表，构建 JSON 对象
        for (String symbol : symbols) {
            JSONObject symbolObject = new JSONObject();
            symbolObject.put("code", symbol);
            symbolList.put(symbolObject);
        }

        // 将 symbol_list 放入 data 对象中
        data.put("symbol_list", symbolList);

        // 创建查询请求对象
        JSONObject query = new JSONObject();
        query.put("trace", generateTraceCode()); // 假设 generateTraceCode() 是生成唯一追踪代码的方法
        query.put("data", data);
        // 返回构建的查询对象
        return buildUrlWithQuery(query);
    }

    // 生成唯一追踪码
    public static String generateTraceCode() {
        // 获取当前时间戳
        long timestamp = System.currentTimeMillis();

        // 生成唯一的 UUID
        String uuid = UUID.randomUUID().toString();

        // 组合时间戳和 UUID，保证唯一性
        return uuid + "-" + timestamp;
    }

    @Scheduled(cron = "0/2 * * * * ?")  // 每6秒执行一次
    public void sync() throws InterruptedException {
        ReentrantLock lock = new ReentrantLock();
        if (lock.tryLock()) { // 尝试获取锁
            try {
                gold();
                Thread.sleep(2000);
                getKDate();
            } finally {
                lock.unlock();  // 释放锁
            }
        } else {
            // 如果锁不可用，可以选择不执行或打印日志等
            System.out.println("任务正在执行，跳过本次执行。");
        }
    }

    public void gold() {
        try {
            // 使用RestTemplate发起HTTP请求
            RestTemplate restTemplate = new RestTemplate();
            String response = restTemplate.getForObject(gold_API_URL, String.class);
            String url = PRICE_URL+getPriceData();
            URL obj = new URL(url);
            HttpURLConnection con = (HttpURLConnection) obj.openConnection();
            con.setRequestMethod("GET");
            int responseCode = con.getResponseCode();
            BufferedReader in = new BufferedReader(new InputStreamReader(con.getInputStream()));
            String inputLine;
            StringBuffer response = new StringBuffer();
            while ((inputLine = in.readLine()) != null) {
                response.append(inputLine);
            }
            in.close();
            if(responseCode == 200){
                // 创建 ObjectMapper 对象
                ObjectMapper objectMapper = new ObjectMapper();

            // 解析返回的CSV格式数据，去除可能存在的换行符
            if (response != null && !response.contains("参数错误")) {
                // 清除换行符并按逗号分割数据
                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("黄金定时任务------没有接收到数据");
                // 解析 JSON 字符串
                JsonNode rootNode = objectMapper.readTree(response.toString());

                // 获取 "data" 节点下的 "tick_list" 数组
                JsonNode tickList = rootNode.path("data").path("tick_list");

                // 遍历 tick_list 数组并提取 code 和 price
                for (JsonNode tick : tickList) {
                    String code = tick.path("code").asText().equals("GOLD") ? "XAUUSD" : tick.path("code").asText();
                    String price = tick.path("price").asText();
                    RedisShardedPoolUtils.set(RedisKeyConstant.getRedisKey(code), price);
                }
            }else{
                log.info("黄金原油获实时价格定时任务------没有接收到数据:"+response);
            }
        } catch (Exception e) {
            e.printStackTrace();
            log.error("黄金定时任务------请求或数据解析失败：" + e.getMessage());
        }
    }

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

            // 解析返回的CSV格式数据，去除可能存在的换行符
            if (response != null && !response.contains("参数错误")) {
                // 清除换行符并按逗号分割数据
                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) {
            // 将 JSON 对象转换为字符串
            String jsonString = jsonObject.toString();
            // 使用 URLEncoder 编码 JSON 字符串
            String encodedJson = URLEncoder.encode(jsonString, "UTF-8");
            return encodedJson;
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
            log.error("原油定时任务------请求或数据解析失败：" + e.getMessage());
            return null;
        }
    }


    //黄金
    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);
            if(!g.contains("参数错误")){
                RedisShardedPoolUtils.set("k_gold_"+str, g);
                Thread.sleep(10000);
        // 创建Map对象
        Map<String, Integer> map = new HashMap<>();

        // 写死键值对
        map.put("d", 8);
        map.put("w", 9);
        map.put("m", 10);
        map.put("1", 1);
        map.put("5", 2);
        map.put("30", 4);

        for (Map.Entry<String, Integer> entry : map.entrySet()) {
            String gold = getResp(K_URL + getKQueryData(entry.getValue(), "XAUUSD"));
            if(StringUtils.isNotEmpty(gold)){
                if(entry.getKey().equals("d")){
                    priceLimit(gold,"XAUUSD");
                }
                RedisShardedPoolUtils.set("k_gold_"+entry.getKey(), gold);
            }
            String c = restTemplate.getForObject(k_crude_oil_API_URL+str, String.class);
            if(!c.contains("参数错误")){
                RedisShardedPoolUtils.set("k_crude_oil_"+str, c);
                Thread.sleep(10000);

            Thread.sleep(2000);
            String c = getResp(K_URL+getKQueryData(entry.getValue(),"USOIL"));
            if(StringUtils.isNotEmpty(c)){
                if(entry.getKey().equals("d")) {
                    priceLimit(c, "USOIL");
                }
                RedisShardedPoolUtils.set("k_crude_oil_"+entry.getKey(), c);
            }
            Thread.sleep(2000);
        }
    }
    /**
     * 同步美国股票
     */
//    @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("");
//
//            }
//        }
//
//    }

    @lombok.Data
    class kData {
        long t;
        String c;
        String o;
        String h;
        String l;
        String v;
        String vo;
    }
    public void priceLimit(String data,String key){
        List<kData> kDataList = new ArrayList<>();
        // 使用 Gson 解析 JSON 字符串
        Gson gson = new Gson();
        kResponse kResponse = gson.fromJson(data, kResponse.class);

        // 打印 kline_list 的内容
        for (kResponse.KlineData item : kResponse.getData().getKlineList()) {
            kData kData = new kData();
            kData.setT(item.getTimestamp());
            kData.setC(item.getClose_price());
            kData.setO(item.getOpen_price());
            kData.setH(item.getHigh_price());
            kData.setL(item.getLow_price());
            kData.setV(item.getVolume());
            kData.setVo(item.getTurnover());
            kDataList.add(kData);
        }
        double oneC = Double.valueOf(kDataList.get(kDataList.size() - 1).getC());
        double twoC = Double.valueOf(kDataList.get(kDataList.size() - 2).getC());
        String h = String.valueOf(((oneC - twoC) / oneC * 100));
        BigDecimal bd = new BigDecimal(h);
        bd = bd.setScale(2, RoundingMode.DOWN);  // RoundingMode.DOWN 表示截断而非四舍五入
        RedisShardedPoolUtils.set(key+"_H",bd.toString());
        RedisShardedPoolUtils.set(key+"_H",bd.toString());
    }

    public String getResp(String url){
        try {
            URL obj = new URL(url);
            HttpURLConnection con = (HttpURLConnection) obj.openConnection();
            con.setRequestMethod("GET");
            int responseCode = con.getResponseCode();
            BufferedReader in = new BufferedReader(new InputStreamReader(con.getInputStream()));
            String inputLine;
            StringBuffer response = new StringBuffer();
            while ((inputLine = in.readLine()) != null) {
                response.append(inputLine);
            }
            in.close();
            if(responseCode == 200){
                return response.toString();
            }
        }catch (Exception e){
            e.printStackTrace();
            log.error("k线请求错误！", e.getMessage());
        }
        return null;
    }

    //k线请求组装数据
    public String getKQueryData(Integer k_data,String code){
        JSONObject queryData = new JSONObject();
        queryData.put("trace", generateTraceCode());
        JSONObject data = new JSONObject();
        data.put("code", code);//产品code
        /**
         * k线类型
         * 1、1是1分钟K，2是5分钟K，3是15分钟K，4是30分钟K，5是小时K，6是2小时K(股票不支持2小时)，7是4小时K(股票不支持4小时)，8是日K，9是周K，10是月K （注：股票不支持2小时K、4小时K）
         * 2、最短的k线只支持1分钟
         * 3、查询昨日收盘价，kline_type 传8
         */
        data.put("kline_type", k_data);
        /**
         * 从指定时间往前查询K线
         * 1、传0表示从当前最新的交易日往前查k线
         * 2、指定时间请传时间戳，传时间戳表示从该时间戳往前查k线
         * 3、只有外汇贵金属加密货币支持传时间戳，股票类的code不支持
         */
        data.put("kline_timestamp_end", 0);
        /**
         * 1、表示查询多少根K线，每次最大请求1000根，可根据时间戳循环往前请求
         * 2、通过该字段可查询昨日收盘价，kline_type 传8，query_kline_num传2，返回2根k线数据中，时间戳较小的数据是昨日收盘价
         */
        data.put("query_kline_num", 200);
        /*
            复权类型,对于股票类的code才有效，例如：0:除权,1:前复权，目前仅支持0
         */
        data.put("adjust_type", 0);
        queryData.put("data", data);
        return buildUrlWithQuery(queryData);
    }


    /**
     * 加载所有股票数据