From 23c07b71a269f2540e5c09eb8d2002d9f20a857e Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Thu, 25 Apr 2024 16:07:43 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/utils/task/stock/StockTask.java |  104 +++++++++++++++++++++++++++-------------------------
 1 files changed, 54 insertions(+), 50 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 1ed3056..0473222 100644
--- a/src/main/java/com/nq/utils/task/stock/StockTask.java
+++ b/src/main/java/com/nq/utils/task/stock/StockTask.java
@@ -1,15 +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.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 +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
@@ -30,6 +37,15 @@
     @Autowired
     StockMapper stockMapper;
 
+    @Autowired
+    IUserPositionService userPositionService;
+
+    @Autowired
+    UserPositionMapper userPositionMapper;
+
+    private final Lock stockConstraintLock = new ReentrantLock();
+
+
 
     @Autowired
     IMandatoryLiquidationService mandatoryLiquidationService;
@@ -38,26 +54,14 @@
 
     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 +70,7 @@
     /**
      * 同步美国股票
      * */
-    @Scheduled(cron = "0 0/30 * * * ?")
+//    @Scheduled(cron = "0 0/30 * * * ?")
     public  void loadStockCompanies(){
         loadAllCompanies();
     }
@@ -128,16 +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);
             }
@@ -148,38 +163,27 @@
         }
     }
 
-
-
-
-    /***
-     * 每3同步一次数据源到数据库
-     *
+    /**
+     * 强制平仓
      */
-    @Scheduled(cron = "0/3 * * * * ?")
-    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/1 * * * * ?")
+    public void stockConstraint(){
+        if (stockConstraintLock.tryLock()) {
+            log.info("强制平仓任务:--------->开始");
             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){
-
+                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("强制平仓任务--------->上次任务还未执行完成,本次任务忽略");
         }
-
     }
 }

--
Gitblit v1.9.3