io.liveoak.stomp.server.SimpleStompServer.java Source code

Introduction

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

import io.liveoak.stomp.common.StompFrameDecoder;
import io.liveoak.stomp.common.StompFrameEncoder;
import io.liveoak.stomp.common.StompMessageDecoder;
import io.liveoak.stomp.common.StompMessageEncoder;
import io.liveoak.stomp.server.protocol.ConnectHandler;
import io.liveoak.stomp.server.protocol.DisconnectHandler;
import io.liveoak.stomp.server.protocol.ReceiptHandler;
import io.liveoak.stomp.server.protocol.SendHandler;
import io.liveoak.stomp.server.protocol.StompErrorHandler;
import io.liveoak.stomp.server.protocol.SubscribeHandler;
import io.liveoak.stomp.server.protocol.UnsubscribeHandler;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.util.concurrent.Future;

/**
 * @author Bob McWhirter
 */
public class SimpleStompServer {

    public SimpleStompServer(String host, int port, StompServerContext serverContext) {
        this.host = host;
        this.port = port;
        this.serverContext = serverContext;
        this.group = new NioEventLoopGroup();
    }

    public void start() throws InterruptedException {
        ServerBootstrap serverBootstrap = new ServerBootstrap();
        serverBootstrap.channel(NioServerSocketChannel.class).group(this.group).localAddress(this.host, this.port)
                .childHandler(createChildHandler());
        ChannelFuture future = serverBootstrap.bind();
        future.sync();
    }

    protected ChannelInitializer<NioSocketChannel> createChildHandler() {
        return new ChannelInitializer<NioSocketChannel>() {
            @Override
            protected void initChannel(NioSocketChannel ch) throws Exception {
                // bytes into frames
                ch.pipeline().addLast(new StompFrameDecoder());
                ch.pipeline().addLast(new StompFrameEncoder());
                //ch.pipeline().addLast( new DebugHandler( "server-head" ) );
                // handle frames
                ch.pipeline().addLast(new ConnectHandler(SimpleStompServer.this.serverContext));
                ch.pipeline().addLast(new DisconnectHandler(SimpleStompServer.this.serverContext));
                ch.pipeline().addLast(new SubscribeHandler(SimpleStompServer.this.serverContext));
                ch.pipeline().addLast(new UnsubscribeHandler(SimpleStompServer.this.serverContext));
                // convert some frames to messages
                ch.pipeline().addLast(new ReceiptHandler());
                ch.pipeline().addLast(new StompMessageDecoder());
                ch.pipeline().addLast(new StompMessageEncoder(true));
                // handle messages
                ch.pipeline().addLast(new SendHandler(SimpleStompServer.this.serverContext));
                // catch errors, return an ERROR message.
                ch.pipeline().addLast(new StompErrorHandler());
            }
        };
    }

    public void stop() throws InterruptedException {
        Future<?> future = this.group.shutdownGracefully();
        future.sync();
    }

    private final String host;
    private final int port;
    private StompServerContext serverContext;
    private EventLoopGroup group;

}