com.opencredo.concourse.domain.events.selection.EventSelection Maven / Gradle / Ivy

Go to download
package com.opencredo.concourse.domain.events.selection;

import com.opencredo.concourse.domain.common.AggregateId;
import com.opencredo.concourse.domain.events.Event;
import com.opencredo.concourse.domain.events.EventType;
import com.opencredo.concourse.domain.events.matching.EventTypeMatcher;
import com.opencredo.concourse.domain.time.TimeRange;

import java.util.*;
import java.util.Map.Entry;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.stream.Collectors;

/**
 * Utility class providing methods for selecting events matching various criteria.
 */
public final class EventSelection {

    private EventSelection() {
    }

    public static Function, List> filterBy(Predicate predicate) {
        return events -> events.stream().filter(predicate).collect(Collectors.toCollection(LinkedList::new));
    }

    public static Predicate inRange(TimeRange timeRange) {
        return event -> timeRange.contains(event.getEventTimestamp().getTimestamp());
    }

    public static Predicate matchedBy(EventTypeMatcher matcher) {
        return event -> matcher.match(EventType.of(event)).isPresent();
    }

    public static List selectEvents(Map> events, Predicate filter, AggregateId aggregateId) {
        return Optional.ofNullable(events.get(aggregateId))
                .map(filterBy(filter))
                .orElseGet(Collections::emptyList);
    }

    public static Map> selectEvents(Map> events, Predicate filter, String aggregateType, Collection aggregateIds) {
        Set aggregateIdSet = aggregateIds.stream()
                .map(id -> AggregateId.of(aggregateType, id))
                .collect(Collectors.toSet());

        return events.entrySet().stream()
                .filter(e -> aggregateIdSet.contains(e.getKey()))
                .collect(Collectors.toMap(
                        Entry::getKey,
                        filterBy(filter).compose(Entry::getValue)));
    }
}