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 {

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

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

            int port = Integer.parseInt(portNo);

            handler = new WarehouseServiceHandler();

            processor = new Processor<Iface>(handler);

	        log.info("DB Connection String is: " + dbHost);

	        log.info("URL read by data source before setting is: " + handler.getDataSourceUrl());

	        handler.setDataSourceUrl(dbHost);

	        log.info("URL read by data source after setting is: " + handler.getDataSourceUrl());

            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.");

    }

    public void init(DaemonContext dc) throws Exception {

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

    }

    public void start() throws Exception {

    	log.info("Starting Warehouse Server");

        initialize();

    }

    public void stop() throws Exception {

        log.info("Stopping Warehouse Server...");

        server.stop();

    }

}