info.bitrich.xchangestream.bitstamp.v2.BitstampStreamingService Maven / Gradle / Ivy

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

import com.fasterxml.jackson.databind.JsonNode;
import info.bitrich.xchangestream.bitstamp.v2.dto.BitstampWebSocketData;
import info.bitrich.xchangestream.bitstamp.v2.dto.BitstampWebSocketSubscriptionMessage;
import info.bitrich.xchangestream.service.netty.JsonNettyStreamingService;
import io.netty.handler.codec.http.websocketx.extensions.WebSocketClientExtensionHandler;
import java.io.IOException;
import java.time.Duration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Bitstamp WebSocket V2 streaming service implementation Created by Pavel Chertalev on 15.03.2018.
 */
public class BitstampStreamingService extends JsonNettyStreamingService {
  private static final Logger LOG = LoggerFactory.getLogger(BitstampStreamingService.class);

  private static final String JSON_CHANNEL = "channel";
  private static final String JSON_EVENT = "event";

  public static final String EVENT_ORDERBOOK = "data";
  public static final String EVENT_TRADE = "trade";
  private static final String EVENT_SUBSCRIPTION_SUCCEEDED = "bts:subscription_succeeded";
  private static final String EVENT_UNSUBSCRIPTION_SUCCEEDED = "bts:unsubscription_succeeded";

  public BitstampStreamingService(String apiUrl) {
    super(apiUrl, Integer.MAX_VALUE);
  }

  public BitstampStreamingService(
      String apiUrl,
      int maxFramePayloadLength,
      Duration connectionTimeout,
      Duration retryDuration,
      int idleTimeoutSeconds) {
    super(apiUrl, maxFramePayloadLength, connectionTimeout, retryDuration, idleTimeoutSeconds);
  }

  @Override
  protected WebSocketClientExtensionHandler getWebSocketClientExtensionHandler() {
    return null;
  }

  @Override
  protected String getChannelNameFromMessage(JsonNode message) throws IOException {
    JsonNode jsonNode = message.get(JSON_CHANNEL);
    if (jsonNode != null) {
      return jsonNode.asText();
    }
    throw new IOException("Channel name can't be evaluated from message");
  }

  @Override
  protected void handleMessage(JsonNode message) {
    JsonNode channelJsonNode = message.get(JSON_CHANNEL);
    JsonNode eventJsonNode = message.get(JSON_EVENT);

    if (channelJsonNode == null || eventJsonNode == null) {
      LOG.error(
          "Received JSON message does not contain {} and {} fields. Skipped...",
          JSON_CHANNEL,
          JSON_EVENT);
      return;
    }

    String channel = channelJsonNode.asText();
    String event = eventJsonNode.asText();

    switch (event) {
      case EVENT_ORDERBOOK:
      case EVENT_TRADE:
        if (!channels.containsKey(channel)) {
          LOG.warn(
              "The message has been received from disconnected channel '{}'. Skipped.", channel);
          return;
        }
        super.handleMessage(message);
        break;
      case EVENT_SUBSCRIPTION_SUCCEEDED:
        LOG.info("Channel {} has been successfully subscribed", channel);
        break;
      case EVENT_UNSUBSCRIPTION_SUCCEEDED:
        LOG.info("Channel {} has been successfully unsubscribed", channel);
        break;
      default:
        LOG.warn("Unsupported event type {} in message {}", event, message.toString());
    }
  }

  @Override
  public String getSubscribeMessage(String channelName, Object... args) throws IOException {
    BitstampWebSocketSubscriptionMessage subscribeMessage =
        generateSubscribeMessage(channelName, "bts:subscribe");
    return objectMapper.writeValueAsString(subscribeMessage);
  }

  @Override
  public String getUnsubscribeMessage(String channelName, Object... args) throws IOException {
    BitstampWebSocketSubscriptionMessage subscribeMessage =
        generateSubscribeMessage(channelName, "bts:unsubscribe");
    return objectMapper.writeValueAsString(subscribeMessage);
  }

  private BitstampWebSocketSubscriptionMessage generateSubscribeMessage(
      String channelName, String channel) {
    return new BitstampWebSocketSubscriptionMessage(
        channel, new BitstampWebSocketData(channelName));
  }
}