info.bitrich.xchangestream.poloniex2.PoloniexStreamingExchange Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xchange-stream-poloniex2 Show documentation
There is a newer version: 5.2.0
Show newest version
package info.bitrich.xchangestream.poloniex2;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import info.bitrich.xchangestream.core.ProductSubscription;
import info.bitrich.xchangestream.core.StreamingExchange;
import info.bitrich.xchangestream.core.StreamingMarketDataService;
import info.bitrich.xchangestream.service.netty.ConnectionStateModel.State;
import info.bitrich.xchangestream.service.netty.StreamingObjectMapperHelper;
import io.reactivex.rxjava3.core.Completable;
import io.reactivex.rxjava3.core.Observable;
import java.io.IOException;
import java.net.URL;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import org.knowm.xchange.ExchangeSpecification;
import org.knowm.xchange.currency.Currency;
import org.knowm.xchange.currency.CurrencyPair;
import org.knowm.xchange.poloniex.PoloniexExchange;

/** Created by Lukas Zaoralek on 10.11.17. */
public class PoloniexStreamingExchange extends PoloniexExchange implements StreamingExchange {
  private static final String API_URI = "wss://api2.poloniex.com";
  private static final String TICKER_URL = "https://poloniex.com/public?command=returnTicker";

  private final PoloniexStreamingService streamingService;
  private PoloniexStreamingMarketDataService streamingMarketDataService;

  public PoloniexStreamingExchange() {
    this.streamingService = new PoloniexStreamingService(API_URI);
  }

  @Override
  protected void initServices() {
    applyStreamingSpecification(getExchangeSpecification(), streamingService);
    super.initServices();
    Map currencyPairMap = getCurrencyPairMap();
    streamingMarketDataService =
        new PoloniexStreamingMarketDataService(streamingService, currencyPairMap);
  }

  private Map getCurrencyPairMap() {
    Map currencyPairMap = new HashMap<>();
    final ObjectMapper mapper = StreamingObjectMapperHelper.getObjectMapper();

    try {
      URL tickerUrl = new URL(TICKER_URL);
      JsonNode jsonRootTickers = mapper.readTree(tickerUrl);
      Iterator pairSymbols = jsonRootTickers.fieldNames();
      pairSymbols.forEachRemaining(
          pairSymbol -> {
            String id = jsonRootTickers.get(pairSymbol).get("id").toString();
            String[] currencies = pairSymbol.split("_");
            CurrencyPair currencyPair =
                new CurrencyPair(new Currency(currencies[1]), new Currency(currencies[0]));
            currencyPairMap.put(Integer.valueOf(id), currencyPair);
          });
    } catch (IOException e) {
      e.printStackTrace();
    }

    return currencyPairMap;
  }

  @Override
  public Completable connect(ProductSubscription... args) {
    return streamingService.connect();
  }

  @Override
  public Completable disconnect() {
    return streamingService.disconnect();
  }

  @Override
  public Observable