Examples with AbortedMessage org.apache.wicket.protocol.ws.api.message.AbortedMessage used on opensource projects

Example 1 with AbortedMessage

use of org.apache.wicket.protocol.ws.api.message.AbortedMessage in project wicket by apache.

the class AbstractWebSocketProcessor method onConnect.

/**
 * A helper that registers the opened connection in the application-level registry.
 *
 * @param connection
 *            the web socket connection to use to communicate with the client
 * @see #onOpen(Object)
 */
protected final void onConnect(final IWebSocketConnection connection) {
    IKey key = getRegistryKey();
    connectionRegistry.setConnection(getApplication(), getSessionId(), key, connection);
    if (connectionFilter != null) {
        ConnectionRejected connectionRejected = connectionFilter.doFilter(servletRequest);
        if (connectionRejected != null) {
            broadcastMessage(new AbortedMessage(getApplication(), getSessionId(), key));
            connectionRegistry.removeConnection(getApplication(), getSessionId(), key);
            connection.close(connectionRejected.getCode(), connectionRejected.getReason());
            return;
        }
    }
    broadcastMessage(new ConnectedMessage(getApplication(), getSessionId(), key));
}

Example 2 with AbortedMessage

use of org.apache.wicket.protocol.ws.api.message.AbortedMessage in project wicket by apache.

the class AbstractWebSocketProcessor method broadcastMessage.

/**
 * Exports the Wicket thread locals and broadcasts the received message from the client to all
 * interested components and behaviors in the page with id {@code #pageId}
 * <p>
 *     Note: ConnectedMessage and ClosedMessage messages are notification-only. I.e. whatever the
 *     components/behaviors write in the WebSocketRequestHandler will be ignored because the protocol
 *     doesn't expect response from the user.
 * </p>
 *
 * @param message
 *      the message to broadcast
 */
public final void broadcastMessage(final IWebSocketMessage message) {
    IKey key = getRegistryKey();
    IWebSocketConnection connection = connectionRegistry.getConnection(application, sessionId, key);
    if (connection != null && (connection.isOpen() || message instanceof ClosedMessage)) {
        Application oldApplication = ThreadContext.getApplication();
        Session oldSession = ThreadContext.getSession();
        RequestCycle oldRequestCycle = ThreadContext.getRequestCycle();
        WebResponse webResponse = webSocketSettings.newWebSocketResponse(connection);
        try {
            WebSocketRequestMapper requestMapper = new WebSocketRequestMapper(application.getRootRequestMapper());
            RequestCycle requestCycle = createRequestCycle(requestMapper, webResponse);
            ThreadContext.setRequestCycle(requestCycle);
            ThreadContext.setApplication(application);
            Session session;
            if (oldSession == null || message instanceof IWebSocketPushMessage) {
                ISessionStore sessionStore = application.getSessionStore();
                session = sessionStore.lookup(webRequest);
                ThreadContext.setSession(session);
            } else {
                session = oldSession;
            }
            IPageManager pageManager = session.getPageManager();
            Page page = getPage(pageManager);
            if (page != null) {
                WebSocketRequestHandler requestHandler = webSocketSettings.newWebSocketRequestHandler(page, connection);
                WebSocketPayload payload = createEventPayload(message, requestHandler);
                if (!(message instanceof ConnectedMessage || message instanceof ClosedMessage || message instanceof AbortedMessage)) {
                    requestCycle.scheduleRequestHandlerAfterCurrent(requestHandler);
                }
                IRequestHandler broadcastingHandler = new WebSocketMessageBroadcastHandler(pageId, resourceName, payload);
                requestMapper.setHandler(broadcastingHandler);
                requestCycle.processRequestAndDetach();
            } else {
                LOG.debug("Page with id '{}' has been expired. No message will be broadcast!", pageId);
            }
        } catch (Exception x) {
            LOG.error("An error occurred during processing of a WebSocket message", x);
        } finally {
            try {
                webResponse.close();
            } finally {
                ThreadContext.setApplication(oldApplication);
                ThreadContext.setRequestCycle(oldRequestCycle);
                ThreadContext.setSession(oldSession);
            }
        }
    } else {
        LOG.debug("Either there is no connection({}) or it is closed.", connection);
    }
}

Example 3 with AbortedMessage

