com.boozallen.aiops.mda.metamodel.element.java.JavaPipeline Maven / Gradle / Ivy

Go to download
package com.boozallen.aiops.mda.metamodel.element.java;

/*-
 * #%L
 * AIOps Foundation::AIOps MDA
 * %%
 * Copyright (C) 2021 Booz Allen
 * %%
 * This software package is licensed under the Booz Allen Public License. All Rights Reserved.
 * #L%
 */

import com.boozallen.aiops.mda.metamodel.element.BasePipelineDecorator;
import com.boozallen.aiops.mda.metamodel.element.Pipeline;

import java.util.Set;
import java.util.TreeSet;
import java.util.List;
import java.util.stream.Collectors;

/**
 * Pipeline decorator to ease generation of Java files.
 */
public class JavaPipeline extends BasePipelineDecorator {

	/**
	 * {@inheritDoc}
	 */
	public JavaPipeline(Pipeline pipelineToDecorate) {
		super(pipelineToDecorate);

	}

	/**
	 * Returns all steps as JavaSteps that have either an inbound type of messaging or an outbound type of messaging.
	 *
	 * @return all Java messaging steps
	 */
	public List getMessagingSteps() {
		return wrapped.getSteps().stream()
				.map(JavaStep::new)
				.filter(step -> step.hasMessagingInbound() || step.hasMessagingOutbound())
				.collect(Collectors.toList());
	}

	/**
	 * Returns all the distinct impl imports from Java steps contained in this pipeline.
	 *
	 * @return all distinct Java steps impl imports in the pipeline
	 */
	public Set getAllStepsImplImports() {
		Set allStepsImplImports = new TreeSet<>();
		wrapped.getSteps().stream()
				.map(JavaStep::new)
				.forEach(step -> {
					allStepsImplImports.addAll(step.getImplImports());
				});
		return allStepsImplImports;

	}
}