org.codice.solr.factory.impl.SolrClientFactoryImpl.java Source code

Introduction

Here is the source code for org.codice.solr.factory.impl.SolrClientFactoryImpl.java

Source

/**
 * Copyright (c) Codice Foundation
 *
 * <p>This is free software: you can redistribute it and/or modify it under the terms of the GNU
 * Lesser General Public License as published by the Free Software Foundation, either version 3 of
 * the License, or any later version.
 *
 * <p>This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details. A copy of the GNU Lesser General Public
 * License is distributed along with this program and can be found at
 * <http://www.gnu.org/licenses/lgpl.html>.
 */
package org.codice.solr.factory.impl;

import static org.apache.commons.lang.Validate.notNull;

import com.google.common.annotations.VisibleForTesting;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.function.BiFunction;
import org.codice.solr.client.solrj.SolrClient;
import org.codice.solr.factory.SolrClientFactory;

/**
 * Factory class used to create the proper {@link SolrClient} based on the current {@code
 * solr.client} system property.
 */
public final class SolrClientFactoryImpl implements SolrClientFactory {

    private final BiFunction<SolrClientFactory, String, SolrClient> newClientFunction;
    private final HttpSolrClientFactory httpSolrClientFactory;

    @SuppressWarnings("unused" /* used by blueprint */)
    public SolrClientFactoryImpl(HttpSolrClientFactory httpSolrClientFactory) {
        this.newClientFunction = (factory, core) -> factory.newClient(core);
        this.httpSolrClientFactory = httpSolrClientFactory;
    }

    @VisibleForTesting
    SolrClientFactoryImpl(HttpSolrClientFactory httpSolrClientFactory,
            BiFunction<SolrClientFactory, String, SolrClient> newClientFunction) {
        this.newClientFunction = newClientFunction;
        this.httpSolrClientFactory = httpSolrClientFactory;
    }

    @Override
    public SolrClient newClient(String core) {
        notNull(core, "Solr core name cannot be null");

        String clientType = AccessController
                .doPrivileged((PrivilegedAction<String>) () -> System.getProperty("solr.client", "HttpSolrClient"));
        SolrClientFactory factory;

        if ("EmbeddedSolrServer".equals(clientType)) {
            factory = new EmbeddedSolrFactory();
        } else if ("CloudSolrClient".equals(clientType)) {
            factory = new SolrCloudClientFactory();
        } else { // Use HttpSolrClient by default
            factory = httpSolrClientFactory;
        }

        return newClientFunction.apply(factory, core);
    }
}