ServerChannelAdapter.java

package org.rakam.kume.network;

import org.rakam.kume.transport.PacketDecoder;
import org.rakam.kume.transport.RemoteOperationContext;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import org.rakam.kume.service.Service;
import org.rakam.kume.transport.Packet;
import org.rakam.kume.transport.Request;
import org.rakam.kume.util.ThrowableNioEventLoopGroup;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.List;


public class ServerChannelAdapter extends ChannelInboundHandlerAdapter {
    final static Logger LOGGER = LoggerFactory.getLogger(PacketDecoder.class);

    List<Service> services;

    ThrowableNioEventLoopGroup eventExecutors;

    public ServerChannelAdapter(List<Service> services, ThrowableNioEventLoopGroup executor) {
        this.services = services;
        eventExecutors = executor;
    }

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) {
        LOGGER.trace("server {} got message {}", ctx.channel().localAddress(), msg);

        Packet read = (Packet) msg;
        Object o = read.getData();

        RemoteOperationContext ctx1 = new RemoteOperationContext(ctx, read.service, read.sequence);
        Service service = services.get(read.service);
        if (o instanceof Request) {
            service.handle(eventExecutors, ctx1, (Request) o);
        } else {
            service.handle(eventExecutors, ctx1, o);
        }
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        LOGGER.warn("Dropped unsupported package.", cause);
        ctx.close();
    }
}