org.elasticsearch.search.aggregations.bucket.terms.StringRareTerms Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of elasticsearch Show documentation
Show all versions of elasticsearch Show documentation
Elasticsearch subproject :server
/*
* Licensed to Elasticsearch 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.search.aggregations.bucket.terms;
import org.apache.lucene.util.BytesRef;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
import org.elasticsearch.common.util.SetBackedScalingCuckooFilter;
import org.elasticsearch.common.xcontent.XContentBuilder;
import org.elasticsearch.search.DocValueFormat;
import org.elasticsearch.search.aggregations.BucketOrder;
import org.elasticsearch.search.aggregations.InternalAggregations;
import org.elasticsearch.search.aggregations.pipeline.PipelineAggregator;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Objects;
public class StringRareTerms extends InternalMappedRareTerms {
public static final String NAME = "srareterms";
public static class Bucket extends InternalRareTerms.Bucket {
BytesRef termBytes;
public Bucket(BytesRef term, long docCount, InternalAggregations aggregations, DocValueFormat format) {
super(docCount, aggregations, format);
this.termBytes = term;
}
/**
* Read from a stream.
*/
public Bucket(StreamInput in, DocValueFormat format) throws IOException {
super(in, format);
termBytes = in.readBytesRef();
}
@Override
protected void writeTermTo(StreamOutput out) throws IOException {
out.writeBytesRef(termBytes);
}
@Override
public Object getKey() {
return getKeyAsString();
}
// this method is needed for scripted numeric aggs
@Override
public Number getKeyAsNumber() {
/*
* If the term is a long greater than 2^52 then parsing as a double would lose accuracy. Therefore, we first parse as a long and
* if this fails then we attempt to parse the term as a double.
*/
try {
return Long.parseLong(termBytes.utf8ToString());
} catch (final NumberFormatException ignored) {
return Double.parseDouble(termBytes.utf8ToString());
}
}
@Override
public String getKeyAsString() {
return format.format(termBytes).toString();
}
@Override
public int compareKey(Bucket other) {
return termBytes.compareTo(other.termBytes);
}
@Override
protected final XContentBuilder keyToXContent(XContentBuilder builder) throws IOException {
return builder.field(CommonFields.KEY.getPreferredName(), getKeyAsString());
}
@Override
public boolean equals(Object obj) {
return super.equals(obj) && Objects.equals(termBytes, ((Bucket) obj).termBytes);
}
@Override
public int hashCode() {
return Objects.hash(super.hashCode(), termBytes);
}
}
StringRareTerms(String name, BucketOrder order, List pipelineAggregators,
Map metaData, DocValueFormat format,
List buckets, long maxDocCount, SetBackedScalingCuckooFilter filter) {
super(name, order, pipelineAggregators, metaData, format, buckets, maxDocCount, filter);
}
/**
* Read from a stream.
*/
public StringRareTerms(StreamInput in) throws IOException {
super(in, StringRareTerms.Bucket::new);
}
@Override
public String getWriteableName() {
return NAME;
}
@Override
public StringRareTerms create(List buckets) {
return new StringRareTerms(name, order, pipelineAggregators(), metaData, format, buckets, maxDocCount, filter);
}
@Override
public StringRareTerms.Bucket createBucket(InternalAggregations aggregations, StringRareTerms.Bucket prototype) {
return new StringRareTerms.Bucket(prototype.termBytes, prototype.getDocCount(), aggregations, prototype.format);
}
@Override
protected StringRareTerms createWithFilter(String name, List buckets,
SetBackedScalingCuckooFilter filterFilter) {
return new StringRareTerms(name, order, pipelineAggregators(), metaData, format,
buckets, maxDocCount, filterFilter);
}
@Override
protected StringRareTerms.Bucket[] createBucketsArray(int size) {
return new StringRareTerms.Bucket[size];
}
@Override
public boolean containsTerm(SetBackedScalingCuckooFilter filter, StringRareTerms.Bucket bucket) {
return filter.mightContain(bucket.termBytes);
}
@Override
public void addToFilter(SetBackedScalingCuckooFilter filter, StringRareTerms.Bucket bucket) {
filter.add(bucket.termBytes);
}
@Override
Bucket createBucket(long docCount, InternalAggregations aggs, StringRareTerms.Bucket prototype) {
return new Bucket(prototype.termBytes, docCount, aggs, format);
}
}