All Downloads are FREE. Search and download functionalities are using the official Maven repository.

io.druid.sql.calcite.filtration.Bounds Maven / Gradle / Ivy

There is a newer version: 0.12.3
Show newest version
/*
 * Licensed to Metamarkets Group Inc. (Metamarkets) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. Metamarkets 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 io.druid.sql.calcite.filtration;

import com.google.common.base.Function;
import com.google.common.collect.BoundType;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Lists;
import com.google.common.collect.Range;
import io.druid.java.util.common.ISE;
import io.druid.query.filter.BoundDimFilter;
import io.druid.query.ordering.StringComparators;
import org.joda.time.Interval;

import java.util.List;

public class Bounds
{
  /**
   * Negates single-ended Bound filters.
   *
   * @param bound filter
   *
   * @return negated filter, or null if this bound is double-ended.
   */
  public static BoundDimFilter not(final BoundDimFilter bound)
  {
    if (bound.getUpper() != null && bound.getLower() != null) {
      return null;
    } else if (bound.getUpper() != null) {
      return new BoundDimFilter(
          bound.getDimension(),
          bound.getUpper(),
          null,
          !bound.isUpperStrict(),
          false,
          null,
          bound.getExtractionFn(),
          bound.getOrdering()
      );
    } else {
      // bound.getLower() != null
      return new BoundDimFilter(
          bound.getDimension(),
          null,
          bound.getLower(),
          false,
          !bound.isLowerStrict(),
          null,
          bound.getExtractionFn(),
          bound.getOrdering()
      );
    }
  }

  public static Range toRange(final BoundDimFilter bound)
  {
    final BoundValue upper = bound.getUpper() != null ? new BoundValue(bound.getUpper(), bound.getOrdering()) : null;
    final BoundValue lower = bound.getLower() != null ? new BoundValue(bound.getLower(), bound.getOrdering()) : null;

    if (lower == null) {
      return bound.isUpperStrict() ? Range.lessThan(upper) : Range.atMost(upper);
    } else if (upper == null) {
      return bound.isLowerStrict() ? Range.greaterThan(lower) : Range.atLeast(lower);
    } else {
      return Range.range(
          lower, bound.isLowerStrict() ? BoundType.OPEN : BoundType.CLOSED,
          upper, bound.isUpperStrict() ? BoundType.OPEN : BoundType.CLOSED
      );
    }
  }

  public static List> toRanges(final List bounds)
  {
    return ImmutableList.copyOf(
        Lists.transform(
            bounds,
            new Function>()
            {
              @Override
              public Range apply(BoundDimFilter bound)
              {
                return toRange(bound);
              }
            }
        )
    );
  }

  public static BoundDimFilter toFilter(final BoundRefKey boundRefKey, final Range range)
  {
    return new BoundDimFilter(
        boundRefKey.getDimension(),
        range.hasLowerBound() ? range.lowerEndpoint().getValue() : null,
        range.hasUpperBound() ? range.upperEndpoint().getValue() : null,
        range.hasLowerBound() && range.lowerBoundType() == BoundType.OPEN,
        range.hasUpperBound() && range.upperBoundType() == BoundType.OPEN,
        null,
        boundRefKey.getExtractionFn(),
        boundRefKey.getComparator()
    );
  }

  public static BoundDimFilter equalTo(final BoundRefKey boundRefKey, final String value)
  {
    return new BoundDimFilter(
        boundRefKey.getDimension(),
        value,
        value,
        false,
        false,
        null,
        boundRefKey.getExtractionFn(),
        boundRefKey.getComparator()
    );
  }

  public static BoundDimFilter greaterThan(final BoundRefKey boundRefKey, final String value)
  {
    return new BoundDimFilter(
        boundRefKey.getDimension(),
        value,
        null,
        true,
        false,
        null,
        boundRefKey.getExtractionFn(),
        boundRefKey.getComparator()
    );
  }

  public static BoundDimFilter greaterThanOrEqualTo(final BoundRefKey boundRefKey, final String value)
  {
    return new BoundDimFilter(
        boundRefKey.getDimension(),
        value,
        null,
        false,
        false,
        null,
        boundRefKey.getExtractionFn(),
        boundRefKey.getComparator()
    );
  }

  public static BoundDimFilter lessThan(final BoundRefKey boundRefKey, final String value)
  {
    return new BoundDimFilter(
        boundRefKey.getDimension(),
        null,
        value,
        false,
        true,
        null,
        boundRefKey.getExtractionFn(),
        boundRefKey.getComparator()
    );
  }

  public static BoundDimFilter lessThanOrEqualTo(final BoundRefKey boundRefKey, final String value)
  {
    return new BoundDimFilter(
        boundRefKey.getDimension(),
        null,
        value,
        false,
        false,
        null,
        boundRefKey.getExtractionFn(),
        boundRefKey.getComparator()
    );
  }

  public static BoundDimFilter interval(final BoundRefKey boundRefKey, final Interval interval)
  {
    if (!boundRefKey.getComparator().equals(StringComparators.NUMERIC)) {
      // Interval comparison only works with NUMERIC comparator.
      throw new ISE("Comparator must be NUMERIC but was[%s]", boundRefKey.getComparator());
    }

    return new BoundDimFilter(
        boundRefKey.getDimension(),
        String.valueOf(interval.getStartMillis()),
        String.valueOf(interval.getEndMillis()),
        false,
        true,
        null,
        boundRefKey.getExtractionFn(),
        boundRefKey.getComparator()
    );
  }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy