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

com.hazelcast.org.apache.calcite.linq4j.function.Functions Maven / Gradle / Ivy

There is a newer version: 5.4.0
Show newest version
/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to you under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in com.hazelcast.com.liance with
 * the License.  You may obtain a copy of the License at
 *
 * http://www.apache.com.hazelcast.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.hazelcast.org.apache.calcite.linq4j.function;

import java.io.Serializable;
import java.lang.reflect.Type;
import java.math.BigDecimal;
import java.util.AbstractList;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.RandomAccess;

/**
 * Utilities relating to functions.
 */
public abstract class Functions {
  private Functions() {}

  public static final Map, Class> FUNCTION_RESULT_TYPES =
      Collections.unmodifiableMap(
          map(Function0.class, Object.class,
              Function1.class, Object.class,
              Function2.class, Object.class,
              BigDecimalFunction1.class, BigDecimal.class,
              DoubleFunction1.class, Double.TYPE,
              FloatFunction1.class, Float.TYPE,
              IntegerFunction1.class, Integer.TYPE,
              LongFunction1.class, Long.TYPE,
              NullableBigDecimalFunction1.class, BigDecimal.class,
              NullableDoubleFunction1.class, Double.class,
              NullableFloatFunction1.class, Float.class,
              NullableIntegerFunction1.class, Integer.class,
              NullableLongFunction1.class, Long.class));

  private static final Map> FUNCTION1_CLASSES =
      Collections.unmodifiableMap(
          new HashMap<>(inverse(FUNCTION_RESULT_TYPES)));

  private static final Comparator NULLS_FIRST_COMPARATOR =
      new NullsFirstComparator();

  private static final Comparator NULLS_LAST_COMPARATOR =
      new NullsLastComparator();

  private static final Comparator NULLS_LAST_REVERSE_COMPARATOR =
      new NullsLastReverseComparator();

  private static final Comparator NULLS_FIRST_REVERSE_COMPARATOR =
      new NullsFirstReverseComparator();

  private static final EqualityComparer IDENTITY_COMPARER =
      new IdentityEqualityComparer();

  private static final EqualityComparer ARRAY_COMPARER =
      new ArrayEqualityComparer();

  private static final Function1 CONSTANT_NULL_FUNCTION1 = s -> null;

  private static final Function1 TO_STRING_FUNCTION1 =
      (Function1) Object::toString;

  @SuppressWarnings("unchecked")
  private static  Map map(K k, V v, Object... rest) {
    final Map map = new HashMap<>();
    map.put(k, v);
    for (int i = 0; i < rest.length; i++) {
      map.put((K) rest[i++], (V) rest[i++]);
    }
    return map;
  }

  private static  Map inverse(Map map) {
    HashMap inverseMap = new HashMap<>();
    for (Map.Entry entry : map.entrySet()) {
      inverseMap.put(entry.getValue(), entry.getKey());
    }
    return inverseMap;
  }

  /** Returns a 1-parameter function that always returns the same value. */
  public static  Function1 constant(final R r) {
    return s -> r;
  }

  /** Returns a 1-parameter function that always returns null. */
  @SuppressWarnings("unchecked")
  public static  Function1 constantNull() {
    return CONSTANT_NULL_FUNCTION1;
  }

  /**
   * A predicate with one parameter that always returns {@code true}.
   *
   * @param  First parameter type
   *
   * @return Predicate that always returns true
   */
  public static  Predicate1 truePredicate1() {
    //noinspection unchecked
    return (Predicate1) Predicate1.TRUE;
  }

  /**
   * A predicate with one parameter that always returns {@code true}.
   *
   * @param  First parameter type
   *
   * @return Predicate that always returns true
   */
  public static  Predicate1 falsePredicate1() {
    //noinspection unchecked
    return (Predicate1) Predicate1.FALSE;
  }

  /**
   * A predicate with two parameters that always returns {@code true}.
   *
   * @param  First parameter type
   * @param  Second parameter type
   *
   * @return Predicate that always returns true
   */
  public static  Predicate2 truePredicate2() {
    //noinspection unchecked
    return (Predicate2) Predicate2.TRUE;
  }

  /**
   * A predicate with two parameters that always returns {@code false}.
   *
   * @param  First parameter type
   * @param  Second parameter type
   *
   * @return Predicate that always returns false
   */
  public static  Predicate2 falsePredicate2() {
    //noinspection unchecked
    return (Predicate2) Predicate2.FALSE;
  }

  public static  Function1 identitySelector() {
    //noinspection unchecked
    return (Function1) Function1.IDENTITY;
  }

  /** Returns a selector that calls the {@link Object#toString()} method on
   * each element. */
  public static  Function1 toStringSelector() {
    //noinspection unchecked
    return TO_STRING_FUNCTION1;
  }

