aQute.bnd.memoize.MemoizingSupplier Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of biz.aQute.bnd.util Show documentation

Util classes

The newest version!

package aQute.bnd.memoize;

import static java.util.Objects.requireNonNull;

import java.util.function.Predicate;
import java.util.function.Supplier;

/**
 * The object can exist in one of two states:
 * 
 * initial which means the source supplier has not been called or has not
 * returned a value acceptable to the predicate and memoized holds the wrapped
 * supplier. From this state, the object can transition to open.
 * open which means memoized is the value from the source supplier. This is
 * the terminal state.
 * 
 */
class MemoizingSupplier implements Memoize {
	private volatile boolean	initial;
	// @GuardedBy("initial")
	private Object				memoized;

	MemoizingSupplier(Supplier supplier) {
		this(supplier, supplied -> true);
	}

	MemoizingSupplier(Supplier supplier, Predicate predicate) {
		requireNonNull(supplier);
		requireNonNull(predicate);
		memoized = (Supplier) () -> {
			T supplied = supplier.get();
			if (predicate.test(supplied)) {
				memoized = supplied;
				// write initial _after_ write memoized
				initial = false;
			}
			return supplied;
		};
		// write initial _after_ write memoized
		initial = true;
	}

	@SuppressWarnings("unchecked")
	@Override
	public T get() {
		// read initial _before_ read memoized
		if (initial) {
			// critical section: only one resolver at a time
			synchronized (this) {
				if (initial) {
					return ((Supplier) memoized).get();
				}
			}
		}
		return (T) memoized;
	}

	@SuppressWarnings("unchecked")
	@Override
	public T peek() {
		// read initial _before_ read memoized
		if (initial) {
			return null;
		}
		return (T) memoized;
	}

	@Override
	public boolean isPresent() {
		return !initial;
	}

	@Override
	public String toString() {
		// read initial _before_ read memoized
		return initial ? "" : String.valueOf(memoized);
	}
}