com.g2forge.gearbox.maven.ProcessOutputHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gb-maven Show documentation
Command line wrapper and helpers for running maven.
The newest version!
package com.g2forge.gearbox.maven;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import org.slf4j.Logger;

import com.g2forge.alexandria.java.core.helpers.HCollector;
import com.g2forge.alexandria.java.function.IFunction1;
import com.g2forge.alexandria.java.io.RuntimeIOException;
import com.g2forge.gearbox.command.process.IProcess;

import lombok.Builder;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.Singular;

@Getter
@Builder(toBuilder = true)
@RequiredArgsConstructor
public class ProcessOutputHandler {
	public static interface IOutputMatcher {
		public boolean isApplicable(boolean success);

		/**
		 * Test if this condition is now a match.
		 * 
		 * @param line A line of output
		 * @param output {@code true} if the line comes from stdout, {@code false} if it comes from stderr.
		 * @return {@code null} if this condition may or may not match, {@code true} if it will and we can stop scanning, {@code false} if it will not match and
		 *         we can stop scanning.
		 */
		public Boolean isMatch(String line, boolean output);
	}

	protected final Logger log;

	@Singular
	protected final Set keys;

	protected final IFunction1 matcherFactory;

	public Set handle(IProcess process) {
		final Map matchers = keys.stream().collect(Collectors.toMap(IFunction1.identity(), getMatcherFactory(), HCollector.mergeFail(), LinkedHashMap::new));
		final Map possible = new LinkedHashMap<>(matchers);
		final Set retVal = new LinkedHashSet<>();

		try {
			stream(process, possible, retVal, true);
			stream(process, possible, retVal, false);
		} catch (IOException e) {
			throw new RuntimeIOException(e);
		}

		final boolean success = process.isSuccess();
		for (K key : new ArrayList<>(retVal)) {
			final IOutputMatcher matcher = matchers.get(key);
			if (!matcher.isApplicable(success)) retVal.remove(key);
		}

		return retVal;
	}

	protected void stream(IProcess process, final Map possible, final Set retVal, boolean output) throws IOException {
		try (final BufferedReader reader = new BufferedReader(new InputStreamReader(output ? process.getStandardOutput() : process.getStandardError()))) {
			while (true) {
				final String line = reader.readLine();
				if (line == null) break;
				if ((log != null) && log.isDebugEnabled()) log.debug(line);

				final Set complete = new HashSet<>();
				for (Map.Entry entry : possible.entrySet()) {
					final Boolean match = entry.getValue().isMatch(line, output);
					if (match == null) continue;
					if (match) retVal.add(entry.getKey());
					complete.add(entry.getKey());
				}
				complete.forEach(possible::remove);
			}
		}
	}
}