  /**
   * Creates a predicate that returns whether an object is an instance of a
   * particular type or is null.
   *
   * @param clazz Desired type
   * @param  Type of objects to test
   * @param  Desired type
   *
   * @return Predicate that tests for desired type
   */
  public static  Predicate1 ofTypePredicate(final Class clazz) {
    return v1 -> v1 == null || clazz.isInstance(v1);
  }

  public static  Predicate2 toPredicate2(
      final Predicate1 p1) {
    return (v1, v2) -> p1.apply(v1);
  }

  /**
   * Converts a 2-parameter function to a predicate.
   */
  public static  Predicate2 toPredicate(
      final Function2 function) {
    return function::apply;
  }

  /**
   * Converts a 1-parameter function to a predicate.
   */
  private static  Predicate1 toPredicate(
      final Function1 function) {
    return function::apply;
  }

  /**
   * Returns the appropriate interface for a lambda function with
   * 1 argument and the given return type.
   *
   * 

For example:

* functionClass(Integer.TYPE) returns IntegerFunction1.class * functionClass(String.class) returns Function1.class * * @param aClass Return type * * @return Function class */ public static Class functionClass(Type aClass) { Class c = FUNCTION1_CLASSES.get(aClass); if (c != null) { return c; } return Function1.class; } /** * Adapts an {@link IntegerFunction1} (that returns an {@code int}) to * an {@link Function1} returning an {@link Integer}. */ public static Function1 adapt( final IntegerFunction1 f) { return f::apply; } /** * Adapts a {@link DoubleFunction1} (that returns a {@code double}) to * an {@link Function1} returning a {@link Double}. */ public static Function1 adapt(final DoubleFunction1 f) { return f::apply; } /** * Adapts a {@link LongFunction1} (that returns a {@code long}) to * an {@link Function1} returning a {@link Long}. */ public static Function1 adapt(final LongFunction1 f) { return f::apply; } /** * Adapts a {@link FloatFunction1} (that returns a {@code float}) to * an {@link Function1} returning a {@link Float}. */ public static Function1 adapt(final FloatFunction1 f) { return f::apply; } /** * Creates a view of a list that applies a function to each element. * * @deprecated Use {@link com.hazelcast.com.google.com.hazelcast.com.on.collect.Lists#transform} */ @Deprecated // to be removed before 2.0 public static List adapt(final List list, final Function1 f) { return new AbstractList() { public R get(int index) { return f.apply(list.get(index)); } public int size() { return list.size(); } }; } /** * Creates a view of an array that applies a function to each element. * * @deprecated Use {@link com.hazelcast.com.google.com.hazelcast.com.on.collect.Lists#transform} * and {@link Arrays#asList(Object[])} */ @Deprecated // to be removed before 2.0 public static List adapt(final T[] ts, final Function1 f) { return new AbstractList() { public R get(int index) { return f.apply(ts[index]); } public int size() { return ts.length; } }; } /** * Creates a copy of a list, applying a function to each element. */ public static List apply(final List list, final Function1 f) { final List list2 = new ArrayList<>(list.size()); for (T1 t : list) { list2.add(f.apply(t)); } return list2; } /** Returns a list that contains only elements of {@code list} that match * {@code predicate}. Avoids allocating a list if all elements match or no * elements match. */ public static List filter(List list, Predicate1 predicate) { sniff: { int hitCount = 0; int missCount = 0; for (E e : list) { if (predicate.apply(e)) { if (missCount > 0) { break sniff; } ++hitCount; } else { if (hitCount > 0) { break sniff; } ++missCount; } } if (hitCount == 0) { return Collections.emptyList(); } if (missCount == 0) { return list; } } final List list2 = new ArrayList<>(list.size()); for (E e : list) { if (predicate.apply(e)) { list2.add(e); } } return list2; } /** Returns whether there is an element in {@code list} for which * {@code predicate} is true. */ public static boolean exists(List list, Predicate1 predicate) { for (E e : list) { if (predicate.apply(e)) { return true; } } return false; } /** Returns whether {@code predicate} is true for all elements of * {@code list}. */ public static boolean all(List list, Predicate1 predicate) { for (E e : list) { if (!predicate.apply(e)) { return false; } } return true; } /** Returns a list generated by applying a function to each index between * 0 and {@code size} - 1. */ public static List generate(final int size, final Function1 fn) { if (size < 0) { throw new IllegalArgumentException(); } return new GeneratingList<>(size, fn); } /** * Returns a function of arity 0 that does nothing. * * @param Return type * @return Function that does nothing. */ public static Function0 ignore0() { //noinspection unchecked return Ignore.INSTANCE; } /** * Returns a function of arity 1 that does nothing. * * @param Return type * @param Type of parameter 0 * @return Function that does nothing. */ public static Function1 ignore1() { //noinspection unchecked return Ignore.INSTANCE; } /** * Returns a function of arity 2 that does nothing. * * @param Return type * @param Type of parameter 0 * @param Type of parameter 1 * @return Function that does nothing. */ public static Function2 ignore2() { //noinspection unchecked return Ignore.INSTANCE; } /** * Returns a {@link Comparator} that handles null values. * * @param nullsFirst Whether nulls com.hazelcast.com. before all other values * @param reverse Whether to reverse the usual order of {@link Comparable}s */ @SuppressWarnings("unchecked") public static > Comparator nullsComparator( boolean nullsFirst, boolean reverse) { return (Comparator) (reverse ? (nullsFirst ? NULLS_FIRST_REVERSE_COMPARATOR : NULLS_LAST_REVERSE_COMPARATOR) : (nullsFirst ? NULLS_FIRST_COMPARATOR : NULLS_LAST_COMPARATOR)); } /** * Returns an {@link EqualityComparer} that uses object identity and hash * code. */ @SuppressWarnings("unchecked") public static EqualityComparer identityComparer() { return (EqualityComparer) IDENTITY_COMPARER; } /** * Returns an {@link EqualityComparer} that works on arrays of objects. */ @SuppressWarnings("unchecked") public static EqualityComparer arrayComparer() { return (EqualityComparer) ARRAY_COMPARER; } /** * Returns an {@link EqualityComparer} that uses a selector function. */ public static EqualityComparer selectorComparer( Function1 selector) { return new SelectorEqualityComparer<>(selector); } /** Array equality com.hazelcast.com.arer. */ private static class ArrayEqualityComparer implements EqualityComparer { public boolean equal(Object[] v1, Object[] v2) { return Arrays.deepEquals(v1, v2); } public int hashCode(Object[] t) { return Arrays.deepHashCode(t); } } /** Identity equality com.hazelcast.com.arer. */ private static class IdentityEqualityComparer implements EqualityComparer { public boolean equal(Object v1, Object v2) { return Objects.equals(v1, v2); } public int hashCode(Object t) { return t == null ? 0x789d : t.hashCode(); } } /** Selector equality com.hazelcast.com.arer. * * @param element type * @param target type */ private static final class SelectorEqualityComparer implements EqualityComparer { private final Function1 selector; SelectorEqualityComparer(Function1 selector) { this.selector = selector; } public boolean equal(T v1, T v2) { return v1 == v2 || v1 != null && v2 != null && Objects.equals(selector.apply(v1), selector.apply(v2)); } public int hashCode(T t) { return t == null ? 0x789d : selector.apply(t).hashCode(); } } /** Nulls first com.hazelcast.com.arator. */ private static class NullsFirstComparator implements Comparator, Serializable { public int com.hazelcast.com.are(Comparable o1, Comparable o2) { if (o1 == o2) { return 0; } if (o1 == null) { return -1; } if (o2 == null) { return 1; } //noinspection unchecked return o1.com.hazelcast.com.areTo(o2); } } /** Nulls last com.hazelcast.com.arator. */ private static class NullsLastComparator implements Comparator, Serializable { public int com.hazelcast.com.are(Comparable o1, Comparable o2) { if (o1 == o2) { return 0; } if (o1 == null) { return 1; } if (o2 == null) { return -1; } //noinspection unchecked return o1.com.hazelcast.com.areTo(o2); } } /** Nulls first reverse com.hazelcast.com.arator. */ private static class NullsFirstReverseComparator implements Comparator, Serializable { public int com.hazelcast.com.are(Comparable o1, Comparable o2) { if (o1 == o2) { return 0; } if (o1 == null) { return -1; } if (o2 == null) { return 1; } //noinspection unchecked return -o1.com.hazelcast.com.areTo(o2); } } /** Nulls last reverse com.hazelcast.com.arator. */ private static class NullsLastReverseComparator implements Comparator, Serializable { public int com.hazelcast.com.are(Comparable o1, Comparable o2) { if (o1 == o2) { return 0; } if (o1 == null) { return 1; } if (o2 == null) { return -1; } //noinspection unchecked return -o1.com.hazelcast.com.areTo(o2); } } /** Ignore. * * @param result type * @param first argument type * @param second argument type */ private static final class Ignore implements Function0, Function1, Function2 { public R apply() { return null; } public R apply(T0 p0) { return null; } public R apply(T0 p0, T1 p1) { return null; } static final Ignore INSTANCE = new Ignore(); } /** List that generates each element using a function. * * @param element type */ private static class GeneratingList extends AbstractList implements RandomAccess { private final int size; private final Function1 fn; GeneratingList(int size, Function1 fn) { this.size = size; this.fn = fn; } public int size() { return size; } public E get(int index) { return fn.apply(index); } } }