wtf.metio.memoization.jool.Function1Memoizer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of memoization-jool Show documentation

The newest version!

/*
 * SPDX-FileCopyrightText: The memoization.java Authors
 * SPDX-License-Identifier: 0BSD
 */

package wtf.metio.memoization.jool;

import org.jooq.lambda.function.Function1;
import wtf.metio.memoization.core.AbstractMemoizer;

import java.util.concurrent.ConcurrentMap;
import java.util.function.Function;

import static java.util.Objects.requireNonNull;

final class Function1Memoizer
        extends AbstractMemoizer
        implements Function1 {

    private final Function keyFunction;
    private final Function1 function;

    Function1Memoizer(
            final ConcurrentMap cache,
            final Function keyFunction,
            final Function1 function) {
        super(cache);
        this.keyFunction = requireNonNull(keyFunction, "Provide a key function.");
        this.function = requireNonNull(function,
                "Cannot memoize a NULL Function1 - provide an actual Function1 to fix this.");
    }

    @Override
    public OUTPUT apply(final INPUT input) {
        return computeIfAbsent(keyFunction.apply(input), key -> function.apply(input));
    }

}