|
import java.io.IOException;
import java.io.Serializable;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.List;import javax.websocket.ClientEndpoint;
import javax.websocket.CloseReason;
import javax.websocket.ContainerProvider;
import javax.websocket.DeploymentException;
import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.WebSocketContainer;
@ClientEndpoint
public class WebSocketClient implements Serializable {private static final long serialVersionUID = 1L;
private Session session = null;
@OnOpen
public void onOpen(Session session) {
setSession(session);
}
@OnClose
public void onClose(Session session, CloseReason closeReason) {
setSession(null);
}
@OnMessage
public void onMessage(String message) {
}
public void connect(String serverURI) throws WebSocketException {
if (isSessionValid())
disconnect();
WebSocketContainer container = ContainerProvider.getWebSocketContainer();
try {
container.connectToServer(this, new URI(serverURI));
} catch (DeploymentException | IOException | URISyntaxException e) {
throw new WebSocketException(“An exception occured during connecting to web socket server.”, e);
}
}
public void disconnect() throws WebSocketException {
if (isSessionValid())
try {
getSession().close();
} catch (IOException e) {
throw new WebSocketException(“An exception occured during closing connection.”, e);
}
}
public boolean isSessionValid() {
return getSession() != null && getSession().isOpen();
}
public Session getSession() {
return session;
}
private void setSession(Session session) {
this.session = session;
}
} |
