com.jaeksoft.searchlib.result.ResultDocuments Maven / Gradle / Ivy
/**
* License Agreement for OpenSearchServer
*
* Copyright (C) 2012-2014 Emmanuel Keller / Jaeksoft
*
* http://www.open-search-server.com
*
* This file is part of OpenSearchServer.
*
* OpenSearchServer is free software: you can redistribute it and/or
* modify it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* OpenSearchServer 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with OpenSearchServer.
* If not, see implements
ResultDocumentsInterface {
transient private ReaderInterface reader = null;
final private TreeSet fieldNameSet;
final private int[] docArray;
private ResultDocuments(ReaderInterface reader, AbstractRequest request,
TreeSet fieldNameSet, int[] docArray) {
super(request);
this.reader = reader;
this.fieldNameSet = fieldNameSet == null ? new TreeSet()
: fieldNameSet;
if (this.fieldNameSet.size() == 0
&& request instanceof RequestInterfaces.ReturnedFieldInterface)
((RequestInterfaces.ReturnedFieldInterface) request)
.getReturnFieldList().populate(this.fieldNameSet);
this.docArray = docArray;
}
public ResultDocuments(ReaderInterface reader, AbstractRequest request,
TreeSet fieldNameSet, List docList) {
this(reader, request, fieldNameSet, toDocArray(docList));
}
private final static int[] toDocArray(List docList) {
if (CollectionUtils.isEmpty(docList))
return null;
int[] docArray = new int[docList.size()];
int i = 0;
for (Integer docId : docList)
docArray[i++] = docId;
return docArray;
}
public ResultDocuments(ReaderLocal reader, DocumentsRequest request)
throws IOException, SearchLibException {
this(reader, request, null, toDocArray(reader, request));
}
private final static int[] toDocArray(ReaderLocal reader,
DocumentsRequest request) throws IOException {
SchemaField schemaField = null;
Schema schema = request.getConfig().getSchema();
String field = request.getField();
if (!StringUtils.isEmpty(field)) {
schemaField = schema.getField(field);
if (schemaField == null)
throw new IOException("Field not found: " + field);
} else {
schemaField = schema.getFieldList().getUniqueField();
if (schemaField == null)
throw new IOException("No unique field");
}
int maxDoc = reader.getStatistics().getMaxDoc();
BitSetInterface bitSet = BitSetFactory.INSTANCE.newInstance(maxDoc);
String fieldName = schemaField.getName();
for (String uniqueKey : request.getUniqueKeyList()) {
TermDocs termDocs = reader.getTermDocs(new Term(fieldName,
uniqueKey));
if (termDocs != null)
while (termDocs.next())
bitSet.set(termDocs.doc());
termDocs.close();
}
if (request.isReverse())
bitSet.flip(0, maxDoc);
IntBufferedArrayInterface intBufferArray = IntBufferedArrayFactory.INSTANCE
.newInstance((int) bitSet.cardinality());
DocIdSetIterator iterator = BitSetFactory.INSTANCE.getDocIdSet(bitSet)
.iterator();
int docId;
while ((docId = iterator.nextDoc()) != DocIdSetIterator.NO_MORE_DOCS)
if (!reader.isDeletedNoLock(docId))
intBufferArray.add(docId);
return intBufferArray.getFinalArray();
}
@Override
public ResultDocument getDocument(int pos, Timer timer)
throws SearchLibException {
if (docArray == null || pos < 0 || pos > docArray.length)
return null;
try {
return new ResultDocument(fieldNameSet, docArray[pos], reader,
getScore(pos), null, timer);
} catch (IOException e) {
throw new SearchLibException(e);
} catch (ParseException e) {
throw new SearchLibException(e);
} catch (SyntaxError e) {
throw new SearchLibException(e);
}
}
@Override
public void populate(List indexDocuments)
throws IOException, SearchLibException {
SchemaFieldList schemaFieldList = request.getConfig().getSchema()
.getFieldList();
for (int docId : docArray) {
IndexDocumentResult indexDocument = new IndexDocumentResult(
schemaFieldList.size());
Map storedFieldMap = reader
.getDocumentStoredField(docId);
for (SchemaField schemaField : schemaFieldList) {
String fieldName = schemaField.getName();
List indexTermList = null;
if (schemaField.checkIndexed(Indexed.YES)) {
if (schemaField.getTermVector() == TermVector.NO) {
indexTermList = IndexTerm.toList(reader, fieldName,
docId);
} else {
TermFreqVector termFreqVector = reader
.getTermFreqVector(docId, fieldName);
indexTermList = IndexTerm.toList(termFreqVector);
}
}
IndexField indexField = new IndexField(fieldName,
storedFieldMap.get(fieldName), indexTermList);
indexDocument.add(indexField);
}
indexDocuments.add(indexDocument);
}
}
@Override
public float getScore(int pos) {
return 0;
}
@Override
public Float getDistance(int pos) {
return null;
}
@Override
public int getCollapseCount(int pos) {
return 0;
}
@Override
public int getNumFound() {
if (docArray == null)
return 0;
return docArray.length;
}
@Override
protected Render getRenderXml() {
return new RenderDocumentsXml(this);
}
@Override
protected Render getRenderCsv() {
// TODO Auto-generated method stub
return null;
}
@Override
protected Render getRenderJson(boolean indent) {
return new RenderDocumentsJson(this, indent);
}
@Override
public Iterator iterator() {
return new ResultDocumentIterator(this, null);
}
@Override
public int getDocumentCount() {
return docArray == null ? 0 : docArray.length;
}
@Override
public int getRequestStart() {
return 0;
}
@Override
public int getRequestRows() {
return docArray == null ? 0 : docArray.length;
}
@Override
public DocIdInterface getDocs() {
return null;
}
public int[] getDocIdArray() {
return docArray;
}
@Override
public float getMaxScore() {
return 0;
}
@Override
public int getCollapsedDocCount() {
return 0;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy