com.github.slem1.await.MojoEntryPoint Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of await-maven-plugin Show documentation

A plugin to pause maven build until some services become available.

There is a newer version: 1.2.2

package com.github.slem1.await;

import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.plugins.annotations.Parameter;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;

/**
 * The plugin entry point. Retrieves the plugin configuration and runs the underlying tasks.
 *
 * @author slemoine
 */
@Mojo(name = "Await")
public class MojoEntryPoint extends AbstractMojo {

    private static final PollingConfig DEFAULT_POLLING_CONFIG = new PollingConfig();

    private static final Comparator TASK_PRIORITY_COMPARATOR = new Comparator() {
        @Override
        public int compare(PollingTask o1, PollingTask o2) {
            return Integer.compare(o1.getPriority(), o2.getPriority());
        }
    };

    @Parameter
    private List tcpConnections;

    @Parameter
    private List httpConnections;

    @Parameter
    private PollingConfig poll;

    //for unit testing
    void setTcpConnections(List tcpConnections) {
        this.tcpConnections = tcpConnections;
    }

    //for unit testing
    void setHttpConnections(List httpConnections) {
        this.httpConnections = httpConnections;
    }

    //for unit testing
    void setPoll(PollingConfig pollingConfig) {
        this.poll = pollingConfig;
    }

    /**
     * {@inheritDoc}
     */
    public void execute() throws MojoFailureException, MojoExecutionException {

        List configs = new ArrayList<>();

        if (tcpConnections != null) {
            configs.addAll(tcpConnections);
        }

        if (httpConnections != null) {
            configs.addAll(httpConnections);
        }

        try {
            Set tasks = toPollingTasks(configs);

            if (tasks.isEmpty()) {
                getLog().warn("No tasks found");
            } else {

                for (PollingTask task : tasks) {
                    task.run();
                }

            }
        } catch (IllegalArgumentException e) {
            throw new MojoFailureException(e.getMessage(), e);
        }
    }

    private Set toPollingTasks(List configs) {

        Set pollingTasks = new TreeSet<>(TASK_PRIORITY_COMPARATOR);

        PollingConfig pollingConfig = poll == null ?
                DEFAULT_POLLING_CONFIG : poll.validate();

        for (MojoConnectionConfig config : configs) {
            Service service = config.buildService();
            PollingTask task = new PollingTask(service,
                    pollingConfig.getAttempts(),
                    pollingConfig.getSleep(), config.getPriority());
            pollingTasks.add(task);
        }

        return pollingTasks;
    }
}