Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Search: Terms Stats Facet, closes #705.
- Loading branch information
Showing
14 changed files
with
2,212 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
39 changes: 39 additions & 0 deletions
39
...arch/src/main/java/org/elasticsearch/search/facet/termsstats/InternalTermsStatsFacet.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,39 @@ | ||
| /* | ||
| * Licensed to Elastic Search and Shay Banon under one | ||
| * or more contributor license agreements. See the NOTICE file | ||
| * distributed with this work for additional information | ||
| * regarding copyright ownership. Elastic Search 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.search.facet.termsstats; | ||
|
|
||
| import org.elasticsearch.search.facet.Facet; | ||
| import org.elasticsearch.search.facet.InternalFacet; | ||
| import org.elasticsearch.search.facet.termsstats.doubles.InternalTermsStatsDoubleFacet; | ||
| import org.elasticsearch.search.facet.termsstats.longs.InternalTermsStatsLongFacet; | ||
| import org.elasticsearch.search.facet.termsstats.strings.InternalTermsStatsStringFacet; | ||
|
|
||
| import java.util.List; | ||
|
|
||
| public abstract class InternalTermsStatsFacet implements TermsStatsFacet, InternalFacet { | ||
|
|
||
| public static void registerStreams() { | ||
| InternalTermsStatsStringFacet.registerStream(); | ||
| InternalTermsStatsLongFacet.registerStream(); | ||
| InternalTermsStatsDoubleFacet.registerStream(); | ||
| } | ||
|
|
||
| public abstract Facet reduce(String name, List<Facet> facets); | ||
| } |
187 changes: 187 additions & 0 deletions
187
...lasticsearch/src/main/java/org/elasticsearch/search/facet/termsstats/TermsStatsFacet.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,187 @@ | ||
| /* | ||
| * Licensed to Elastic Search and Shay Banon under one | ||
| * or more contributor license agreements. See the NOTICE file | ||
| * distributed with this work for additional information | ||
| * regarding copyright ownership. Elastic Search 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.search.facet.termsstats; | ||
|
|
||
| import org.elasticsearch.ElasticSearchIllegalArgumentException; | ||
| import org.elasticsearch.search.facet.Facet; | ||
|
|
||
| import java.util.Comparator; | ||
| import java.util.List; | ||
|
|
||
| public interface TermsStatsFacet extends Facet, Iterable<TermsStatsFacet.Entry> { | ||
|
|
||
| public static final String TYPE = "terms_stats"; | ||
|
|
||
|
|
||
| /** | ||
| * The number of docs missing a value. | ||
| */ | ||
| long missingCount(); | ||
|
|
||
| /** | ||
| * The number of docs missing a value. | ||
| */ | ||
| long getMissingCount(); | ||
|
|
||
| /** | ||
| * The terms and counts. | ||
| */ | ||
| List<? extends TermsStatsFacet.Entry> entries(); | ||
|
|
||
| /** | ||
| * The terms and counts. | ||
| */ | ||
| List<? extends TermsStatsFacet.Entry> getEntries(); | ||
|
|
||
| /** | ||
| * Controls how the terms facets are ordered. | ||
| */ | ||
| public static enum ComparatorType { | ||
| /** | ||
| * Order by the (higher) count of each term. | ||
| */ | ||
| COUNT((byte) 0, new Comparator<Entry>() { | ||
|
|
||
| @Override public int compare(Entry o1, Entry o2) { | ||
| int i = o2.count() - o1.count(); | ||
| if (i == 0) { | ||
| i = o2.term().compareTo(o1.term()); | ||
| if (i == 0) { | ||
| i = System.identityHashCode(o2) - System.identityHashCode(o1); | ||
| } | ||
| } | ||
| return i; | ||
| } | ||
| }), | ||
| /** | ||
| * Order by the (lower) count of each term. | ||
| */ | ||
| REVERSE_COUNT((byte) 1, new Comparator<Entry>() { | ||
|
|
||
| @Override public int compare(Entry o1, Entry o2) { | ||
| return -COUNT.comparator().compare(o1, o2); | ||
| } | ||
| }), | ||
| /** | ||
| * Order by the terms. | ||
| */ | ||
| TERM((byte) 2, new Comparator<Entry>() { | ||
|
|
||
| @Override public int compare(Entry o1, Entry o2) { | ||
| return o1.compareTo(o2); | ||
| } | ||
| }), | ||
| /** | ||
| * Order by the terms. | ||
| */ | ||
| REVERSE_TERM((byte) 3, new Comparator<Entry>() { | ||
|
|
||
| @Override public int compare(Entry o1, Entry o2) { | ||
| return -TERM.comparator().compare(o1, o2); | ||
| } | ||
| }), | ||
|
|
||
| TOTAL((byte) 4, new Comparator<Entry>() { | ||
| @Override public int compare(Entry o1, Entry o2) { | ||
| return (o2.total() < o1.total() ? -1 : (o2.total() == o1.total() ? 0 : 1)); | ||
| } | ||
| }), | ||
|
|
||
| REVERSE_TOTAL((byte) 5, new Comparator<Entry>() { | ||
| @Override public int compare(Entry o1, Entry o2) { | ||
| return (o1.total() < o2.total() ? -1 : (o1.total() == o2.total() ? 0 : 1)); | ||
| } | ||
| }); | ||
|
|
||
| private final byte id; | ||
|
|
||
| private final Comparator<Entry> comparator; | ||
|
|
||
| ComparatorType(byte id, Comparator<Entry> comparator) { | ||
| this.id = id; | ||
| this.comparator = comparator; | ||
| } | ||
|
|
||
| public byte id() { | ||
| return this.id; | ||
| } | ||
|
|
||
| public Comparator<Entry> comparator() { | ||
| return comparator; | ||
| } | ||
|
|
||
| public static ComparatorType fromId(byte id) { | ||
| if (id == COUNT.id()) { | ||
| return COUNT; | ||
| } else if (id == REVERSE_COUNT.id()) { | ||
| return REVERSE_COUNT; | ||
| } else if (id == TERM.id()) { | ||
| return TERM; | ||
| } else if (id == REVERSE_TERM.id()) { | ||
| return REVERSE_TERM; | ||
| } else if (id == TOTAL.id()) { | ||
| return TOTAL; | ||
| } else if (id == REVERSE_TOTAL.id()) { | ||
| return REVERSE_TOTAL; | ||
| } | ||
| throw new ElasticSearchIllegalArgumentException("No type argument match for terms facet comparator [" + id + "]"); | ||
| } | ||
|
|
||
| public static ComparatorType fromString(String type) { | ||
| if ("count".equals(type)) { | ||
| return COUNT; | ||
| } else if ("term".equals(type)) { | ||
| return TERM; | ||
| } else if ("reverse_count".equals(type) || "reverseCount".equals(type)) { | ||
| return REVERSE_COUNT; | ||
| } else if ("reverse_term".equals(type) || "reverseTerm".equals(type)) { | ||
| return REVERSE_TERM; | ||
| } else if ("total".equals(type)) { | ||
| return TOTAL; | ||
| } else if ("reverse_total".equals(type) || "reverseTotal".equals(type)) { | ||
| return REVERSE_TOTAL; | ||
| } | ||
| throw new ElasticSearchIllegalArgumentException("No type argument match for terms stats facet comparator [" + type + "]"); | ||
| } | ||
| } | ||
|
|
||
| public interface Entry extends Comparable<Entry> { | ||
|
|
||
| String term(); | ||
|
|
||
| String getTerm(); | ||
|
|
||
| Number termAsNumber(); | ||
|
|
||
| Number getTermAsNumber(); | ||
|
|
||
| int count(); | ||
|
|
||
| int getCount(); | ||
|
|
||
| double total(); | ||
|
|
||
| double getTotal(); | ||
|
|
||
| double mean(); | ||
|
|
||
| double getMean(); | ||
| } | ||
| } |
Oops, something went wrong.