org.aksw.commons.collections.trees.ReclaimingSupplier Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of aksw-commons-collections Show documentation

There is a newer version: 0.9.9

Show newest version

package org.aksw.commons.collections.trees;

import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;
import java.util.function.Supplier;

public class ReclaimingSupplier implements Supplier {
    protected Supplier generator;
    protected Set reclaims = new TreeSet();

    public ReclaimingSupplier(Supplier generator) {
        super();
        this.generator = generator;
    }

    public void reclaim(T item) {
        reclaims.add(item);
    }

    @Override
    public T get() {
        T result;
        Iterator it = reclaims.iterator();
        if (it.hasNext()) {
            result = it.next();
            it.remove();
        } else {
            result = generator.get();
        }
        return result;
    }
}