ThriftServer.java

package com.breakersoft.plow.thrift;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;

import org.apache.thrift.TProcessor;
import org.apache.thrift.protocol.TProtocolFactory;
import org.apache.thrift.server.TServer;
import org.apache.thrift.server.TThreadedSelectorServer;
import org.apache.thrift.transport.TFramedTransport;
import org.apache.thrift.transport.TNonblockingServerSocket;
import org.apache.thrift.transport.TTransportException;
import org.slf4j.Logger;

public class ThriftServer {

    private Logger logger = org.slf4j.LoggerFactory.getLogger(ThriftServer.class);

    private final int port;
    private int threads;
    private final String name;
    private final TProcessor processor;
    private final Thread thread;
    private final TProtocolFactory protocolFactory;

    private TNonblockingServerSocket transport;
    private TServer server;

    public ThriftServer(TProcessor processor, TProtocolFactory protocolFactory, int threads, int port) {
        this.port = port;
        this.name = processor.toString();
        this.processor = processor;
        this.protocolFactory = protocolFactory;
        this.threads = threads;
        thread = new ServerThread();
    }

    @PostConstruct
    public void start() {
        logger.info("Starting thift server " + name + " on port " + port);

        try {
            transport = new TNonblockingServerSocket(port);
            server = new TThreadedSelectorServer(
                    new TThreadedSelectorServer.Args(transport)
                .processor(processor)
                .workerThreads(threads)
                .selectorThreads((threads / 16) + 1)
                .protocolFactory(protocolFactory)
                .transportFactory(new TFramedTransport.Factory()));
            thread.start();

        } catch (TTransportException e) {
            throw new RuntimeException("Unable to start thrift server " + e, e);
        }
    }

    @PreDestroy
    public void stop() {
        logger.info("Stopping thift server " + name + " on port " + port);
        server.stop();
    }

    private class ServerThread extends Thread {

        public void run() {
            server.serve();
        }
    }
}