TestExpressionCalculatorSelector.java

package org.jtwig.render.expression.test;

import com.google.common.base.Optional;
import org.jtwig.model.expression.test.TestExpression;
import org.jtwig.render.expression.test.calculator.TestExpressionCalculator;

import java.util.Map;

public class TestExpressionCalculatorSelector {
    private final Map<Class<? extends TestExpression>, TestExpressionCalculator> calculatorMap;

    public TestExpressionCalculatorSelector(Map<Class<? extends TestExpression>, TestExpressionCalculator> calculatorMap) {
        this.calculatorMap = calculatorMap;
    }

    public Optional<TestExpressionCalculator> calculatorFor (TestExpression expression) {
        return Optional.fromNullable(calculatorMap.get(expression.getClass()));
    }
}