[ https://issues.apache.org/jira/browse/GORA-266?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16400605#comment-16400605 ]
ASF GitHub Bot commented on GORA-266: ------------------------------------- Github user renato2099 commented on a diff in the pull request: https://github.com/apache/gora/pull/131#discussion_r174719493 --- Diff: gora-lucene/src/main/java/org/apache/gora/lucene/query/LuceneResult.java --- @@ -0,0 +1,122 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you 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.apache.gora.lucene.query; + +import com.google.common.primitives.Ints; +import org.apache.gora.lucene.store.LuceneStore; +import org.apache.gora.persistency.impl.PersistentBase; +import org.apache.gora.query.Query; +import org.apache.gora.query.impl.PartitionQueryImpl; +import org.apache.gora.query.impl.ResultBase; +import org.apache.gora.store.DataStore; +import org.apache.lucene.document.Document; +import org.apache.lucene.search.IndexSearcher; +import org.apache.lucene.search.ScoreDoc; +import org.apache.lucene.search.SearcherManager; + +import java.io.IOException; +import java.util.Arrays; +import java.util.Collection; +import java.util.HashSet; +import java.util.Set; + +public class LuceneResult<K, T extends PersistentBase> extends ResultBase<K, T> { + + private ScoreDoc[] scoreDocs = null; + private final LuceneStore<K, T> store; + private String[] fields; + private int pos = 0; + private final SearcherManager searcherManager; + private IndexSearcher searcher; + + public LuceneResult(DataStore<K, T> dataStore, Query<K, T> query, + SearcherManager sm) throws IOException { + super(dataStore, query); + + searcherManager = sm; + store = (LuceneStore<K, T>)dataStore; + if (query instanceof PartitionQueryImpl) { + query = ((PartitionQueryImpl<K, T>)query).getBaseQuery(); + } + fields = query.getFields(); + if (fields != null) { + HashSet<String> uniqFields = new HashSet<>(Arrays.asList(fields)); + String keyFld = ((LuceneStore<K, T>)dataStore).getMapping().getPrimaryKey(); + uniqFields.add(keyFld); // return also primary key + query.setFields(fields); + } + else { + Collection<String> c = ((LuceneStore<K, T>)dataStore).getMapping().getLuceneFields(); + String[] a = {}; + fields = c.toArray(a); + query.setFields(fields); + } + // This is based on the limits of IndexSearcher.search(Query, int) + // A custom Collector could go larger than Integer.MAX_VALUE + // (NB: TotalHitCountCollector uses an int internally) + if (limit < 1L) + limit = Integer.MAX_VALUE; + + searcher = searcherManager.acquire(); + scoreDocs = searcher.search(((LuceneQuery<K, T>)query).toLuceneQuery(), Ints.checkedCast(limit)).scoreDocs; + } + + public ScoreDoc[] getScoreDocs() { + return scoreDocs; + } + + @SuppressWarnings("unchecked") + @Override + protected boolean nextInner() throws IOException { + if (scoreDocs == null || pos >= scoreDocs.length) { + return false; + } + + Set<String> f = null; + if (fields != null) { + f = new HashSet<>(fields.length); + f.addAll(Arrays.asList(fields)); + f.add(store.getMapping().getPrimaryKey()); + } + else { + Collection<String> c = ((LuceneStore<K, T>)dataStore).getMapping().getLuceneFields(); --- End diff -- The same comment as before, you should just use the "store" class variable to avoid more casts. > Lucene datastore for Gora > ------------------------- > > Key: GORA-266 > URL: https://issues.apache.org/jira/browse/GORA-266 > Project: Apache Gora > Issue Type: New Feature > Components: gora-lucene > Reporter: Lewis John McGibbney > Assignee: Lewis John McGibbney > Priority: Minor > Labels: features > Fix For: 0.9 > > Attachments: GORA-266.patch > > > Sometimes we don't need to set up, administer, maintain and configure a > running solr server(s). > Sometime we just want to write to Lucene index files and read directly from > them. -- This message was sent by Atlassian JIRA (v7.6.3#76005)