Subversion Repositories SmartDukaan

/**

 * 

 */

package in.shop2020.warehouse.service;

import in.shop2020.thrift.clients.config.ConfigClient;

import in.shop2020.utils.ConfigClientKeys;

import in.shop2020.warehouse.WarehouseService.Iface;

import in.shop2020.warehouse.WarehouseService.Processor;

import in.shop2020.warehouse.service.handler.WarehouseServiceHandler;

import org.apache.commons.daemon.Daemon;

import org.apache.commons.daemon.DaemonContext;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.apache.thrift.protocol.TBinaryProtocol;

import org.apache.thrift.protocol.TProtocolFactory;

import org.apache.thrift.server.TServer;

import org.apache.thrift.server.TThreadPoolServer;

import org.apache.thrift.server.TThreadPoolServer.Args;

import org.apache.thrift.transport.TFramedTransport;

import org.apache.thrift.transport.TServerSocket;

import org.apache.thrift.transport.TServerTransport;

import org.apache.thrift.transport.TTransportFactory;

/**

 * @author mandeep

 *

 */

public class WarehouseServer implements Daemon{

    private static Log log = LogFactory.getLog(WarehouseServer.class);

    private static WarehouseServiceHandler handler;

    private static Processor<Iface> processor;

    private static String dbHost;

    private static TServer server;

    public static void main(String[] args){

        initialize();

    }

    private static void initialize() {

        try {

            int port = 9013;

            try {

                dbHost = "jdbc:mysql://" + ConfigClient.getClient().get(ConfigClientKeys.warehouse_service_db_hostname.toString()) + "/warehouse";

                String portNo = ConfigClient.getClient().get(ConfigClientKeys.warehouse_service_server_port.toString());

                port = Integer.parseInt(portNo);

            } catch(Exception e) {

                log.warn("Unable to get port number from the Config server because of:", e);

            }

            // Setting dbHost env property to make this value accessible in context.xml 

            // while spring context creation.

            System.setProperty("dbHost", dbHost);

            handler = new WarehouseServiceHandler();

            processor = new Processor<Iface>(handler);

            TServerTransport serverTransport = new TServerSocket(port);

            TTransportFactory tFactory = new TFramedTransport.Factory();

            TProtocolFactory pFactory = new TBinaryProtocol.Factory();

            Args serverParams = new Args(serverTransport);

            serverParams.processor(processor);

            serverParams.transportFactory(tFactory);

            serverParams.protocolFactory(pFactory);

            server = new TThreadPoolServer(serverParams);

            log.info("Warehouse service started on port " + port);

            server.serve();

        }catch(Exception ex){

            ex.printStackTrace();

        }

    }

    public void destroy() {

        System.out.println("Warehouse Server stopped.");

        // TODO Auto-generated method stub

    }

    public void init(DaemonContext dc) throws Exception {

        System.out.println("Initializing Warehouse Server...");

        // TODO Auto-generated method stub

    }

    public void start() throws Exception {

        System.out.println("Starting Warehouse Server");

        initialize();

        // TODO Auto-generated method stub

    }

    public void stop() throws Exception {

        System.out.println("Stopping Warehouse Server...");

        server.stop();

        System.exit(0);

        // TODO Auto-generated method stub

    }

}