org.janusgraph.diskstorage.solr.SolrRunner.java Source code

Introduction

Here is the source code for org.janusgraph.diskstorage.solr.SolrRunner.java

Source

// Copyright 2017 JanusGraph Authors
//
// 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 org.janusgraph.diskstorage.solr;

import com.google.common.base.Joiner;
import org.apache.commons.io.FileUtils;
import org.apache.solr.cloud.MiniSolrCloudCluster;
import org.apache.solr.cloud.ZkController;
import org.apache.solr.servlet.SolrDispatchFilter;

import java.io.File;

public class SolrRunner {

    protected static final int NUM_SERVERS = 1;
    protected static final String[] COLLECTIONS = new String[] { "store1", "store2", "vertex", "edge", "namev",
            "namee", "composite", "psearch", "esearch", "vsearch", "mi", "mixed", "index1", "index2", "index3",
            "ecategory", "vcategory", "pcategory", "theIndex", "vertices", "edges", "booleanIndex", "dateIndex",
            "instantIndex", "uuidIndex" };

    protected static final String[] KEY_FIELDS = new String[0];

    private static final String TMP_DIRECTORY = System.getProperty("java.io.tmpdir");
    private static final String TEMPLATE_DIRECTORY = "core-template";

    private static MiniSolrCloudCluster miniSolrCloudCluster;

    public static void start() throws Exception {
        String userDir = System.getProperty("user.dir");
        String solrHome = userDir.contains("janusgraph-solr")
                ? Joiner.on(File.separator).join(userDir, "target", "test-classes", "solr")
                : Joiner.on(File.separator).join(userDir, "janusgraph-solr", "target", "test-classes", "solr");

        File templateDirectory = new File(solrHome + File.separator + TEMPLATE_DIRECTORY);
        assert templateDirectory.exists();

        File temp = new File(TMP_DIRECTORY + File.separator + "solr-" + System.nanoTime());
        assert !temp.exists();

        temp.mkdirs();
        temp.deleteOnExit();

        File solrXml = new File(solrHome, "solr.xml");
        miniSolrCloudCluster = new MiniSolrCloudCluster(NUM_SERVERS, null, temp, solrXml, null, null);

        for (String core : COLLECTIONS) {
            File coreDirectory = new File(temp.getAbsolutePath() + File.separator + core);
            assert coreDirectory.mkdirs();
            FileUtils.copyDirectory(templateDirectory, coreDirectory);
            miniSolrCloudCluster.uploadConfigDir(coreDirectory.getAbsoluteFile(), core);
        }
    }

    public static MiniSolrCloudCluster getMiniCluster() {
        return miniSolrCloudCluster;
    }

    public static void stop() throws Exception {
        System.clearProperty("solr.solrxml.location");
        System.clearProperty("zkHost");
        miniSolrCloudCluster.shutdown();
    }

    private static ZkController getZkController() {
        SolrDispatchFilter dispatchFilter = (SolrDispatchFilter) miniSolrCloudCluster.getJettySolrRunners().get(0)
                .getDispatchFilter().getFilter();
        return dispatchFilter.getCores().getZkController();
    }

}