ezbake.IntentQuery.Sample.MongoDatasource.Server.MongoExternalDataSourceServer.java Source code

Introduction

Here is the source code for ezbake.IntentQuery.Sample.MongoDatasource.Server.MongoExternalDataSourceServer.java

Source

/*   Copyright (C) 2013-2014 Computer Sciences Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License. */

package ezbake.IntentQuery.Sample.MongoDatasource.Server;

import com.cloudera.impala.extdatasource.thrift.ExternalDataSourceService;

import org.apache.thrift.server.TServer;
//import org.apache.thrift.server.TServer.Args;
//import org.apache.thrift.server.TSimpleServer;
import org.apache.thrift.server.TThreadPoolServer;
import org.apache.thrift.transport.TSSLTransportFactory;
import org.apache.thrift.transport.TServerSocket;
import org.apache.thrift.transport.TServerTransport;
import org.apache.thrift.transport.TSSLTransportFactory.TSSLTransportParameters;

/**
 * Mock of an application which use MongoDB as its storage,
 * It provides Thrift interface to Impala to access its exported table
 *
 */
public class MongoExternalDataSourceServer {
    public static int SIMPLE_PORT = 9003;
    public static int SECURE_PORT = 9091;

    public static MongoExternalDataSourceHandler handler;

    public static ExternalDataSourceService.Processor processor;

    public static void main(String[] args) {
        try {
            handler = new MongoExternalDataSourceHandler();
            processor = new ExternalDataSourceService.Processor(handler);

            Runnable simple = new Runnable() {
                public void run() {
                    simple(processor);
                }
            };
            Runnable secure = new Runnable() {
                public void run() {
                    secure(processor);
                }
            };

            new Thread(simple).start();

            //new Thread(secure).start();
        } catch (Exception x) {
            x.printStackTrace();
        }

    }

    public static void simple(ExternalDataSourceService.Processor processor) {
        try {
            TServerTransport serverTransport = new TServerSocket(SIMPLE_PORT);
            //TServer server = new TSimpleServer(new Args(serverTransport).processor(processor));

            // Use this for a multithreaded server
            TServer server = new TThreadPoolServer(
                    new TThreadPoolServer.Args(serverTransport).processor(processor));

            System.out.println("Starting the simple server...");
            server.serve();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void secure(ExternalDataSourceService.Processor processor) {
        try {
            /*
             * Use TSSLTransportParameters to setup the required SSL parameters. In this example
             * we are setting the keystore and the keystore password. Other things like algorithms,
             * cipher suites, client auth etc can be set. 
             */
            TSSLTransportParameters params = new TSSLTransportParameters();
            // The Keystore contains the private key
            params.setKeyStore("/home/training/ezBake/thrift-0.9.1/lib/java/test/.keystore", "thrift", null, null);

            /*
             * Use any of the TSSLTransportFactory to get a server transport with the appropriate
             * SSL configuration. You can use the default settings if properties are set in the command line.
             * Ex: -Djavax.net.ssl.keyStore=.keystore and -Djavax.net.ssl.keyStorePassword=thrift
             * 
             * Note: You need not explicitly call open(). The underlying server socket is bound on return
             * from the factory class. 
             */
            TServerTransport serverTransport = TSSLTransportFactory.getServerSocket(SECURE_PORT, 0, null, params);
            //TServer server = new TSimpleServer(new Args(serverTransport).processor(processor));

            // Use this for a multi threaded server
            TServer server = new TThreadPoolServer(
                    new TThreadPoolServer.Args(serverTransport).processor(processor));

            System.out.println("Starting the secure server...");
            server.serve();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}