# AfterUpdateSupport.java

    package org.codefilarete.stalactite.engine.cascade;

    import java.util.function.BiConsumer;
    import java.util.function.Function;
    import java.util.function.Predicate;
    import java.util.stream.Collectors;

    import org.codefilarete.tool.Duo;
    import org.codefilarete.tool.collection.Iterables;
    import org.codefilarete.tool.function.Predicates;
    import org.codefilarete.stalactite.engine.listener.UpdateListener;

    /**
     * @author Guillaume Mary
     */
    public class AfterUpdateSupport<TRIGGER, TARGET> implements UpdateListener<TRIGGER> {

    	private final BiConsumer<Iterable<Duo<TARGET, TARGET>>, Boolean> afterUpdateAction;

    	private final Function<TRIGGER, TARGET> targetProvider;

    	private final Predicate<Duo<TARGET, TARGET>> targetFilter;

    	public AfterUpdateSupport(BiConsumer<Iterable<Duo<TARGET, TARGET>>, Boolean> afterUpdateAction, Function<TRIGGER, TARGET> targetProvider) {
    		this(afterUpdateAction, targetProvider, Predicates.acceptAll());
    	}

    	public AfterUpdateSupport(BiConsumer<Iterable<Duo<TARGET, TARGET>>, Boolean> afterUpdateAction, Function<TRIGGER, TARGET> targetProvider, Predicate<Duo<TARGET, TARGET>> targetFilter) {
    		this.afterUpdateAction = afterUpdateAction;
    		this.targetProvider = targetProvider;
    		this.targetFilter = targetFilter;
    	}

    	@Override
    	public void afterUpdate(Iterable<? extends Duo<TRIGGER, TRIGGER>> entities, boolean allColumnsStatement) {
    		afterUpdateAction.accept(Iterables.stream(entities)
    						.map(e -> new Duo<>(targetProvider.apply(e.getLeft()), targetProvider.apply(e.getRight())))
    						.filter(targetFilter).collect(Collectors.toList()),
    				allColumnsStatement);
    	}
    }