JobMutexGroups.java

package de.otto.edison.jobs.service;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.HashSet;
import java.util.Set;

import static java.util.Collections.emptySet;

/**
 * Component used to determine the mutually exclusive job types for a given type.
 *
 * @since 1.0.0
 */
@Component
public class JobMutexGroups {

    private Set<JobMutexGroup> mutexGroups = emptySet();

    /**
     * Returns the set of configured JobMutextGroups.
     *
     * @return set of mutex groups
     */
    public Set<JobMutexGroup> getMutexGroups() {
        return mutexGroups;
    }

    /**
     * Returns the set of mutually exclusive job types for a given type.
     * <p>
     * Because every job type is mutually exclusive with itself, the returned set will at least contain the
     * type provided as parameter.
     * </p>
     * @param jobType the given job type
     * @return set of mutually exclusive job types
     */
    public Set<String> mutexJobTypesFor(final String jobType) {
        final Set<String> result = new HashSet<>();
        this.mutexGroups
                .stream()
                .map(JobMutexGroup::getJobTypes)
                .filter(g -> g.contains(jobType))
                .forEach(result::addAll);
        result.remove(jobType);
        return result;
    }

    @Autowired(required = false)
    public JobMutexGroups setMutexGroups(Set<JobMutexGroup> mutexGroups) {
        this.mutexGroups = mutexGroups;
        return this;
    }


}