MockEventSocket.java

package org.apache.zeppelin.notebook.repo.zeppelinhub.websocket.mock;

import org.eclipse.jetty.websocket.api.Session;
import org.eclipse.jetty.websocket.api.WebSocketAdapter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class MockEventSocket extends WebSocketAdapter {
  private static final Logger LOG = LoggerFactory.getLogger(MockEventServlet.class);
  private Session session;

  @Override
  public void onWebSocketConnect(Session session) {
    super.onWebSocketConnect(session);
    this.session = session;
    LOG.info("Socket Connected: " + session);
  }

  @Override
  public void onWebSocketText(String message) {
    super.onWebSocketText(message);
    session.getRemote().sendStringByFuture(message);
    LOG.info("Received TEXT message: {}", message);
    
  }

  @Override
  public void onWebSocketClose(int statusCode, String reason) {
    super.onWebSocketClose(statusCode, reason);
    LOG.info("Socket Closed: [{}] {}", statusCode, reason);
  }

  @Override
  public void onWebSocketError(Throwable cause) {
    super.onWebSocketError(cause);
    LOG.error("Websocket error: {}", cause);
  }
}