Examples with AbstractReceiveListener io.undertow.websockets.core.AbstractReceiveListener used on opensource projects

Example 21 with AbstractReceiveListener

use of io.undertow.websockets.core.AbstractReceiveListener 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);
    }
}