info.bitrich.xchangestream.bitfinex.BitfinexStreamingAccountService 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

The newest version!

package info.bitrich.xchangestream.bitfinex;

import info.bitrich.xchangestream.bitfinex.dto.BitfinexWebSocketAuthBalance;
import info.bitrich.xchangestream.core.StreamingAccountService;
import io.reactivex.rxjava3.core.Observable;
import org.knowm.xchange.currency.Currency;
import org.knowm.xchange.dto.account.Balance;
import org.knowm.xchange.exceptions.ExchangeException;
import org.knowm.xchange.exceptions.ExchangeSecurityException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class BitfinexStreamingAccountService implements StreamingAccountService {

  private static final Logger LOG = LoggerFactory.getLogger(BitfinexStreamingAccountService.class);

  private final BitfinexStreamingService service;

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

  @Override
  public Observable getBalanceChanges(Currency currency, Object... args) {
    if (args.length == 0 || !String.class.isInstance(args[0])) {
      throw new ExchangeException("Specify wallet id to monitor balance stream");
    }
    String walletId = (String) args[0];
    return getRawAuthenticatedBalances()
        .filter(b -> b.getWalletType().equalsIgnoreCase(walletId))
        .filter(b -> currency.getCurrencyCode().equals(b.getCurrency()))
        .filter(
            b -> {
              if (b.getBalanceAvailable() == null) {
                LOG.debug(
                    "Ignoring uncalculated balance on {}-{}, scheduling calculated fetch",
                    walletId,
                    b.getCurrency());
                service.scheduleCalculatedBalanceFetch(b.getCurrency());
                return false;
              }
              return true;
            })
        .map(BitfinexStreamingAdapters::adaptBalance);
  }

  public Observable getRawAuthenticatedBalances() {
    if (!service.isAuthenticated()) {
      throw new ExchangeSecurityException("Not authenticated");
    }
    return service.getAuthenticatedBalances();
  }
}