io.liveoak.container.protocols.websocket.WebSocketStompFrameDecoder.java Source code

Introduction

Here is the source code for io.liveoak.container.protocols.websocket.WebSocketStompFrameDecoder.java

Source

/*
 * Copyright 2013 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed under the Eclipse Public License version 1.0, available at http://www.eclipse.org/legal/epl-v10.html
 */
package io.liveoak.container.protocols.websocket;

import io.liveoak.stomp.common.StompFrameDecoder;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.MessageToMessageDecoder;
import io.netty.handler.codec.http.websocketx.BinaryWebSocketFrame;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.handler.codec.http.websocketx.WebSocketFrame;

import java.util.List;

/**
 * WebSocket to STOMP frame decoder.
 *
 * <p>This decoder also adds the normal "native" STOMP frame decoder upstream from itself,
 * providing for the pipeline of:</p>
 *
 * <p>
 * WebSocket Frame to bytes to STOMP frame
 * </p>
 *
 * @author Bob McWhirter
 */
public class WebSocketStompFrameDecoder extends MessageToMessageDecoder<WebSocketFrame> {

    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        ctx.pipeline().addAfter(ctx.name(), "stomp-frame-decoder", new StompFrameDecoder());
    }

    @Override
    protected void decode(ChannelHandlerContext ctx, WebSocketFrame msg, List<Object> out) throws Exception {
        if (msg instanceof TextWebSocketFrame || msg instanceof BinaryWebSocketFrame) {
            out.add(msg.content().retain());
        } else {
            out.add(msg.retain());
        }
    }

}