org.apache.tomcat.websocket.WsSession Java Examples

private void registerAuthenticatedSession(WsSession wsSession,
        String httpSessionId) {
    Set<WsSession> wsSessions = authenticatedSessions.get(httpSessionId);
    if (wsSessions == null) {
        wsSessions = Collections.newSetFromMap(
                 new ConcurrentHashMap<WsSession,Boolean>());
         authenticatedSessions.putIfAbsent(httpSessionId, wsSessions);
         wsSessions = authenticatedSessions.get(httpSessionId);
    }
    wsSessions.add(wsSession);
}
 

public void closeAuthenticatedSession(String httpSessionId) {
    Set<WsSession> wsSessions = authenticatedSessions.remove(httpSessionId);

    if (wsSessions != null && !wsSessions.isEmpty()) {
        for (WsSession 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.
            }
        }
    }
}
 

private void unregisterAuthenticatedSession(WsSession wsSession,
        String httpSessionId) {
    Set<WsSession> wsSessions = authenticatedSessions.get(httpSessionId);
    // wsSessions will be null if the HTTP session has ended
    if (wsSessions != null) {
        wsSessions.remove(wsSession);
    }
}
 

private void registerAuthenticatedSession(WsSession wsSession,
        String httpSessionId) {
    Set<WsSession> wsSessions = authenticatedSessions.get(httpSessionId);
    if (wsSessions == null) {
        wsSessions = Collections.newSetFromMap(
                 new ConcurrentHashMap<WsSession,Boolean>());
         authenticatedSessions.putIfAbsent(httpSessionId, wsSessions);
         wsSessions = authenticatedSessions.get(httpSessionId);
    }
    wsSessions.add(wsSession);
}
 

/**
 * {@inheritDoc}
 *
 * Overridden to make it visible to other classes in this package.
 */
@Override
protected void unregisterSession(Endpoint endpoint, WsSession wsSession) {
    if (wsSession.getUserPrincipal() != null &&
            wsSession.getHttpSessionId() != null) {
        unregisterAuthenticatedSession(wsSession,
                wsSession.getHttpSessionId());
    }
    super.unregisterSession(endpoint, wsSession);
}
 

/**
 * {@inheritDoc}
 *
 * Overridden to make it visible to other classes in this package.
 */
@Override
protected void registerSession(Endpoint endpoint, WsSession wsSession) {
    super.registerSession(endpoint, wsSession);
    if (wsSession.isOpen() &&
            wsSession.getUserPrincipal() != null &&
            wsSession.getHttpSessionId() != null) {
        registerAuthenticatedSession(wsSession,
                wsSession.getHttpSessionId());
    }
}
 

@Override
public final void onMessage(T message, boolean last) {
    if (params.length == 1 && params[0] instanceof DecodeException) {
        ((WsSession) session).getLocal().onError(session,
                (DecodeException) params[0]);
        return;
    }
    Object[] parameters = params.clone();
    if (indexBoolean != -1) {
        parameters[indexBoolean] = Boolean.valueOf(last);
    }
    if (indexSession != -1) {
        parameters[indexSession] = session;
    }
    if (convert) {
        parameters[indexPayload] = ((ByteBuffer) message).array();
    } else {
        parameters[indexPayload] = message;
    }
    Object result = null;
    try {
        result = method.invoke(pojo, parameters);
    } catch (IllegalAccessException | InvocationTargetException e) {
        handlePojoMethodException(e);
    }
    processResult(result);
}
 

/**
 * {@inheritDoc}
 *
 * Overridden to make it visible to other classes in this package.
 */
@Override
protected void registerSession(Endpoint endpoint, WsSession wsSession) {
    super.registerSession(endpoint, wsSession);
    if (wsSession.isOpen() &&
            wsSession.getUserPrincipal() != null &&
            wsSession.getHttpSessionId() != null) {
        registerAuthenticatedSession(wsSession,
                wsSession.getHttpSessionId());
    }
}
 

/**
 * {@inheritDoc}
 *
 * Overridden to make it visible to other classes in this package.
 */
@Override
protected void unregisterSession(Endpoint endpoint, WsSession wsSession) {
    if (wsSession.getUserPrincipal() != null &&
            wsSession.getHttpSessionId() != null) {
        unregisterAuthenticatedSession(wsSession,
                wsSession.getHttpSessionId());
    }
    super.unregisterSession(endpoint, wsSession);
}
 

private void unregisterAuthenticatedSession(WsSession wsSession,
        String httpSessionId) {
    Set<WsSession> 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<WsSession> wsSessions = authenticatedSessions.remove(httpSessionId);

    if (wsSessions != null && !wsSessions.isEmpty()) {
        for (WsSession 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.
            }
        }
    }
}
 

private void unregisterAuthenticatedSession(WsSession wsSession,
        String httpSessionId) {
    Set<WsSession> wsSessions = authenticatedSessions.get(httpSessionId);
    // wsSessions will be null if the HTTP session has ended
    if (wsSessions != null) {
        wsSessions.remove(wsSession);
    }
}
 

private void registerAuthenticatedSession(WsSession wsSession,
        String httpSessionId) {
    Set<WsSession> wsSessions = authenticatedSessions.get(httpSessionId);
    if (wsSessions == null) {
        wsSessions = Collections.newSetFromMap(
                 new ConcurrentHashMap<WsSession,Boolean>());
         authenticatedSessions.putIfAbsent(httpSessionId, wsSessions);
         wsSessions = authenticatedSessions.get(httpSessionId);
    }
    wsSessions.add(wsSession);
}
 

