info.bitrich.xchangestream.bitfinex.BitfinexStreamingTradeService Maven / Gradle / Ivy

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

import info.bitrich.xchangestream.bitfinex.dto.BitfinexWebSocketAuthOrder;
import info.bitrich.xchangestream.bitfinex.dto.BitfinexWebSocketAuthPreTrade;
import info.bitrich.xchangestream.bitfinex.dto.BitfinexWebSocketAuthTrade;
import info.bitrich.xchangestream.core.StreamingTradeService;
import io.reactivex.Observable;
import java.util.function.Function;
import org.knowm.xchange.currency.CurrencyPair;
import org.knowm.xchange.dto.Order;
import org.knowm.xchange.dto.trade.UserTrade;
import org.knowm.xchange.exceptions.ExchangeSecurityException;

public class BitfinexStreamingTradeService implements StreamingTradeService {

  private final BitfinexStreamingService service;

  public BitfinexStreamingTradeService(BitfinexStreamingService service) {
    this.service = service;
  }

  public Observable getOrderChanges() {
    return getRawAuthenticatedOrders()
        .filter(o -> o.getId() != 0)
        .map(BitfinexStreamingAdapters::adaptOrder)
        .doOnNext(
            o -> {
              service.scheduleCalculatedBalanceFetch(o.getCurrencyPair().base.getCurrencyCode());
              service.scheduleCalculatedBalanceFetch(o.getCurrencyPair().counter.getCurrencyCode());
            });
  }

  @Override
  public Observable getOrderChanges(CurrencyPair currencyPair, Object... args) {
    return getOrderChanges().filter(o -> currencyPair.equals(o.getCurrencyPair()));
  }

  /**
   * Gets a stream of all user trades to which we are subscribed.
   *
   * @return The stream of user trades.
   */
  public Observable getUserTrades() {
    return getRawAuthenticatedTrades()
        .filter(o -> o.getId() != 0)
        .map(BitfinexStreamingAdapters::adaptUserTrade)
        .doOnNext(
            t -> {
              service.scheduleCalculatedBalanceFetch(t.getCurrencyPair().base.getCurrencyCode());
              service.scheduleCalculatedBalanceFetch(t.getCurrencyPair().counter.getCurrencyCode());
            });
  }

  @Override
  public Observable getUserTrades(CurrencyPair currencyPair, Object... args) {
    return getUserTrades().filter(t -> currencyPair.equals(t.getCurrencyPair()));
  }

  public Observable getRawAuthenticatedOrders() {
    return withAuthenticatedService(BitfinexStreamingService::getAuthenticatedOrders);
  }

  public Observable getRawAuthenticatedPreTrades() {
    return withAuthenticatedService(BitfinexStreamingService::getAuthenticatedPreTrades);
  }

  public Observable getRawAuthenticatedTrades() {
    return withAuthenticatedService(BitfinexStreamingService::getAuthenticatedTrades);
  }

  private  Observable withAuthenticatedService(
      Function> serviceConsumer) {
    if (!service.isAuthenticated()) {
      throw new ExchangeSecurityException("Not authenticated");
    }
    return serviceConsumer.apply(service);
  }
}