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);
}
}