io.liveoak.container.protocols.ProtocolDetector.java Source code

Introduction

Here is the source code for io.liveoak.container.protocols.ProtocolDetector.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;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.ReplayingDecoder;

import java.nio.charset.Charset;
import java.util.List;

/**
 * @author Bob McWhirter
 */
public class ProtocolDetector extends ReplayingDecoder<Void> {

    private static final Charset UTF_8 = Charset.forName("UTF-8");

    public ProtocolDetector(PipelineConfigurator configurator) {
        this.configurator = configurator;
    }

    @Override
    protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception {

        int nonNewlineBytes = in.bytesBefore((byte) '\n');

        in.markReaderIndex();

        if (nonNewlineBytes > 0) {
            ByteBuf lineBuffer = in.readBytes(nonNewlineBytes);
            String line = lineBuffer.toString(UTF_8);

            //SslHandler sslHandler = context.getPipeline().writeState( SslHandler.class );

            in.resetReaderIndex();
            ByteBuf fullBuffer = in.readBytes(super.actualReadableBytes());

            if (line.startsWith("CONNECT") || line.startsWith("STOMP")) {
                this.configurator.switchToPureStomp(ctx.pipeline());
            } else {
                this.configurator.switchToHttpWebSockets(ctx.pipeline());
            }

            ctx.pipeline().fireChannelRead(fullBuffer);
        }
    }

    private PipelineConfigurator configurator;

}