mirror of
https://github.com/honeymoose/OpenSearch.git
synced 2025-04-01 04:48:28 +00:00
f01d3a18ad
Removing #getValue and #hasValue to have a simple and consistent API for multiple values.
478 lines
24 KiB
Java
478 lines
24 KiB
Java
/*
|
|
* Licensed to ElasticSearch and Shay Banon under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. ElasticSearch 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.elasticsearch.index.fielddata;
|
|
|
|
import org.apache.lucene.document.Document;
|
|
import org.apache.lucene.document.Field;
|
|
import org.apache.lucene.document.StringField;
|
|
import org.apache.lucene.index.DirectoryReader;
|
|
import org.apache.lucene.search.*;
|
|
import org.elasticsearch.index.fielddata.fieldcomparator.SortMode;
|
|
import org.junit.Test;
|
|
|
|
import static org.hamcrest.Matchers.equalTo;
|
|
|
|
/**
|
|
*/
|
|
public abstract class AbstractNumericFieldDataTests extends AbstractFieldDataImplTests {
|
|
|
|
protected abstract FieldDataType getFieldDataType();
|
|
|
|
@Test
|
|
public void testSingleValueAllSetNumber() throws Exception {
|
|
fillSingleValueAllSet();
|
|
IndexNumericFieldData indexFieldData = getForField("value");
|
|
AtomicNumericFieldData fieldData = indexFieldData.load(refreshReader());
|
|
|
|
assertThat(fieldData.getNumDocs(), equalTo(3));
|
|
|
|
LongValues longValues = fieldData.getLongValues();
|
|
|
|
assertThat(longValues.isMultiValued(), equalTo(false));
|
|
|
|
assertThat(longValues.setDocument(0), equalTo(1));
|
|
assertThat(longValues.nextValue(), equalTo(2l));
|
|
|
|
assertThat(longValues.setDocument(1), equalTo(1));
|
|
assertThat(longValues.nextValue(), equalTo(1l));
|
|
|
|
assertThat(longValues.setDocument(2), equalTo(1));
|
|
assertThat(longValues.nextValue(), equalTo(3l));
|
|
|
|
DoubleValues doubleValues = fieldData.getDoubleValues();
|
|
|
|
assertThat(doubleValues.isMultiValued(), equalTo(false));
|
|
|
|
assertThat(1, equalTo(doubleValues.setDocument(0)));
|
|
assertThat(doubleValues.nextValue(), equalTo(2d));
|
|
|
|
assertThat(1, equalTo(doubleValues.setDocument(1)));
|
|
assertThat(doubleValues.nextValue(), equalTo(1d));
|
|
|
|
assertThat(1, equalTo(doubleValues.setDocument(2)));
|
|
assertThat(doubleValues.nextValue(), equalTo(3d));
|
|
|
|
IndexSearcher searcher = new IndexSearcher(readerContext.reader());
|
|
TopFieldDocs topDocs;
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource(null, SortMode.MIN))));
|
|
assertThat(topDocs.totalHits, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(1));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(2));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource(null, SortMode.MAX), true)));
|
|
assertThat(topDocs.totalHits, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(2));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(1));
|
|
}
|
|
|
|
@Test
|
|
public void testSingleValueWithMissingNumber() throws Exception {
|
|
fillSingleValueWithMissing();
|
|
IndexNumericFieldData indexFieldData = getForField("value");
|
|
AtomicNumericFieldData fieldData = indexFieldData.load(refreshReader());
|
|
|
|
assertThat(fieldData.getNumDocs(), equalTo(3));
|
|
|
|
LongValues longValues = fieldData.getLongValues();
|
|
|
|
assertThat(longValues.isMultiValued(), equalTo(false));
|
|
|
|
assertThat(longValues.setDocument(0), equalTo(1));
|
|
assertThat(longValues.nextValue(), equalTo(2l));
|
|
|
|
assertThat(longValues.setDocument(1), equalTo(0));
|
|
|
|
assertThat(longValues.setDocument(2), equalTo(1));
|
|
assertThat(longValues.nextValue(), equalTo(3l));
|
|
|
|
DoubleValues doubleValues = fieldData.getDoubleValues();
|
|
|
|
assertThat(doubleValues.isMultiValued(), equalTo(false));
|
|
|
|
assertThat(1, equalTo(doubleValues.setDocument(0)));
|
|
assertThat(doubleValues.nextValue(), equalTo(2d));
|
|
|
|
assertThat(0, equalTo(doubleValues.setDocument(1)));
|
|
|
|
assertThat(1, equalTo(doubleValues.setDocument(2)));
|
|
assertThat(doubleValues.nextValue(), equalTo(3d));
|
|
|
|
IndexSearcher searcher = new IndexSearcher(readerContext.reader());
|
|
TopFieldDocs topDocs;
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource(null, SortMode.MIN)))); // defaults to _last
|
|
assertThat(topDocs.totalHits, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(2));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(1));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource(null, SortMode.MAX), true))); // defaults to _last
|
|
assertThat(topDocs.totalHits, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(2));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(1));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource("_first", SortMode.MIN))));
|
|
assertThat(topDocs.totalHits, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(1));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(2));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource("_first", SortMode.MAX), true)));
|
|
assertThat(topDocs.totalHits, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(1));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(2));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(0));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource("1", SortMode.MIN))));
|
|
assertThat(topDocs.totalHits, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(1));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(2));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource("1", SortMode.MAX), true)));
|
|
assertThat(topDocs.totalHits, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(2));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(1));
|
|
}
|
|
|
|
@Test
|
|
public void testMultiValueAllSetNumber() throws Exception {
|
|
fillMultiValueAllSet();
|
|
IndexNumericFieldData indexFieldData = getForField("value");
|
|
AtomicNumericFieldData fieldData = indexFieldData.load(refreshReader());
|
|
|
|
assertThat(fieldData.getNumDocs(), equalTo(3));
|
|
|
|
LongValues longValues = fieldData.getLongValues();
|
|
|
|
assertThat(longValues.isMultiValued(), equalTo(true));
|
|
|
|
assertThat(longValues.setDocument(0), equalTo(2));
|
|
assertThat(longValues.nextValue(), equalTo(2l));
|
|
assertThat(longValues.nextValue(), equalTo(4l));
|
|
|
|
assertThat(longValues.setDocument(1), equalTo(1));
|
|
assertThat(longValues.nextValue(), equalTo(1l));
|
|
|
|
assertThat(longValues.setDocument(2), equalTo(1));
|
|
assertThat(longValues.nextValue(), equalTo(3l));
|
|
|
|
DoubleValues doubleValues = fieldData.getDoubleValues();
|
|
|
|
assertThat(doubleValues.isMultiValued(), equalTo(true));
|
|
|
|
assertThat(2, equalTo(doubleValues.setDocument(0)));
|
|
assertThat(doubleValues.nextValue(), equalTo(2d));
|
|
assertThat(doubleValues.nextValue(), equalTo(4d));
|
|
|
|
assertThat(1, equalTo(doubleValues.setDocument(1)));
|
|
assertThat(doubleValues.nextValue(), equalTo(1d));
|
|
|
|
assertThat(1, equalTo(doubleValues.setDocument(2)));
|
|
assertThat(doubleValues.nextValue(), equalTo(3d));
|
|
}
|
|
|
|
@Test
|
|
public void testMultiValueWithMissingNumber() throws Exception {
|
|
fillMultiValueWithMissing();
|
|
IndexNumericFieldData indexFieldData = getForField("value");
|
|
AtomicNumericFieldData fieldData = indexFieldData.load(refreshReader());
|
|
|
|
assertThat(fieldData.getNumDocs(), equalTo(3));
|
|
|
|
LongValues longValues = fieldData.getLongValues();
|
|
|
|
assertThat(longValues.isMultiValued(), equalTo(true));
|
|
|
|
assertThat(longValues.setDocument(0), equalTo(2));
|
|
assertThat(longValues.nextValue(), equalTo(2l));
|
|
assertThat(longValues.nextValue(), equalTo(4l));
|
|
|
|
assertThat(longValues.setDocument(1), equalTo(0));
|
|
|
|
assertThat(longValues.setDocument(2), equalTo(1));
|
|
assertThat(longValues.nextValue(), equalTo(3l));
|
|
|
|
DoubleValues doubleValues = fieldData.getDoubleValues();
|
|
|
|
assertThat(doubleValues.isMultiValued(), equalTo(true));
|
|
|
|
|
|
assertThat(2, equalTo(doubleValues.setDocument(0)));
|
|
assertThat(doubleValues.nextValue(), equalTo(2d));
|
|
assertThat(doubleValues.nextValue(), equalTo(4d));
|
|
|
|
assertThat(0, equalTo(doubleValues.setDocument(1)));
|
|
|
|
assertThat(1, equalTo(doubleValues.setDocument(2)));
|
|
assertThat(doubleValues.nextValue(), equalTo(3d));
|
|
|
|
}
|
|
|
|
@Test
|
|
public void testMissingValueForAll() throws Exception {
|
|
fillAllMissing();
|
|
IndexNumericFieldData indexFieldData = getForField("value");
|
|
AtomicNumericFieldData fieldData = indexFieldData.load(refreshReader());
|
|
|
|
assertThat(fieldData.getNumDocs(), equalTo(3));
|
|
|
|
// long values
|
|
|
|
LongValues longValues = fieldData.getLongValues();
|
|
|
|
assertThat(longValues.isMultiValued(), equalTo(false));
|
|
|
|
assertThat(longValues.setDocument(0), equalTo(0));
|
|
assertThat(longValues.setDocument(1), equalTo(0));
|
|
assertThat(longValues.setDocument(2), equalTo(0));
|
|
|
|
// double values
|
|
|
|
DoubleValues doubleValues = fieldData.getDoubleValues();
|
|
|
|
assertThat(doubleValues.isMultiValued(), equalTo(false));
|
|
|
|
assertThat(0, equalTo(doubleValues.setDocument(0)));
|
|
|
|
assertThat(0, equalTo(doubleValues.setDocument(1)));
|
|
|
|
assertThat(0, equalTo(doubleValues.setDocument(2)));
|
|
}
|
|
|
|
|
|
protected void fillAllMissing() throws Exception {
|
|
Document d = new Document();
|
|
d.add(new StringField("_id", "1", Field.Store.NO));
|
|
writer.addDocument(d);
|
|
|
|
d = new Document();
|
|
d.add(new StringField("_id", "2", Field.Store.NO));
|
|
writer.addDocument(d);
|
|
|
|
d = new Document();
|
|
d.add(new StringField("_id", "3", Field.Store.NO));
|
|
writer.addDocument(d);
|
|
}
|
|
|
|
@Test
|
|
public void testSortMultiValuesFields() throws Exception {
|
|
fillExtendedMvSet();
|
|
IndexFieldData indexFieldData = getForField("value");
|
|
|
|
IndexSearcher searcher = new IndexSearcher(DirectoryReader.open(writer, true));
|
|
TopFieldDocs topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource(null, SortMode.MIN)))); // defaults to _last
|
|
assertThat(topDocs.totalHits, equalTo(8));
|
|
assertThat(topDocs.scoreDocs.length, equalTo(8));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(7));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[0]).fields[0]).intValue(), equalTo(-10));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(0));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[1]).fields[0]).intValue(), equalTo(2));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(2));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[2]).fields[0]).intValue(), equalTo(3));
|
|
assertThat(topDocs.scoreDocs[3].doc, equalTo(3));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[3]).fields[0]).intValue(), equalTo(4));
|
|
assertThat(topDocs.scoreDocs[4].doc, equalTo(4));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[4]).fields[0]).intValue(), equalTo(6));
|
|
assertThat(topDocs.scoreDocs[5].doc, equalTo(6));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[5]).fields[0]).intValue(), equalTo(8));
|
|
assertThat(topDocs.scoreDocs[6].doc, equalTo(1));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[6]).fields[0], equalTo(null));
|
|
assertThat(topDocs.scoreDocs[7].doc, equalTo(5));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[7]).fields[0], equalTo(null));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource(null, SortMode.MAX), true))); // defaults to _last
|
|
assertThat(topDocs.totalHits, equalTo(8));
|
|
assertThat(topDocs.scoreDocs.length, equalTo(8));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(6));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[0]).fields[0]).intValue(), equalTo(10));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(4));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[1]).fields[0]).intValue(), equalTo(8));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(3));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[2]).fields[0]).intValue(), equalTo(6));
|
|
assertThat(topDocs.scoreDocs[3].doc, equalTo(0));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[3]).fields[0]).intValue(), equalTo(4));
|
|
assertThat(topDocs.scoreDocs[4].doc, equalTo(2));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[4]).fields[0]).intValue(), equalTo(3));
|
|
assertThat(topDocs.scoreDocs[5].doc, equalTo(7));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[5]).fields[0]).intValue(), equalTo(-8));
|
|
assertThat(topDocs.scoreDocs[6].doc, equalTo(1));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[6]).fields[0], equalTo(null));
|
|
assertThat(topDocs.scoreDocs[7].doc, equalTo(5));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[7]).fields[0], equalTo(null));
|
|
|
|
searcher = new IndexSearcher(DirectoryReader.open(writer, true));
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource(null, SortMode.SUM)))); // defaults to _last
|
|
assertThat(topDocs.totalHits, equalTo(8));
|
|
assertThat(topDocs.scoreDocs.length, equalTo(8));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(7));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[0]).fields[0]).intValue(), equalTo(-27));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(2));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[1]).fields[0]).intValue(), equalTo(3));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(0));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[2]).fields[0]).intValue(), equalTo(6));
|
|
assertThat(topDocs.scoreDocs[3].doc, equalTo(3));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[3]).fields[0]).intValue(), equalTo(15));
|
|
assertThat(topDocs.scoreDocs[4].doc, equalTo(4));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[4]).fields[0]).intValue(), equalTo(21));
|
|
assertThat(topDocs.scoreDocs[5].doc, equalTo(6));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[5]).fields[0]).intValue(), equalTo(27));
|
|
assertThat(topDocs.scoreDocs[6].doc, equalTo(1));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[6]).fields[0], equalTo(null));
|
|
assertThat(topDocs.scoreDocs[7].doc, equalTo(5));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[7]).fields[0], equalTo(null));
|
|
|
|
searcher = new IndexSearcher(DirectoryReader.open(writer, true));
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource(null, SortMode.SUM), true))); // defaults to _last
|
|
assertThat(topDocs.totalHits, equalTo(8));
|
|
assertThat(topDocs.scoreDocs.length, equalTo(8));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(6));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[0]).fields[0]).intValue(), equalTo(27));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(4));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[1]).fields[0]).intValue(), equalTo(21));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(3));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[2]).fields[0]).intValue(), equalTo(15));
|
|
assertThat(topDocs.scoreDocs[3].doc, equalTo(0));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[3]).fields[0]).intValue(), equalTo(6));
|
|
assertThat(topDocs.scoreDocs[4].doc, equalTo(2));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[4]).fields[0]).intValue(), equalTo(3));
|
|
assertThat(topDocs.scoreDocs[5].doc, equalTo(7));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[5]).fields[0]).intValue(), equalTo(-27));
|
|
assertThat(topDocs.scoreDocs[6].doc, equalTo(1));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[6]).fields[0], equalTo(null));
|
|
assertThat(topDocs.scoreDocs[7].doc, equalTo(5));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[7]).fields[0], equalTo(null));
|
|
|
|
searcher = new IndexSearcher(DirectoryReader.open(writer, true));
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource(null, SortMode.AVG)))); // defaults to _last
|
|
assertThat(topDocs.totalHits, equalTo(8));
|
|
assertThat(topDocs.scoreDocs.length, equalTo(8));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(7));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[0]).fields[0]).intValue(), equalTo(-9));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(0));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[1]).fields[0]).intValue(), equalTo(3));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(2));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[2]).fields[0]).intValue(), equalTo(3));
|
|
assertThat(topDocs.scoreDocs[3].doc, equalTo(3));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[3]).fields[0]).intValue(), equalTo(5));
|
|
assertThat(topDocs.scoreDocs[4].doc, equalTo(4));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[4]).fields[0]).intValue(), equalTo(7));
|
|
assertThat(topDocs.scoreDocs[5].doc, equalTo(6));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[5]).fields[0]).intValue(), equalTo(9));
|
|
assertThat(topDocs.scoreDocs[6].doc, equalTo(1));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[6]).fields[0], equalTo(null));
|
|
assertThat(topDocs.scoreDocs[7].doc, equalTo(5));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[7]).fields[0], equalTo(null));
|
|
|
|
searcher = new IndexSearcher(DirectoryReader.open(writer, true));
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource(null, SortMode.AVG), true))); // defaults to _last
|
|
assertThat(topDocs.totalHits, equalTo(8));
|
|
assertThat(topDocs.scoreDocs.length, equalTo(8));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(6));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[0]).fields[0]).intValue(), equalTo(9));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(4));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[1]).fields[0]).intValue(), equalTo(7));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(3));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[2]).fields[0]).intValue(), equalTo(5));
|
|
assertThat(topDocs.scoreDocs[3].doc, equalTo(0));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[3]).fields[0]).intValue(), equalTo(3));
|
|
assertThat(topDocs.scoreDocs[4].doc, equalTo(2));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[4]).fields[0]).intValue(), equalTo(3));
|
|
assertThat(topDocs.scoreDocs[5].doc, equalTo(7));
|
|
assertThat(((Number) ((FieldDoc) topDocs.scoreDocs[5]).fields[0]).intValue(), equalTo(-9));
|
|
assertThat(topDocs.scoreDocs[6].doc, equalTo(1));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[6]).fields[0], equalTo(null));
|
|
assertThat(topDocs.scoreDocs[7].doc, equalTo(5));
|
|
// assertThat(((FieldDoc) topDocs.scoreDocs[7]).fields[0], equalTo(null));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource("_first", SortMode.MIN))));
|
|
assertThat(topDocs.totalHits, equalTo(8));
|
|
assertThat(topDocs.scoreDocs.length, equalTo(8));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(1));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(5));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(7));
|
|
assertThat(topDocs.scoreDocs[3].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[4].doc, equalTo(2));
|
|
assertThat(topDocs.scoreDocs[5].doc, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[6].doc, equalTo(4));
|
|
assertThat(topDocs.scoreDocs[7].doc, equalTo(6));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource("_first", SortMode.MAX), true)));
|
|
assertThat(topDocs.totalHits, equalTo(8));
|
|
assertThat(topDocs.scoreDocs.length, equalTo(8));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(1));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(5));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(6));
|
|
assertThat(topDocs.scoreDocs[3].doc, equalTo(4));
|
|
assertThat(topDocs.scoreDocs[4].doc, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[5].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[6].doc, equalTo(2));
|
|
assertThat(topDocs.scoreDocs[7].doc, equalTo(7));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource("-9", SortMode.MIN))));
|
|
assertThat(topDocs.totalHits, equalTo(8));
|
|
assertThat(topDocs.scoreDocs.length, equalTo(8));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(7));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(1));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(5));
|
|
assertThat(topDocs.scoreDocs[3].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[4].doc, equalTo(2));
|
|
assertThat(topDocs.scoreDocs[5].doc, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[6].doc, equalTo(4));
|
|
assertThat(topDocs.scoreDocs[7].doc, equalTo(6));
|
|
|
|
topDocs = searcher.search(new MatchAllDocsQuery(), 10,
|
|
new Sort(new SortField("value", indexFieldData.comparatorSource("9", SortMode.MAX), true)));
|
|
assertThat(topDocs.totalHits, equalTo(8));
|
|
assertThat(topDocs.scoreDocs.length, equalTo(8));
|
|
assertThat(topDocs.scoreDocs[0].doc, equalTo(6));
|
|
assertThat(topDocs.scoreDocs[1].doc, equalTo(1));
|
|
assertThat(topDocs.scoreDocs[2].doc, equalTo(5));
|
|
assertThat(topDocs.scoreDocs[3].doc, equalTo(4));
|
|
assertThat(topDocs.scoreDocs[4].doc, equalTo(3));
|
|
assertThat(topDocs.scoreDocs[5].doc, equalTo(0));
|
|
assertThat(topDocs.scoreDocs[6].doc, equalTo(2));
|
|
assertThat(topDocs.scoreDocs[7].doc, equalTo(7));
|
|
}
|
|
|
|
}
|