use of org.apache.wicket.protocol.ws.api.message.AbortedMessage in project wicket by apache.

the class WebSocketResource method onPayload.

void onPayload(WebSocketPayload<?> payload) {
    WebSocketRequestHandler webSocketHandler = payload.getHandler();
    if (payload instanceof WebSocketTextPayload) {
        WebSocketTextPayload textPayload = (WebSocketTextPayload) payload;
        TextMessage data = textPayload.getMessage();
        onMessage(webSocketHandler, data);
    } else if (payload instanceof WebSocketBinaryPayload) {
        WebSocketBinaryPayload binaryPayload = (WebSocketBinaryPayload) payload;
        BinaryMessage binaryData = binaryPayload.getMessage();
        onMessage(webSocketHandler, binaryData);
    } else if (payload instanceof WebSocketConnectedPayload) {
        WebSocketConnectedPayload connectedPayload = (WebSocketConnectedPayload) payload;
        ConnectedMessage message = connectedPayload.getMessage();
        onConnect(message);
    } else if (payload instanceof WebSocketClosedPayload) {
        WebSocketClosedPayload connectedPayload = (WebSocketClosedPayload) payload;
        ClosedMessage message = connectedPayload.getMessage();
        onClose(message);
    } else if (payload instanceof WebSocketErrorPayload) {
        WebSocketErrorPayload errorPayload = (WebSocketErrorPayload) payload;
        ErrorMessage message = errorPayload.getMessage();
        onError(webSocketHandler, message);
    } else if (payload instanceof WebSocketAbortedPayload) {
        WebSocketAbortedPayload abortedPayload = (WebSocketAbortedPayload) payload;
        AbortedMessage message = abortedPayload.getMessage();
        onAbort(message);
    } else if (payload instanceof WebSocketPushPayload) {
        WebSocketPushPayload pushPayload = (WebSocketPushPayload) payload;
        IWebSocketPushMessage message = pushPayload.getMessage();
        onPush(webSocketHandler, message);
    }
}

Example 4 with AbortedMessage

use of org.apache.wicket.protocol.ws.api.message.AbortedMessage in project wicket by apache.

the class WebSocketBehavior method onEvent.

@Override
public void onEvent(Component component, IEvent<?> event) {
    super.onEvent(component, event);
    Object payload = event.getPayload();
    if (payload instanceof WebSocketPayload) {
        WebSocketPayload<?> wsPayload = (WebSocketPayload<?>) payload;
        WebSocketRequestHandler webSocketHandler = wsPayload.getHandler();
        if (payload instanceof WebSocketTextPayload) {
            WebSocketTextPayload textPayload = (WebSocketTextPayload) payload;
            TextMessage data = textPayload.getMessage();
            onMessage(webSocketHandler, data);
        } else if (wsPayload instanceof WebSocketBinaryPayload) {
            WebSocketBinaryPayload binaryPayload = (WebSocketBinaryPayload) wsPayload;
            BinaryMessage binaryData = binaryPayload.getMessage();
            onMessage(webSocketHandler, binaryData);
        } else if (wsPayload instanceof WebSocketConnectedPayload) {
            WebSocketConnectedPayload connectedPayload = (WebSocketConnectedPayload) wsPayload;
            ConnectedMessage message = connectedPayload.getMessage();
            onConnect(message);
        } else if (wsPayload instanceof WebSocketClosedPayload) {
            WebSocketClosedPayload closedPayload = (WebSocketClosedPayload) wsPayload;
            ClosedMessage message = closedPayload.getMessage();
            onClose(message);
        } else if (wsPayload instanceof WebSocketErrorPayload) {
            WebSocketErrorPayload errorPayload = (WebSocketErrorPayload) wsPayload;
            ErrorMessage message = errorPayload.getMessage();
            onError(webSocketHandler, message);
        } else if (wsPayload instanceof WebSocketAbortedPayload) {
            WebSocketAbortedPayload abortedPayload = (WebSocketAbortedPayload) wsPayload;
            AbortedMessage message = abortedPayload.getMessage();
            onAbort(message);
        } else if (wsPayload instanceof WebSocketPushPayload) {
            WebSocketPushPayload pushPayload = (WebSocketPushPayload) wsPayload;
            IWebSocketPushMessage message = pushPayload.getMessage();
            onPush(webSocketHandler, message);
        }
    }
}