com.senseidb.util.SingleNodeStarter.java Source code

Introduction

Here is the source code for com.senseidb.util.SingleNodeStarter.java

Source

/**
 * This software is licensed to you under the Apache License, Version 2.0 (the
 * "Apache License").
 *
 * LinkedIn's contributions are made under the Apache License. If you contribute
 * to the Software, the contributions will be deemed to have been made under the
 * Apache License, unless you expressly indicate otherwise. Please do not make any
 * contributions that would be inconsistent with the Apache License.
 *
 * You may obtain a copy of the Apache License at http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, this software
 * distributed under the Apache License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the Apache
 * License for the specific language governing permissions and limitations for the
 * software governed under the Apache License.
 *
 *  2012 LinkedIn Corp. All Rights Reserved.
 */
package com.senseidb.util;

import java.io.File;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.HashMap;

import com.linkedin.norbert.network.Serializer;
import com.senseidb.svc.impl.CoreSenseiServiceImpl;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.eclipse.jetty.server.Server;

import com.linkedin.norbert.javacompat.network.PartitionedLoadBalancerFactory;
import com.senseidb.cluster.routing.SenseiPartitionedLoadBalancerFactory;
import com.senseidb.conf.SenseiConfParams;
import com.senseidb.conf.SenseiServerBuilder;
import com.senseidb.search.node.SenseiBroker;
import com.senseidb.search.node.SenseiServer;
import com.senseidb.search.node.broker.BrokerConfig;
import com.senseidb.search.req.SenseiRequest;
import com.senseidb.search.req.SenseiResult;

public class SingleNodeStarter {
    private static boolean serverStarted = false;
    private static Server jettyServer;
    private static SenseiServer server;

    public static void start(String localPath, int expectedDocs) {
        start(new File(getUri(localPath)), expectedDocs);
    }

    public static void start(File confDir, int expectedDocs) {
        if (!serverStarted) {
            try {
                PropertiesConfiguration senseiConfiguration = new PropertiesConfiguration(
                        new File(confDir, "sensei.properties"));
                final String indexDir = senseiConfiguration.getString(SenseiConfParams.SENSEI_INDEX_DIR);
                // rmrf(new File(indexDir));
                SenseiServerBuilder senseiServerBuilder = new SenseiServerBuilder(confDir, null);
                server = senseiServerBuilder.buildServer();
                jettyServer = senseiServerBuilder.buildHttpRestServer();
                server.start(true);
                jettyServer.start();
                Runtime.getRuntime().addShutdownHook(new Thread() {
                    @Override
                    public void run() {
                        shutdown();
                    }
                });
                PartitionedLoadBalancerFactory balancerFactory = new SenseiPartitionedLoadBalancerFactory(1000);
                Serializer<SenseiRequest, SenseiResult> serializer = CoreSenseiServiceImpl.JAVA_SERIALIZER;
                BrokerConfig brokerConfig = new BrokerConfig(senseiConfiguration, balancerFactory, serializer,
                        senseiServerBuilder.getPluginRegistry());
                brokerConfig.init();
                SenseiBroker senseiBroker = brokerConfig.buildSenseiBroker();
                while (true) {
                    SenseiResult senseiResult = senseiBroker.browse(new SenseiRequest());
                    int totalDocs = senseiResult.getTotalDocs();
                    System.out.println("TotalDocs = " + totalDocs);
                    if (totalDocs >= expectedDocs) {
                        break;
                    }
                    Thread.sleep(100);
                }
            } catch (Exception ex) {
                throw new RuntimeException(ex);
            }
        }
    }

    public static boolean rmrf(File f) {
        if (f == null || !f.exists()) {
            return true;
        }
        if (f.isDirectory()) {
            for (File sub : f.listFiles()) {
                if (!rmrf(sub))
                    return false;
            }
        }
        return f.delete();
    }

    public static boolean isServerStarted() {
        return serverStarted;
    }

    private static URI getUri(String localPath) {
        try {
            return SingleNodeStarter.class.getClassLoader().getResource(localPath).toURI();
        } catch (URISyntaxException ex) {
            throw new RuntimeException(ex);
        }
    }

    public static void shutdown() {
        try {
            jettyServer.stop();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            server.shutdown();
            serverStarted = false;
        }
    }
}