com.sindicetech.siren.solr.SolrServerWrapper.java Source code

Introduction

Here is the source code for com.sindicetech.siren.solr.SolrServerWrapper.java

Source

/**
 * Copyright (c) 2014, Sindice Limited. All Rights Reserved.
 *
 * This file is part of the SIREn project.
 *
 * 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 com.sindicetech.siren.solr;

import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.SolrRequest;
import org.apache.solr.client.solrj.SolrServer;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.client.solrj.embedded.EmbeddedSolrServer;
import org.apache.solr.client.solrj.request.LukeRequest;
import org.apache.solr.client.solrj.request.UpdateRequest;
import org.apache.solr.client.solrj.response.LukeResponse;
import org.apache.solr.client.solrj.response.LukeResponse.FieldInfo;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.SolrDocumentList;
import org.apache.solr.common.SolrInputDocument;
import org.apache.solr.common.params.ModifiableSolrParams;
import org.apache.solr.common.params.SolrParams;
import org.apache.solr.core.CoreContainer;
import org.apache.solr.core.SolrCore;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.util.Map;
import java.util.Set;

/**
 * Wraps a {@link CoreContainer} and provides utility methods over the
 * Solr index.
 */
public class SolrServerWrapper {

    final CoreContainer coreContainer;
    private final SolrServer server;
    private Map<String, FieldInfo> fieldInfos;

    private static final Logger logger = LoggerFactory.getLogger(SolrServerWrapper.class);

    public SolrServerWrapper(final CoreContainer coreContainer) {
        this.coreContainer = coreContainer;
        server = new EmbeddedSolrServer(coreContainer, "");
    }

    protected SolrCore getCore() {
        return coreContainer.getCore(coreContainer.getDefaultCoreName());
    }

    /**
     * Add a document.
     */
    public void add(final SolrInputDocument doc) throws SolrServerException, IOException {
        final UpdateRequest request = new UpdateRequest();
        request.add(doc);
        request.process(this.getServer());
    }

    /**
     * Return the number of documents currently indexed.
     */
    public int getNumberOfDocuments() throws SolrServerException, IOException {
        final SolrRequest request = new FastLukeRequest();
        final LukeResponse response = (LukeResponse) request.process(this.getServer());
        return response.getNumDocs();
    }

    public String[] search(final SolrQuery query, final String retrievedField)
            throws SolrServerException, IOException {
        final QueryResponse response = this.getServer().query(query);
        final SolrDocumentList docList = response.getResults();

        final int size = docList.size();
        final String docIDs[] = new String[size];
        SolrDocument doc = null;
        for (int i = 0; i < size; i++) {
            doc = docList.get(i);
            docIDs[i] = (String) doc.getFieldValue(retrievedField);
        }
        return docIDs;
    }

    public SolrDocumentList search(final SolrQuery query) throws SolrServerException, IOException {
        final QueryResponse response = this.getServer().query(query);
        return response.getResults();
    }

    public String[] search(final String q, final String retrievedField, final String qt)
            throws SolrServerException, IOException {
        final SolrQuery query = new SolrQuery();
        query.setRequestHandler(qt);
        query.setQuery(q);
        return this.search(query, retrievedField);
    }

    public int getNumberOfSegments() {
        return this.getCore().getSearcher().get().getIndexReader().leaves().size();
    }

    /**
     * Execute a Luke Request and extract the field names.
     */
    public Set<String> getFieldNames() throws SolrServerException, IOException {
        final Map<String, FieldInfo> fieldInfos = this.getFieldInfos();
        return fieldInfos.keySet();
    }

    private Map<String, FieldInfo> getFieldInfos() throws SolrServerException, IOException {
        if (fieldInfos == null) {
            final LukeRequest request = new FastLukeRequest();
            request.setShowSchema(true);
            final LukeResponse response = request.process(this.getServer());
            fieldInfos = response.getFieldInfo();
        }
        return fieldInfos;
    }

    /**
     * Commit all documents that have been submitted
     */
    public void commit() throws SolrServerException, IOException {
        this.getServer().commit();
    }

    /**
     * Execute optimisation of the Solr index
     */
    public void optimize() throws SolrServerException, IOException {
        this.optimize(1);
    }

    /**
     * Execute optimisation of the Solr index
     */
    public void optimize(final int maxSegments) throws SolrServerException, IOException {
        this.getServer().optimize(true, true, maxSegments);
    }

    /**
     * Delete all the documents
     */
    public void clear() throws SolrServerException, IOException {
        this.getServer().deleteByQuery("*:*");
        this.getServer().commit();
    }

    public SolrServer getServer() {
        return server;
    }

    /**
     * Extends LukeRequest in order to disable the fetching of the top K terms
     * for each field.
     */
    private static class FastLukeRequest extends LukeRequest {

        private static final long serialVersionUID = 3210430588671309573L;

        @Override
        public SolrParams getParams() {
            final ModifiableSolrParams params = (ModifiableSolrParams) super.getParams();
            params.add("numTerms", "0");
            return params;
        }

    }

}