com.speedment.runtime.compute.internal.ToIntNullableImpl Maven / Gradle / Ivy

Go to download
/*
 *
 * Copyright (c) 2006-2020, Speedment, Inc. All Rights Reserved.
 *
 * 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.speedment.runtime.compute.internal;

import com.speedment.runtime.compute.ToDoubleNullable;
import com.speedment.runtime.compute.ToInt;
import com.speedment.runtime.compute.ToIntNullable;
import com.speedment.runtime.compute.expression.NullableExpression;

import java.util.Objects;
import java.util.function.IntToDoubleFunction;
import java.util.function.IntUnaryOperator;
import java.util.function.Predicate;

import static java.util.Objects.requireNonNull;

/**
 * @author Emil Forslund
 * @since  3.1.0
 */
public final class ToIntNullableImpl
    implements NullableExpression>, ToIntNullable {

    private final ToInt original;
    private final Predicate isNull;

    public ToIntNullableImpl(ToInt original, Predicate isNull) {
        this.original = requireNonNull(original);
        this.isNull   = requireNonNull(isNull);
    }

    @Override
    public ToInt inner() {
        return original;
    }

    @Override
    public Predicate isNullPredicate() {
        return isNull;
    }

    @Override
    public Integer apply(T t) {
        return isNull.test(t) ? null : original.applyAsInt(t);
    }

    @Override
    public int applyAsInt(T t) {
        return original.applyAsInt(t);
    }

    @Override
    public ToInt orThrow() {
        return original;
    }

    @Override
    public ToInt orElseGet(ToInt getter) {
        return t -> isNull.test(t)
            ? getter.applyAsInt(t)
            : original.applyAsInt(t);
    }

    @Override
    public ToInt orElse(Integer value) {
        return t -> isNull.test(t) ? value : original.applyAsInt(t);
    }

    @Override
    public ToDoubleNullable mapToDoubleIfPresent(IntToDoubleFunction mapper) {
        return t -> isNull.test(t) ? null
            : mapper.applyAsDouble(original.applyAsInt(t));
    }

    @Override
    public ToIntNullable mapIfPresent(IntUnaryOperator mapper) {
        return t -> isNull.test(t) ? null
            : mapper.applyAsInt(original.applyAsInt(t));
    }

    @Override
    public long hash(T object) {
        return isNull.test(object)
            ? (0xffffffffL + 1) // Will never occur naturally
            : original.applyAsInt(object);
    }

    @Override
    public int compare(T first, T second) {
        final boolean f = isNull(first);
        final boolean s = isNull(second);
        if (f && s) return 0;
        else if (f) return 1;
        else if (s) return -1;
        else return Integer.compare(
            original.applyAsInt(first),
            original.applyAsInt(second)
        );
    }

    @Override
    public boolean isNull(T object) {
        return isNull.test(object);
    }

    @Override
    public boolean isNotNull(T object) {
        return !isNull.test(object);
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        else if (!(o instanceof NullableExpression)) return false;
        final NullableExpression that = (NullableExpression) o;
        return Objects.equals(original, that.inner()) &&
            Objects.equals(isNull, that.isNullPredicate());
    }

    @Override
    public int hashCode() {
        return Objects.hash(original, isNull);
    }
}