io.github.amayaframework.options.ClosedOptionSet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of amaya-options Show documentation

Amaya Options Module

The newest version!

package io.github.amayaframework.options;

import com.github.romanqed.jfunc.Runnable1;

import java.util.*;
import java.util.function.Supplier;

/**
 * The implementation of {@link OptionSet} that accepts keys only from limited key set.
 */
public class ClosedOptionSet extends AbstractOptionSet {
    private final Set keys;

    /**
     * Constructs {@link ClosedOptionSet} instance with the map given by supplier and the specified key set.
     *
     * @param supplier supplier providing key-value map instance
     * @param keys     the specified key set
     */
    public ClosedOptionSet(Supplier> supplier, Set keys) {
        super(Objects.requireNonNull(supplier.get()));
        this.keys = Collections.unmodifiableSet(keys);
    }

    /**
     * Constructs {@link ClosedOptionSet} with {@link HashMap} and the specified key set.
     *
     * @param keys the specified key set
     */
    public ClosedOptionSet(Set keys) {
        super(new HashMap<>());
        this.keys = Collections.unmodifiableSet(keys);
    }

    @Override
    public Object set(String key, Object value) {
        Objects.requireNonNull(key);
        if (!keys.contains(key)) {
            throw new IllegalKeyException(key, keys);
        }
        return body.put(key, value);
    }

    @Override
    public Object remove(String key) {
        return body.remove(key);
    }

    @Override
    public Set getKeys() {
        return keys;
    }

    @Override
    public void forEach(Runnable1 action) {
        try {
            for (var key : keys) {
                action.run(key);
            }
        } catch (Error | RuntimeException e) {
            throw e;
        } catch (Throwable e) {
            throw new RuntimeException(e);
        }
    }
}