Search in sources :

Example 1 with WebSocketContext

use of act.ws.WebSocketContext in project actframework by actframework.

the class UndertowWebSocketConnectionHandler method handle.

@Override
public void handle(final ActionContext context) {
    if (logger.isTraceEnabled()) {
        logger.trace("handle websocket connection request to %s", context.req().url());
    }
    final UndertowRequest req = (UndertowRequest) context.req();
    HttpServerExchange exchange = req.exchange();
    try {
        Handlers.websocket(new WebSocketConnectionCallback() {

            @Override
            public void onConnect(WebSocketHttpExchange exchange, WebSocketChannel channel) {
                final WebSocketConnection connection = new UndertowWebSocketConnection(channel, context.session());
                channel.setAttribute("act_conn", connection);
                connectionManager.registerNewConnection(connection, context);
                final WebSocketContext wsCtx = new WebSocketContext(req.url(), connection, connectionManager, context, connectionManager.app());
                if (logger.isTraceEnabled()) {
                    logger.trace("websocket context[%s] created for %s", connection.sessionId(), context.req().url());
                }
                channel.getReceiveSetter().set(new AbstractReceiveListener() {

                    @Override
                    protected void onFullTextMessage(WebSocketChannel channel, BufferedTextMessage message) throws IOException {
                        WebSocketContext.current(wsCtx);
                        String payload = message.getData();
                        if (logger.isTraceEnabled()) {
                            logger.trace("websocket message received: %s", payload);
                        }
                        wsCtx.messageReceived(payload);
                        invoke(wsCtx);
                    }

                    @Override
                    protected void onClose(WebSocketChannel webSocketChannel, StreamSourceFrameChannel channel) throws IOException {
                        if (logger.isTraceEnabled()) {
                            logger.trace("websocket closed: ", connection.sessionId());
                        }
                        WebSocketContext.current(wsCtx);
                        super.onClose(webSocketChannel, channel);
                        connection.destroy();
                        context.app().eventBus().emit(new WebSocketCloseEvent(wsCtx));
                    }
                });
                channel.resumeReceives();
                Act.eventBus().emit(new WebSocketConnectEvent(wsCtx));
            }
        }).handleRequest(exchange);
    } catch (RuntimeException e) {
        throw e;
    } catch (Exception e) {
        throw ActErrorResult.of(e);
    }
}
Also used : WebSocketCloseEvent(act.ws.WebSocketCloseEvent) WebSocketConnectEvent(act.ws.WebSocketConnectEvent) WebSocketConnection(act.xio.WebSocketConnection) WebSocketChannel(io.undertow.websockets.core.WebSocketChannel) IOException(java.io.IOException) BufferedTextMessage(io.undertow.websockets.core.BufferedTextMessage) IOException(java.io.IOException) WebSocketContext(act.ws.WebSocketContext) HttpServerExchange(io.undertow.server.HttpServerExchange) WebSocketHttpExchange(io.undertow.websockets.spi.WebSocketHttpExchange) StreamSourceFrameChannel(io.undertow.websockets.core.StreamSourceFrameChannel) AbstractReceiveListener(io.undertow.websockets.core.AbstractReceiveListener) WebSocketConnectionCallback(io.undertow.websockets.WebSocketConnectionCallback)

Aggregations

WebSocketCloseEvent (act.ws.WebSocketCloseEvent)1 WebSocketConnectEvent (act.ws.WebSocketConnectEvent)1 WebSocketContext (act.ws.WebSocketContext)1 WebSocketConnection (act.xio.WebSocketConnection)1 HttpServerExchange (io.undertow.server.HttpServerExchange)1 WebSocketConnectionCallback (io.undertow.websockets.WebSocketConnectionCallback)1 AbstractReceiveListener (io.undertow.websockets.core.AbstractReceiveListener)1 BufferedTextMessage (io.undertow.websockets.core.BufferedTextMessage)1 StreamSourceFrameChannel (io.undertow.websockets.core.StreamSourceFrameChannel)1 WebSocketChannel (io.undertow.websockets.core.WebSocketChannel)1 WebSocketHttpExchange (io.undertow.websockets.spi.WebSocketHttpExchange)1 IOException (java.io.IOException)1