org.openrdf.sail.elasticsearch.ElasticsearchQuery.java Source code

Introduction

Here is the source code for org.openrdf.sail.elasticsearch.ElasticsearchQuery.java

Source

/* 
 * Licensed to Aduna under one or more contributor license agreements.  
 * See the NOTICE.txt file distributed with this work for additional 
 * information regarding copyright ownership. 
 *
 * Aduna licenses this file to you under the terms of the Aduna BSD 
 * License (the "License"); you may not use this file except in compliance 
 * with the License. See the LICENSE.txt file distributed with this work 
 * for the full License.
 *
 * 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.openrdf.sail.elasticsearch;

import java.io.IOException;

import org.elasticsearch.action.search.SearchRequestBuilder;
import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.SearchHits;
import org.openrdf.model.Resource;
import org.openrdf.model.URI;
import org.openrdf.sail.lucene.DocumentScore;
import org.openrdf.sail.lucene.SearchFields;
import org.openrdf.sail.lucene.SearchQuery;

import com.google.common.base.Function;
import com.google.common.collect.Iterables;

/**
 * To be removed, no longer used.
 */
@Deprecated
public class ElasticsearchQuery implements SearchQuery {
    private final SearchRequestBuilder request;
    private final QueryBuilder qb;
    private ElasticsearchIndex index;

    public ElasticsearchQuery(SearchRequestBuilder request, QueryBuilder qb, ElasticsearchIndex index) {
        this.request = request;
        this.qb = qb;
        this.index = index;
    }

    @Override
    public Iterable<? extends DocumentScore> query(Resource resource) throws IOException {
        SearchHits hits;
        if (resource != null) {
            hits = index.search(resource, request, qb);
        } else {
            hits = index.search(request, qb);
        }
        return Iterables.transform(hits, new Function<SearchHit, DocumentScore>() {
            @Override
            public DocumentScore apply(SearchHit hit) {
                return new ElasticsearchDocumentScore(hit, null);
            }
        });
    }

    /**
     * Highlights the given field or all fields if null.
     */
    @Override
    public void highlight(URI property) {
        String field = (property != null) ? SearchFields.getPropertyField(property) : "*";
        request.addHighlightedField(field);
        request.setHighlighterPreTags(SearchFields.HIGHLIGHTER_PRE_TAG);
        request.setHighlighterPostTags(SearchFields.HIGHLIGHTER_POST_TAG);
        // Elastic Search doesn't really have the same support for fragments as Lucene.
        // So, we have to get back the whole highlighted value (comma-separated if it is a list)
        // and then post-process it into fragments ourselves.
        request.setHighlighterNumOfFragments(0);
    }
}