xiaoyong931011
2021-12-08 f5e6133809c553cfd9fb28ee61019927c547c374
src/main/java/com/xcong/excoin/quartz/job/CoinTradeInitJob.java
@@ -1,21 +1,27 @@
package com.xcong.excoin.quartz.job;
import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSON;
import com.huobi.client.SubscriptionClient;
import com.huobi.client.SubscriptionOptions;
import com.huobi.client.model.Candlestick;
import com.huobi.client.model.enums.CandlestickInterval;
import com.xcong.excoin.common.contants.AppContants;
import com.xcong.excoin.modules.coin.dao.OrderCoinDealDao;
import com.xcong.excoin.modules.coin.dao.OrderCoinsDao;
import com.xcong.excoin.modules.coin.entity.OrderCoinsEntity;
import com.xcong.excoin.modules.coin.service.OrderCoinService;
import com.xcong.excoin.modules.symbols.constants.SymbolsConstats;
import com.xcong.excoin.modules.symbols.service.SymbolsService;
import com.xcong.excoin.processor.CoinProcessor;
import com.xcong.excoin.processor.CoinProcessorFactory;
import com.xcong.excoin.processor.DefaultCoinProcessor;
import com.xcong.excoin.processor.MarketService;
import com.xcong.excoin.rabbit.pricequeue.WebsocketPriceService;
import com.xcong.excoin.rabbit.producer.ExchangeProducer;
import com.xcong.excoin.trade.CoinTrader;
import com.xcong.excoin.trade.CoinTraderFactory;
import com.xcong.excoin.trade.ExchangeTrade;
import com.xcong.excoin.utils.CoinTypeConvert;
import com.xcong.excoin.utils.RedisUtils;
import lombok.extern.slf4j.Slf4j;
@@ -28,7 +34,9 @@
import java.math.BigDecimal;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
 *  开启撮合交易
@@ -38,17 +46,19 @@
 **/
@Slf4j
@Component
//@ConditionalOnProperty(prefix = "app", name = "trade", havingValue = "true")
@ConditionalOnProperty(prefix = "app", name = "exchange-trade", havingValue = "true")
public class CoinTradeInitJob {
    @Resource
    private OrderCoinsDao orderCoinsDao;
    @Resource
    private OrderCoinDealDao orderCoinDealDao;
    @Resource
    private CoinTraderFactory factory;
    @Resource
    private OrderCoinService coinService;
    private RedisUtils redisUtils;
    @Resource
    private MarketService marketService;
@@ -56,12 +66,15 @@
    @Resource
    private CoinProcessorFactory processorFactory;
    @Resource
    ExchangeProducer exchangeProducer;
    @PostConstruct
    public void initCoinTrade() {
        log.info("#=======撮合交易器开启=======#");
        String symbol = "NEKK";
        String symbol = SymbolsConstats.ROC;
        CoinTrader newTrader = new CoinTrader(symbol);
        newTrader.setOrderCoinService(coinService);
        newTrader.setExchangeProducer(exchangeProducer);
        //newTrader.setKafkaTemplate(kafkaTemplate);
        //newTrader.setBaseCoinScale(coin.getBaseCoinScale());
        //newTrader.setCoinScale(coin.getCoinScale());
@@ -70,10 +83,18 @@
        // 创建成功以后需要对未处理订单预处理
        log.info("======CoinTrader Process: " + symbol + "======");
        List<OrderCoinsEntity> orders = orderCoinsDao.selectAllEntrustingCoinOrderList();
        List<String> symbolList = new ArrayList<>();
        symbolList.add(symbol);
        List<OrderCoinsEntity> orders = orderCoinsDao.selectCoinOrderOnTrade(symbolList);
        List<OrderCoinsEntity> tradingOrders = new ArrayList<>();
        List<OrderCoinsEntity> completedOrders = new ArrayList<>();
        orders.forEach(order -> {
            if(order.getDealCnt()==null){
                order.setDealCnt(BigDecimal.ZERO);
            }
            if(order.getDealAmount()==null){
                order.setDealAmount(BigDecimal.ZERO);
            }
            tradingOrders.add(order);
        });
        try {
@@ -88,11 +109,42 @@
        // 创建K线生成器
        CoinProcessor processor = new DefaultCoinProcessor(symbol, "USDT");
        processor.setMarketService(marketService);
        processor.setRedisUtils(redisUtils);
        //processor.setExchangeRate(exchangeRate);
        processor.initializeThumb();
        //processor.initializeUsdRate();
        processor.setIsHalt(false);
        List<ExchangeTrade> nekk = orderCoinDealDao.selectOrderCoinDealByTime(SymbolsConstats.ROC, null, null);
        processor.process(nekk);
        String symbolUsdt = symbol;
        if(!symbol.contains("USDT")){
            symbolUsdt = symbol+"/USDT";
        }
        String key = "NEW_KINE_{}";
        key = StrUtil.format(key, symbolUsdt);
        Object o = redisUtils.get(key);
        if(o!=null){
            Map<String, Candlestick> currentKlineMap = (Map<String, Candlestick> )o;
            ((DefaultCoinProcessor) processor).setCurrentKlineMap(currentKlineMap);
        }else{
            // 当最新K线不存在时 需要初始化
            // 1min 5min 15min 30min  1hour 4hour 1day 1week
            String[] rang = {"1min","5min","15min","30min","1hour","4hour","1day","1week"};
            Map<String, Candlestick> currentKlineMap = new HashMap<>();
            long currentTimeMillis = System.currentTimeMillis();
            for (String s : rang) {
                Candlestick candlestick = new Candlestick();
                candlestick.setClose(AppContants.DEFAULT_PRICE);
                candlestick.setHigh(AppContants.DEFAULT_PRICE);
                candlestick.setLow(AppContants.DEFAULT_PRICE);
                candlestick.setOpen(AppContants.DEFAULT_PRICE);
                candlestick.setTimestamp(currentTimeMillis);
                currentKlineMap.put(s,candlestick);
            }
            redisUtils.set(key,currentKlineMap);
        }
        processorFactory.addProcessor(symbol, processor);
    }
}