com.bull.aurocontrol.csst.poc.index.interval.InNumericIntervalQuery.java Source code

Introduction

Here is the source code for com.bull.aurocontrol.csst.poc.index.interval.InNumericIntervalQuery.java

Source

/*
 * Copyright 2010 The Lucene Interval Field Authors.
 *
 * Licensed 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 com.bull.aurocontrol.csst.poc.index.interval;

import org.apache.lucene.analysis.NumericTokenStream;
import org.apache.lucene.analysis.TokenStream;
import org.apache.lucene.analysis.tokenattributes.TermAttribute;
import org.apache.lucene.index.Term;
import org.apache.lucene.search.BooleanClause;
import org.apache.lucene.search.BooleanQuery;
import org.apache.lucene.search.TermQuery;

import java.io.IOException;

/**
 * Query that finds intervals containing a given value.
 */
public final class InNumericIntervalQuery extends BooleanQuery {
    /**
     * The search value.
     */
    private final long value;

    /**
     * Creates a query to find intervals a number is in.
     * @param name The name of the field to search.
     * @param value The value to find containing intervals.
     */
    public InNumericIntervalQuery(final String name, final long value) {
        this(name, value, NumericIntervalField.DEFAULT_PRECISION_STEP);
    }

    /**
     * Creates a query to find intervals a number is in.
     * @param name The name of the field to search.
     * @param value The search value.
     * @param precisionStep The precision step used when indexing the field.
     */
    public InNumericIntervalQuery(final String name, final long value, final int precisionStep) {
        super(true);
        this.value = value;

        TokenStream stream = new NumericTokenStream(precisionStep).setLongValue(value);

        try {
            stream.reset();
            while (stream.incrementToken()) {
                this.add(new TermQuery(new Term(name, stream.getAttribute(TermAttribute.class).term())),
                        BooleanClause.Occur.SHOULD);
            }
        } catch (IOException e) {
            throw new IllegalStateException("This should never happen - NumericTokenStream does no IO.");
        }
    }

    @Override
    public String toString(final String field) {
        return String.format("inInterval(%d, %s)", value, field);
    }

    @Override
    public int hashCode() {
        return (int) value;
    }

    @Override
    public boolean equals(final Object o) {
        return o != null && o.getClass() == this.getClass() && ((InNumericIntervalQuery) o).value == value;
    }
}