io.liveoak.stomp.common.StompMessageDecoder.java Source code

Introduction

Here is the source code for io.liveoak.stomp.common.StompMessageDecoder.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.stomp.common;

import java.util.List;

import io.liveoak.stomp.Stomp;
import io.liveoak.stomp.StompMessage;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.MessageToMessageDecoder;
import io.netty.util.ReferenceCountUtil;

/**
 * @author Bob McWhirter
 */
public class StompMessageDecoder extends MessageToMessageDecoder<StompFrame> {

    @Override
    protected void decode(ChannelHandlerContext ctx, StompFrame msg, List<Object> out) throws Exception {
        if (msg instanceof StompContentFrame) {
            if (msg.command() == Stomp.Command.MESSAGE || msg.command() == Stomp.Command.SEND
                    || msg.command() == Stomp.Command.ERROR) {
                StompMessage stompMessage = new DefaultStompMessage(msg.headers(),
                        ((StompContentFrame) msg).content().retain(), msg.command() == Stomp.Command.ERROR);
                out.add(stompMessage);
            } else {
                ReferenceCountUtil.retain(msg);
                out.add(msg);
            }
        } else {
            ReferenceCountUtil.retain(msg);
            out.add(msg);
        }

    }
}