package org.example.timedTask;
|
|
import cn.hutool.core.date.DateUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
|
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
import com.google.gson.Gson;
|
import lombok.extern.log4j.Log4j;
|
import lombok.extern.slf4j.Slf4j;
|
import org.example.dao.StockMapper;
|
import org.example.dao.StockMarketNewMapper;
|
import org.example.enums.EStockType;
|
import org.example.pojo.DataStockBean;
|
import org.example.pojo.ReponseBase;
|
import org.example.pojo.Stock;
|
import org.example.pojo.StockMarketNew;
|
import org.example.util.HttpClientRequest;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.stereotype.Component;
|
|
import java.math.BigDecimal;
|
import java.sql.Timestamp;
|
import java.text.SimpleDateFormat;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.concurrent.locks.Lock;
|
import java.util.concurrent.locks.ReentrantLock;
|
|
|
/**
|
* @program: webSocketProject
|
* @description:
|
* @create: 2024-03-26 19:11
|
**/
|
@Component
|
@Slf4j
|
public class StockTask {
|
|
|
@Autowired
|
StockMapper stockMapper;
|
|
@Autowired
|
StockMarketNewMapper stockMarketNewMapper;
|
|
private final Lock lock = new ReentrantLock();
|
|
@Scheduled(cron = "0 * 0/6 * * ?")
|
public void syncINStockData() {
|
if (lock.tryLock()) {
|
log.info("同步印度股票--------->开始");
|
try {
|
loadAllStock(EStockType.IN);
|
} finally {
|
lock.unlock();
|
log.info("同步印度股票--------->结束");
|
}
|
} else {
|
log.info("同步印度股票--------->上次任务还未执行完成，本次任务忽略");
|
}
|
}
|
|
/**
|
* 加载所有股票数据
|
*/
|
public void loadAllStock(EStockType eStockType) {
|
log.info("同步股票 数据 {}", eStockType.getCode());
|
List<DataStockBean> list = new ArrayList<>();
|
int totleStock = 1;
|
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);
|
list.addAll(reponseBase.getData());
|
page++;
|
totleStock = reponseBase.getTotal();
|
} catch (Exception e) {
|
e.printStackTrace();
|
break;
|
}
|
}
|
for (DataStockBean o : list) {
|
Stock stock = stockMapper.selectOne(new LambdaQueryWrapper<Stock>().eq(Stock::getStockCode, o.getId()));
|
if (stock == null) {
|
stock = new Stock();
|
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());
|
}
|
if (eStockType.getCode().equals("US")) {
|
stock.setStockSpell(o.getSymbol());
|
} else {
|
stock.setStockSpell(o.getTicker());
|
}
|
|
stock.setIsLock(0);
|
stock.setIsShow(0);
|
stock.setDataBase(0);
|
stock.setAddTime(new Date());
|
stockMapper.insert(stock);
|
}
|
StockMarketNew stockMarketNew = new StockMarketNew();
|
conv(o,stockMarketNew);
|
StockMarketNew selectById = stockMarketNewMapper.selectById(o.getId());
|
if(null == selectById){
|
stockMarketNewMapper.insert(stockMarketNew);
|
}else{
|
stockMarketNewMapper.updateById(stockMarketNew);
|
}
|
}
|
log.info("同步股票 数据 成功 {} 总共同步数据 {}", eStockType.getCode(), list.size());
|
} catch (
|
Exception e) {
|
log.error("同步出错", e);
|
}
|
}
|
|
public void conv(DataStockBean o,StockMarketNew stockMarketNew){
|
stockMarketNew.setAvgVolume(o.getAvgVolume());
|
stockMarketNew.setChg(o.getChg());
|
stockMarketNew.setChgPct(o.getChgPct());
|
stockMarketNew.setFundamentalBeta(o.getFundamentalBeta());
|
stockMarketNew.setFundamentalDividend(o.getFundamentalDividend());
|
stockMarketNew.setFundamentalEps(o.getFundamentalEps());
|
stockMarketNew.setFundamentalMarketCap(o.getFundamentalMarketCap());
|
stockMarketNew.setFundamentalRatio(o.getFundamentalRatio());
|
stockMarketNew.setFundamentalRevenue(o.getFundamentalRevenue());
|
stockMarketNew.setFundamentalSharesOutstanding(o.getFundamentalSharesOutstanding());
|
stockMarketNew.setHigh(o.getHigh());
|
stockMarketNew.setId(Integer.parseInt(o.getId()));
|
stockMarketNew.setIsOpen(o.getIsOpen());
|
stockMarketNew.setLast(o.getLast());
|
stockMarketNew.setLastPairDecimal(o.getLastPairDecimal());
|
stockMarketNew.setLow(o.getLow());
|
stockMarketNew.setName(o.getName());
|
stockMarketNew.setOpen(o.getOpen());
|
stockMarketNew.setPrevClose(o.getPrevClose());
|
stockMarketNew.setSymbol(o.getSymbol());
|
stockMarketNew.setTime(o.getTime());
|
stockMarketNew.setVolume(o.getVolume());
|
stockMarketNew.setType(o.getType());
|
}
|
|
}
|
