From da335bddc242b2872029003332ef7549cceb0ce8 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 29 May 2024 19:55:01 +0800
Subject: [PATCH] 1
---
src/main/java/com/nq/utils/task/stock/StockTask.java | 125 ++++++++++++++++++++++++++++++++++++++++-
1 files changed, 121 insertions(+), 4 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 9d11dee..8bacf30 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,23 @@
package com.nq.utils.task.stock;
+import cn.hutool.core.date.DateUtil;
+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.common.ServerResponse;
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.IUserAssetsServices;
+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 +29,9 @@
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
+import java.util.concurrent.locks.Lock;
+import java.util.concurrent.locks.ReentrantLock;
+import java.util.stream.Collectors;
@Component
@@ -30,10 +41,34 @@
@Autowired
StockMapper stockMapper;
+ @Autowired
+ IUserPositionService userPositionService;
+
+ @Autowired
+ UserPositionMapper userPositionMapper;
@Autowired
IMandatoryLiquidationService mandatoryLiquidationService;
+ @Autowired
+ UserStockSubscribeMapper userStockSubscribeMapper;
+
+
+ @Autowired
+ StockSubscribeMapper stockSubscribeMapper;
+
+ @Autowired
+ IUserPositionService iUserPositionService;
+
+
+ @Autowired
+ IUserAssetsServices userAssetsServices;
+
+ private final Lock ballotLock = new ReentrantLock();
+
+ private final Lock subscriptionLock = new ReentrantLock();
+
+ private final Lock stockConstraintLock = new ReentrantLock();
private static final Logger log = LoggerFactory.getLogger(StockTask.class);
@@ -42,10 +77,9 @@
/**
* 同步系统所需要的股票
* */
- @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);
}
@@ -55,7 +89,7 @@
/**
* 同步美国股票
* */
- @Scheduled(cron = "0 0/30 * * * ?")
+// @Scheduled(cron = "0 0/30 * * * ?")
public void loadStockCompanies(){
loadAllCompanies();
}
@@ -147,4 +181,87 @@
log.error("同步出错", e);
}
}
+
+ @Scheduled(cron = "0 0/10 * * * ?")
+ public void ballot() {
+ if (ballotLock.tryLock()) {
+ try {
+ log.info("新股上市转持仓定时任务--------->开始");
+ List<StockSubscribe> stockSubscribes = stockSubscribeMapper.selectList(new LambdaQueryWrapper<StockSubscribe>()
+ .lt(StockSubscribe::getListDate, DateUtil.date()));
+ if (CollectionUtils.isNotEmpty(stockSubscribes)) {
+ List<String> codeList = stockSubscribes.stream().map(StockSubscribe::getCode).collect(Collectors.toList());
+ List<UserStockSubscribe> userStockSubscribes = userStockSubscribeMapper.selectList(new LambdaQueryWrapper<UserStockSubscribe>()
+ .eq(UserStockSubscribe::getStatus, 4).in(UserStockSubscribe::getNewCode, codeList));
+ //订单转持仓
+ userStockSubscribes.forEach(f -> {
+ ServerResponse serverResponse = iUserPositionService.newStockToPosition(f.getId());//转持仓
+ if(serverResponse.isSuccess()){
+ f.setStatus(5);
+ userStockSubscribeMapper.updateById(f);
+ }else{
+ log.info("新股上市转持仓失败申购订单id:"+f.getId()+",失败原因："+serverResponse.getMsg());
+ }
+ });
+ }
+ log.info("新股上市转持仓定时任务--------->结束");
+ } catch (Exception e) {
+ log.error("新股上市转持仓定时任务发生异常", e);
+ } finally {
+ ballotLock.unlock();
+ }
+ } else {
+ log.info("新股上市转持仓定时任务--------->上次任务还未执行完成，本次任务忽略");
+ }
+ }
+
+// @Scheduled(cron = "0/5 * * * * ?")
+// public void subscription() {
+// if (subscriptionLock.tryLock()) {
+// try {
+// log.info("自动转已认缴--------->开始");
+// List<UserStockSubscribe> userStockSubscribes = userStockSubscribeMapper.selectList(new LambdaQueryWrapper<UserStockSubscribe>()
+// .eq(UserStockSubscribe::getStatus, 3));
+// userStockSubscribes.forEach(f->{
+// Stock stock = stockMapper.selectOne(new LambdaQueryWrapper<Stock>().eq(Stock::getStockCode, f.getNewCode()));
+// UserAssets userAssets = userAssetsServices.assetsByTypeAndUserId(stock.getStockType(),f.getUserId());
+// if(null != userAssets && userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) == 0){
+// f.setStatus(4);
+// userStockSubscribeMapper.updateById(f);
+// }
+// });
+// log.info("自动转已认缴--------->结束");
+// } catch (Exception e) {
+// log.error("自动转已认缴定时任务发生异常", e);
+// } finally {
+// subscriptionLock.unlock();
+// }
+// } else {
+// log.info("自动转已认缴定时任务--------->上次任务还未执行完成，本次任务忽略");
+// }
+// }
+
+ /**
+ * 强制平仓
+ */
+// @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("强制平仓任务--------->上次任务还未执行完成，本次任务忽略");
+ }
+ }
}
--
Gitblit v1.9.3