io.contek.invoker.deribit.api.websocket.WebSocketMessageParser Maven / Gradle / Ivy

Go to download
package io.contek.invoker.deribit.api.websocket;

import com.google.gson.Gson;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import io.contek.invoker.commons.websocket.AnyWebSocketMessage;
import io.contek.invoker.commons.websocket.IWebSocketComponent;
import io.contek.invoker.commons.websocket.WebSocketTextMessageParser;
import io.contek.invoker.deribit.api.websocket.common.WebSocketInboundMessage;
import io.contek.invoker.deribit.api.websocket.common.WebSocketResponse;
import io.contek.invoker.deribit.api.websocket.common.constants.WebSocketChannelKeys;
import io.contek.invoker.deribit.api.websocket.market.BookChangeChannel;
import io.contek.invoker.deribit.api.websocket.market.BookSnapshotChannel;
import io.contek.invoker.deribit.api.websocket.market.TradesChannel;
import io.contek.invoker.deribit.api.websocket.user.UserChangesChannel;
import io.contek.invoker.deribit.api.websocket.user.UserOrdersChannel;

import javax.annotation.concurrent.ThreadSafe;
import java.util.HashMap;
import java.util.Map;

import static java.lang.String.format;

@ThreadSafe
final class WebSocketMessageParser extends WebSocketTextMessageParser {

  private final Gson gson = new Gson();

  private final Map>> pendingRequests =
      new HashMap<>();

  public void register(Integer id, Class> type) {
    synchronized (pendingRequests) {
      pendingRequests.put(id, type);
    }
  }

  @Override
  public void register(IWebSocketComponent component) {}

  @Override
  protected AnyWebSocketMessage fromText(String text) {
    JsonElement json = gson.fromJson(text, JsonElement.class);
    if (!json.isJsonObject()) {
      throw new IllegalArgumentException(text);
    }
    JsonObject obj = json.getAsJsonObject();
    if (obj.has("id")) {
      return toResponseMessage(obj);
    } else if (obj.has("params")) {
      return toDataMessage(obj);
    } else {
      throw new IllegalArgumentException(text);
    }
  }

  private WebSocketResponse toResponseMessage(JsonObject obj) {
    int id = obj.get("id").getAsInt();
    Class> type = pendingRequests.remove(id);
    if (type == null) {
      throw new IllegalStateException(format("Expected response type not found: %d", id));
    }
    return gson.fromJson(obj, type);
  }

  private WebSocketInboundMessage toDataMessage(JsonObject obj) {
    JsonObject params = obj.getAsJsonObject("params");
    String channel = params.get("channel").getAsString();
    if (channel.startsWith(WebSocketChannelKeys._book)) {
      JsonObject data = params.getAsJsonObject("data");
      if (data.has("type")) {
        return gson.fromJson(obj, BookChangeChannel.Message.class);
      }
      return gson.fromJson(obj, BookSnapshotChannel.Message.class);
    } else if (channel.startsWith(WebSocketChannelKeys._trades)) {
      return gson.fromJson(obj, TradesChannel.Message.class);
    } else if (channel.startsWith(WebSocketChannelKeys._user_changes)) {
      return gson.fromJson(obj, UserChangesChannel.Message.class);
    }
    if (channel.startsWith(WebSocketChannelKeys._user_orders)) {
      return gson.fromJson(obj, UserOrdersChannel.Message.class);
    } else {
      throw new IllegalArgumentException(obj.toString());
    }
  }
}