Call onMissing if doc has no value in the field.

Closes #2807
This commit is contained in:
Simon Willnauer 2013-03-21 22:45:17 +01:00
parent 064d272916
commit 075779a397
7 changed files with 121 additions and 0 deletions

View File

@ -485,6 +485,8 @@ public abstract class ByteArrayAtomicFieldData implements AtomicNumericFieldData
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, (long) values[docId]);
} else {
proc.onMissing(docId);
}
}
}
@ -549,6 +551,8 @@ public abstract class ByteArrayAtomicFieldData implements AtomicNumericFieldData
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, (double) values[docId]);
} else {
proc.onMissing(docId);
}
}
}

View File

@ -642,6 +642,8 @@ public abstract class DoubleArrayAtomicFieldData implements AtomicNumericFieldDa
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, values[docId]);
} else {
proc.onMissing(docId);
}
}
}

View File

@ -484,6 +484,8 @@ public abstract class FloatArrayAtomicFieldData implements AtomicNumericFieldDat
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, (long) values[docId]);
} else {
proc.onMissing(docId);
}
}
}
@ -548,6 +550,8 @@ public abstract class FloatArrayAtomicFieldData implements AtomicNumericFieldDat
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, (double) values[docId]);
} else {
proc.onMissing(docId);
}
}
}

View File

@ -483,6 +483,8 @@ public abstract class IntArrayAtomicFieldData implements AtomicNumericFieldData
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, (long) values[docId]);
} else {
proc.onMissing(docId);
}
}
}
@ -547,6 +549,8 @@ public abstract class IntArrayAtomicFieldData implements AtomicNumericFieldData
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, (double) values[docId]);
} else {
proc.onMissing(docId);
}
}
}

View File

@ -639,6 +639,8 @@ public abstract class LongArrayAtomicFieldData implements AtomicNumericFieldData
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, values[docId]);
} else {
proc.onMissing(docId);
}
}
}
@ -703,6 +705,8 @@ public abstract class LongArrayAtomicFieldData implements AtomicNumericFieldData
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, (double) values[docId]);
} else {
proc.onMissing(docId);
}
}
}

View File

@ -483,6 +483,8 @@ public abstract class ShortArrayAtomicFieldData implements AtomicNumericFieldDat
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, (long) values[docId]);
} else {
proc.onMissing(docId);
}
}
}
@ -547,6 +549,8 @@ public abstract class ShortArrayAtomicFieldData implements AtomicNumericFieldDat
public void forEachValueInDoc(int docId, ValueInDocProc proc) {
if (set.get(docId)) {
proc.onValue(docId, (double) values[docId]);
} else {
proc.onMissing(docId);
}
}
}

View File

@ -19,6 +19,7 @@
package org.elasticsearch.test.integration.search.facet;
import org.elasticsearch.ElasticSearchException;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.action.search.SearchType;
import org.elasticsearch.action.search.ShardSearchFailure;
@ -46,6 +47,7 @@ import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;
import java.io.IOException;
import java.util.concurrent.TimeUnit;
import static org.elasticsearch.common.xcontent.XContentFactory.jsonBuilder;
@ -140,6 +142,103 @@ public class SimpleFacetsTests extends AbstractNodesTests {
}
}
@Test
public void testFacetNumeric() throws ElasticSearchException, IOException {
// TODO we should test this with more complex queries
try {
client.admin().indices().prepareDelete("test").execute().actionGet();
} catch (Exception e) {
// ignore
}
client.admin().indices().prepareCreate("test")
.addMapping("type", jsonBuilder().startObject().startObject("type").startObject("properties")
.startObject("byte").field("type", "byte").endObject()
.startObject("short").field("type", "short").endObject()
.startObject("integer").field("type", "integer").endObject()
.startObject("long").field("type", "long").endObject()
.startObject("float").field("type", "float").endObject()
.startObject("double").field("type", "double").endObject()
.endObject().endObject().endObject())
.execute().actionGet();
client.admin().cluster().prepareHealth().setWaitForEvents(Priority.LANGUID).setWaitForGreenStatus().execute().actionGet();
for (int i = 0; i < 100; i++) {
client.prepareIndex("test", "type", ""+i).setSource(jsonBuilder().startObject()
.field("name", ""+i)
.field("byte", i )
.field("short", i + Byte.MAX_VALUE)
.field("integer", i + Short.MAX_VALUE)
.field("long", i + Integer.MAX_VALUE)
.field("float", (float)i)
.field("double", (double)i)
.endObject()).execute().actionGet();
}
for (int i = 0; i < 10; i++) {
client.prepareIndex("test", "type", ""+(i + 100)).setSource(jsonBuilder().startObject()
.field("foo", ""+i)
.endObject()).execute().actionGet();
}
client.admin().indices().prepareFlush().setRefresh(true).execute().actionGet();
SearchResponse searchResponse = client.prepareSearch()
.setQuery(matchAllQuery())
.addFacet(termsFacet("double").field("double").size(10))
.addFacet(termsFacet("float").field("float").size(10))
.addFacet(termsFacet("integer").field("integer").size(10))
.addFacet(termsFacet("long").field("long").size(10))
.addFacet(termsFacet("short").field("short").size(10))
.addFacet(termsFacet("byte").field("byte").size(10))
.addFacet(termsFacet("termFacet").field("name").size(10))
.execute().actionGet();
assertThat(searchResponse.getHits().getTotalHits(), equalTo(110l));
TermsFacet facet = searchResponse.getFacets().facet("termFacet");
assertThat(facet.getName(), equalTo("termFacet"));
assertThat(facet.getEntries().size(), equalTo(10));
assertThat(facet.getTotalCount(), equalTo(100l));
assertThat(facet.getOtherCount(), equalTo(90l));
assertThat(facet.getMissingCount(), equalTo(10l));
facet = searchResponse.getFacets().facet("double");
assertThat(facet.getName(), equalTo("double"));
assertThat(facet.getEntries().size(), equalTo(10));
assertThat(facet.getTotalCount(), equalTo(100l));
assertThat(facet.getOtherCount(), equalTo(90l));
assertThat(facet.getMissingCount(), equalTo(10l));
facet = searchResponse.getFacets().facet("float");
assertThat(facet.getName(), equalTo("float"));
assertThat(facet.getEntries().size(), equalTo(10));
assertThat(facet.getTotalCount(), equalTo(100l));
assertThat(facet.getOtherCount(), equalTo(90l));
assertThat(facet.getMissingCount(), equalTo(10l));
facet = searchResponse.getFacets().facet("long");
assertThat(facet.getName(), equalTo("long"));
assertThat(facet.getEntries().size(), equalTo(10));
assertThat(facet.getTotalCount(), equalTo(100l));
assertThat(facet.getOtherCount(), equalTo(90l));
assertThat(facet.getMissingCount(), equalTo(10l));
facet = searchResponse.getFacets().facet("integer");
assertThat(facet.getName(), equalTo("integer"));
assertThat(facet.getEntries().size(), equalTo(10));
assertThat(facet.getTotalCount(), equalTo(100l));
assertThat(facet.getOtherCount(), equalTo(90l));
assertThat(facet.getMissingCount(), equalTo(10l));
facet = searchResponse.getFacets().facet("short");
assertThat(facet.getName(), equalTo("short"));
assertThat(facet.getEntries().size(), equalTo(10));
assertThat(facet.getTotalCount(), equalTo(100l));
assertThat(facet.getOtherCount(), equalTo(90l));
assertThat(facet.getMissingCount(), equalTo(10l));
}
@Test
public void testSearchFilter() throws Exception {
try {