/**
 * {@inheritDoc}
 *
 * Overridden to make it visible to other classes in this package.
 */
@Override
protected void unregisterSession(Object key, WsSession wsSession) {
    if (wsSession.getUserPrincipal() != null &&
            wsSession.getHttpSessionId() != null) {
        unregisterAuthenticatedSession(wsSession,
                wsSession.getHttpSessionId());
    }
    super.unregisterSession(key, wsSession);
}
 

/**
 * {@inheritDoc}
 *
 * Overridden to make it visible to other classes in this package.
 */
@Override
protected void registerSession(Object key, WsSession wsSession) {
    super.registerSession(key, wsSession);
    if (wsSession.isOpen() &&
            wsSession.getUserPrincipal() != null &&
            wsSession.getHttpSessionId() != null) {
        registerAuthenticatedSession(wsSession,
                wsSession.getHttpSessionId());
    }
}
 

public void closeAuthenticatedSession(String httpSessionId) {
    Set<WsSession> wsSessions = authenticatedSessions.remove(httpSessionId);

    if (wsSessions != null && !wsSessions.isEmpty()) {
        for (WsSession 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.
            }
        }
    }
}
 

@Override
protected void resumeReceiving() {
	if (SUSPENDED.compareAndSet(this, 1, 0)) {
		((WsSession) getDelegate()).resume();
	}
}
 

public WsFrameServer(AbstractServletInputStream sis, WsSession wsSession,
        Transformation transformation) {
    super(wsSession, transformation);
    this.sis = sis;
}
 

public WsFrameServer(AbstractServletInputStream sis, WsSession wsSession,
        Transformation transformation) {
    super(wsSession, transformation);
    this.sis = sis;
}
 

@Override
protected void suspendReceiving() {
	if (SUSPENDED.compareAndSet(this, 0, 1)) {
		((WsSession) getDelegate()).suspend();
	}
}
 

@Override
protected void resumeReceiving() {
	if (SUSPENDED.compareAndSet(this, 1, 0)) {
		((WsSession) getDelegate()).resume();
	}
}
 

@Override
protected void suspendReceiving() {
	if (SUSPENDED.compareAndSet(this, 0, 1)) {
		((WsSession) getDelegate()).suspend();
	}
}
 

public WsFrameServer(SocketWrapperBase<?> socketWrapper, WsSession wsSession,
        Transformation transformation, ClassLoader applicationClassLoader) {
    super(wsSession, transformation);
    this.socketWrapper = socketWrapper;
    this.applicationClassLoader = applicationClassLoader;
}
 

@Override
public void init(WebConnection connection) {
    if (ep == null) {
        throw new IllegalStateException(
                sm.getString("wsHttpUpgradeHandler.noPreInit"));
    }

    String httpSessionId = null;
    Object session = handshakeRequest.getHttpSession();
    if (session != null ) {
        httpSessionId = ((HttpSession) session).getId();
    }

    // Need to call onOpen using the web application's class loader
    // Create the frame using the application's class loader so it can pick
    // up application specific config from the ServerContainerImpl
    Thread t = Thread.currentThread();
    ClassLoader cl = t.getContextClassLoader();
    t.setContextClassLoader(applicationClassLoader);
    try {
        wsRemoteEndpointServer = new WsRemoteEndpointImplServer(socketWrapper, webSocketContainer);
        wsSession = new WsSession(ep, wsRemoteEndpointServer,
                webSocketContainer, handshakeRequest.getRequestURI(),
                handshakeRequest.getParameterMap(),
                handshakeRequest.getQueryString(),
                handshakeRequest.getUserPrincipal(), httpSessionId,
                negotiatedExtensions, subProtocol, pathParameters, secure,
                serverEndpointConfig);
        wsFrame = new WsFrameServer(socketWrapper, wsSession, transformation,
                applicationClassLoader);
        // WsFrame adds the necessary final transformations. Copy the
        // completed transformation chain to the remote end point.
        wsRemoteEndpointServer.setTransformation(wsFrame.getTransformation());
        ep.onOpen(wsSession, serverEndpointConfig);
        webSocketContainer.registerSession(serverEndpointConfig.getPath(), wsSession);
    } catch (DeploymentException e) {
        throw new IllegalArgumentException(e);
    } finally {
        t.setContextClassLoader(cl);
    }
}
 

@Override
public final void onMessage(T message) {

    if (params.length == 1 && params[0] instanceof DecodeException) {
        ((WsSession) session).getLocal().onError(session,
                (DecodeException) params[0]);
        return;
    }

    // Can this message be decoded?
    Object payload;
    try {
        payload = decode(message);
    } catch (DecodeException de) {
        ((WsSession) session).getLocal().onError(session, de);
        return;
    }

    if (payload == null) {
        // Not decoded. Convert if required.
        if (convert) {
            payload = convert(message);
        } else {
            payload = message;
        }
    }

    Object[] parameters = params.clone();
    if (indexSession != -1) {
        parameters[indexSession] = session;
    }
    parameters[indexPayload] = payload;

    Object result = null;
    try {
        result = method.invoke(pojo, parameters);
    } catch (IllegalAccessException | InvocationTargetException e) {
        handlePojoMethodException(e);
    }
    processResult(result);
}