com.github.netty.protocol.servlet.websocket.WebSocketServerContainer Maven / Gradle / Ivy
package com.github.netty.protocol.servlet.websocket;
import javax.websocket.*;
import javax.websocket.server.ServerContainer;
import javax.websocket.server.ServerEndpointConfig;
import java.io.IOException;
import java.net.URI;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
/**
* WebSocket Server Container. Note: only the server is implemented, but the client is not
*
* @author wangzihao
*/
public class WebSocketServerContainer implements WebSocketContainer, ServerContainer {
private static final CloseReason AUTHENTICATED_HTTP_SESSION_CLOSED = new CloseReason(CloseReason.CloseCodes.VIOLATED_POLICY,
"This connection was established under an authenticated " +
"HTTP session that has ended.");
private final Map> endpointSessionMap = new HashMap<>();
private final Map sessions = new ConcurrentHashMap<>();
private final Object endPointSessionMapLock = new Object();
private final ConcurrentMap> authenticatedSessions = new ConcurrentHashMap<>();
private long defaultAsyncTimeout = -1;
private int maxBinaryMessageBufferSize = 8 * 1024;
private int maxTextMessageBufferSize = 8 * 1024;
private volatile long defaultMaxSessionIdleTimeout = 0;
public WebSocketServerContainer() {
}
@Override
public long getDefaultMaxSessionIdleTimeout() {
return defaultMaxSessionIdleTimeout;
}
@Override
public void setDefaultMaxSessionIdleTimeout(long timeout) {
this.defaultMaxSessionIdleTimeout = timeout;
}
@Override
public int getDefaultMaxBinaryMessageBufferSize() {
return maxBinaryMessageBufferSize;
}
@Override
public void setDefaultMaxBinaryMessageBufferSize(int max) {
maxBinaryMessageBufferSize = max;
}
@Override
public int getDefaultMaxTextMessageBufferSize() {
return maxTextMessageBufferSize;
}
@Override
public void setDefaultMaxTextMessageBufferSize(int max) {
maxTextMessageBufferSize = max;
}
/**
* {@inheritDoc}
*
* Currently, this implementation does not support any extensions.
*/
@Override
public Set getInstalledExtensions() {
WsExtension deflate = new WsExtension("permessage-deflate");
// boolean clientContextTakeover = true;
// boolean serverContextTakeover = true;
// int serverMaxWindowBits = -1;
// int clientMaxWindowBits = -1;
// if (!serverContextTakeover) {
// deflate.addParameter("server_no_context_takeover", null);
// }
// if (serverMaxWindowBits != -1) {
// deflate.addParameter("server_max_window_bits",
// Integer.toString(serverMaxWindowBits));
// }
// if (!clientContextTakeover) {
// deflate.addParameter("client_no_context_takeover", null);
// }
// if (clientMaxWindowBits != -1) {
// deflate.addParameter("client_max_window_bits",
// Integer.toString(clientMaxWindowBits));
// }
return Collections.singleton(deflate);
}
/**
* {@inheritDoc}
*
* The default value for this implementation is -1.
*/
@Override
public long getDefaultAsyncSendTimeout() {
return defaultAsyncTimeout;
}
/**
* {@inheritDoc}
*
* The default value for this implementation is -1.
*/
@Override
public void setAsyncSendTimeout(long timeout) {
this.defaultAsyncTimeout = timeout;
}
@Override
public Session connectToServer(Object endpoint, URI path) throws DeploymentException, IOException {
return null;
}
@Override
public Session connectToServer(Class annotatedEndpointClass, URI path) throws DeploymentException, IOException {
return null;
}
@Override
public Session connectToServer(Endpoint endpoint, ClientEndpointConfig clientEndpointConfiguration, URI path) throws DeploymentException, IOException {
return null;
}
@Override
public Session connectToServer(Class endpoint, ClientEndpointConfig clientEndpointConfiguration, URI path) throws DeploymentException, IOException {
return null;
}
@Override
public void addEndpoint(Class clazz) throws DeploymentException {
}
@Override
public void addEndpoint(ServerEndpointConfig sec) throws DeploymentException {
}
public void registerSession(Endpoint endpoint, WebSocketSession wsSession) {
if (!wsSession.isOpen()) {
// The session was closed during onOpen. No need to supportPipeline it.
return;
}
synchronized (endPointSessionMapLock) {
// if (endpointSessionMap.size() == 0) {
// BackgroundProcessManager.getInstance().supportPipeline(this);
// }
Set wsSessions = endpointSessionMap.get(endpoint);
if (wsSessions == null) {
wsSessions = new HashSet<>();
endpointSessionMap.put(endpoint, wsSessions);
}
wsSessions.add(wsSession);
}
sessions.put(wsSession, wsSession);
}
public void unregisterSession(Endpoint endpoint, WebSocketSession wsSession) {
synchronized (endPointSessionMapLock) {
Set wsSessions = endpointSessionMap.get(endpoint);
if (wsSessions != null) {
wsSessions.remove(wsSession);
if (wsSessions.isEmpty()) {
endpointSessionMap.remove(endpoint);
}
}
}
sessions.remove(wsSession);
}
public void registerAuthenticatedSession(WebSocketSession wsSession,
String httpSessionId) {
Set wsSessions = authenticatedSessions.get(httpSessionId);
if (wsSessions == null) {
wsSessions = Collections.newSetFromMap(
new ConcurrentHashMap());
authenticatedSessions.putIfAbsent(httpSessionId, wsSessions);
wsSessions = authenticatedSessions.get(httpSessionId);
}
wsSessions.add(wsSession);
}
public void unregisterAuthenticatedSession(WebSocketSession wsSession,
String httpSessionId) {
Set wsSessions = authenticatedSessions.get(httpSessionId);
// wsSessions will be null if the HTTP session has ended
if (wsSessions != null) {
wsSessions.remove(wsSession);
}
}
public void closeAuthenticatedSession(String httpSessionId) {
Set wsSessions = authenticatedSessions.remove(httpSessionId);
if (wsSessions != null && !wsSessions.isEmpty()) {
for (Session wsSession : wsSessions) {
try {
wsSession.close(AUTHENTICATED_HTTP_SESSION_CLOSED);
} catch (IOException e) {
// Any IOExceptions during close will have been caught and the
// onError method called.
}
}
}
}
public Set getOpenSessions(Endpoint endpoint) {
HashSet result = new HashSet<>();
synchronized (endPointSessionMapLock) {
Set sessions = endpointSessionMap.get(endpoint);
if (sessions != null) {
result.addAll(sessions);
}
}
return result;
}
public static class WsExtension implements Extension {
private final String name;
private final List parameters = new ArrayList<>();
WsExtension(String name) {
this.name = name;
}
@Override
public boolean equals(Object obj) {
if (obj instanceof Extension) {
return Objects.equals(name, ((Extension) obj).getName());
} else {
return false;
}
}
@Override
public int hashCode() {
return name.hashCode();
}
public void addParameter(String name, String value) {
parameters.add(new WsExtensionParameter(name, value));
}
@Override
public String getName() {
return name;
}
@Override
public List getParameters() {
return parameters;
}
public static class WsExtensionParameter implements Parameter {
private final String name;
private final String value;
WsExtensionParameter(String name, String value) {
this.name = name;
this.value = value;
}
@Override
public String getName() {
return name;
}
@Override
public String getValue() {
return value;
}
}
}
}