com.applitools.eyes.universal.USDKConnection Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of eyes-sdk-core-java5 Show documentation

Applitools Eyes SDK base for Java

There is a newer version: 5.76.0

package com.applitools.eyes.universal;

import com.applitools.eyes.EyesException;
import com.applitools.eyes.Logger;
import com.applitools.eyes.SyncTaskListener;
import com.applitools.eyes.universal.dto.Command;
import com.applitools.eyes.universal.dto.EventDto;
import com.applitools.eyes.universal.dto.RequestDto;
import com.applitools.eyes.universal.dto.ResponseDto;
import com.applitools.eyes.universal.server.UniversalSdkNativeLoader;
import org.asynchttpclient.AsyncHttpClient;
import org.asynchttpclient.DefaultAsyncHttpClientConfig;
import org.asynchttpclient.Dsl;
import org.asynchttpclient.ws.WebSocket;
import org.asynchttpclient.ws.WebSocketUpgradeHandler;

import java.io.IOException;

/**
 * universal sdk connection
 */
public class USDKConnection implements AutoCloseable {
  private WebSocket webSocket;
  protected Logger logger = new Logger();
  private AbstractSDKListener listener;
  private AsyncHttpClient client;

  public USDKConnection(AbstractSDKListener listener) {
    this.listener = listener;
  }

  public void init() {
   webSocket = openWebsocket();
   listener.setWebSocket(webSocket);
  }

  private WebSocket openWebsocket() {

    try {

      String url = String.format("ws://localhost:%s/eyes", UniversalSdkNativeLoader.getPort());
      client = Dsl.asyncHttpClient(new DefaultAsyncHttpClientConfig.Builder().setWebSocketMaxFrameSize(268435456));
      return client.prepareGet(url)
          .execute(new WebSocketUpgradeHandler.Builder().addWebSocketListener(listener).build()).get();

    } catch (Exception e) {
      throw new EyesException("Communication with server failed, reason: " + e.getMessage(), e);
    }

  }

  public synchronized SyncTaskListener executeCommand(Command command) throws Exception {
    if (command instanceof EventDto) {
      webSocket.sendTextFrame(listener.objectMapper.writeValueAsString(command));
      return null;
    }

    RequestDto request = (RequestDto) command;
    SyncTaskListener> syncTaskListener = new SyncTaskListener<>(logger, request.getKey());
    listener.map.put(request.getKey(), syncTaskListener);
    webSocket.sendTextFrame(listener.objectMapper.writeValueAsString(request));
    return syncTaskListener;
  }

  @Override
  public void close() {
    try {
      client.close();
    } catch (IOException e) {
      throw new EyesException("Failed to stop the HTTP Client!" + e.getMessage());
    }
  }
}