website.automate.jwebrobot.executor.action.ActionExecutorFactory Maven / Gradle / Ivy

Go to download
package website.automate.jwebrobot.executor.action;

import website.automate.jwebrobot.exceptions.ActionExecutorMissingException;
import website.automate.waml.io.model.action.Action;

import javax.inject.Inject;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

public class ActionExecutorFactory {

    private final Map, ActionExecutor> executorRegistry = new HashMap<>();

    private final Set> executors;

    @Inject
    public ActionExecutorFactory(Set> executors) {
        this.executors = executors;

        init();
    }

    private void init() {
        for (ActionExecutor executor : executors) {
            executorRegistry.put(executor.getSupportedType(), executor);
        }
    }

    @SuppressWarnings("unchecked")
    public ActionExecutor getInstance(Class actionClazz) {
        ActionExecutor actionExecutor = executorRegistry.get(actionClazz);

        if (actionExecutor == null) {
            throw new ActionExecutorMissingException(actionClazz);
        }

        return (ActionExecutor)actionExecutor;
    }
}