From decfc2ff7244cec365d3f32a5090edf5faeae861 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 23 Apr 2025 19:08:18 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/utils/task/stock/StockTask.java |  252 ++++++++++++++++++++++++++++++++++---------------
 1 files changed, 173 insertions(+), 79 deletions(-)

diff --git a/src/main/java/com/nq/utils/task/stock/StockTask.java b/src/main/java/com/nq/utils/task/stock/StockTask.java
index 0473222..f7787b7 100644
--- a/src/main/java/com/nq/utils/task/stock/StockTask.java
+++ b/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 0 0/2  * * ?")
+     */
+    @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("强制平仓任务--------->上次任务还未执行完成,本次任务忽略");
-        }
-    }
+//    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("强制平仓任务--------->上次任务还未执行完成,本次任务忽略");
+//        }
+//    }
 }

--
Gitblit v1.9.3