Examples with WebSocketPayload org.apache.wicket.protocol.ws.api.event.WebSocketPayload used on opensource projects

Example 1 with WebSocketPayload

use of org.apache.wicket.protocol.ws.api.event.WebSocketPayload 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 2 with WebSocketPayload

use of org.apache.wicket.protocol.ws.api.event.WebSocketPayload 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);
        }
    }
}