io.hyperfoil.api.config.BaseSequenceBuilder Maven / Gradle / Ivy
package io.hyperfoil.api.config;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.ServiceLoader;
import java.util.stream.Collectors;
import io.hyperfoil.impl.StepCatalogFactory;
public abstract class BaseSequenceBuilder> implements BuilderBase {
private static final Map, StepCatalogFactory> factories = new HashMap<>();
protected final BaseSequenceBuilder parent;
protected final List> steps = new ArrayList<>();
static {
ServiceLoader.load(StepCatalogFactory.class).forEach(factory -> factories.put(factory.clazz(), factory));
}
public BaseSequenceBuilder(BaseSequenceBuilder parent) {
this.parent = parent;
}
public D step(Class catalogClass) {
StepCatalogFactory factory = factories.get(catalogClass);
if (factory == null) {
throw new IllegalStateException("Cannot load step catalog");
}
Step.Catalog catalog = factory.create(this);
if (catalogClass.isInstance(catalog)) {
return catalogClass.cast(catalog);
} else {
throw new IllegalStateException("Unknown step catalog " + catalog + ", want: " + catalogClass);
}
}
@SuppressWarnings("unchecked")
public S self() {
return (S) this;
}
public S step(Step step) {
steps.add(new ProvidedStepBuilder(step));
return self();
}
public S step(SimpleBuilder builder) {
steps.add(new SimpleAdapter(builder));
return self();
}
// Calling this method step() would cause ambiguity with step(Step) defined through lambda
public S stepBuilder(StepBuilder stepBuilder) {
steps.add(stepBuilder);
return self();
}
public BaseSequenceBuilder end() {
return parent;
}
public SequenceBuilder rootSequence() {
return parent.rootSequence();
}
public ScenarioBuilder endSequence() {
return rootSequence().endSequence();
}
public String name() {
return rootSequence().name();
}
public BaseSequenceBuilder insertBefore(Locator locator) {
return insertWithOffset(locator, 0);
}
public BaseSequenceBuilder insertAfter(Locator locator) {
return insertWithOffset(locator, 1);
}
private StepInserter insertWithOffset(Locator locator, int offset) {
for (int i = 0; i < steps.size(); ++i) {
if (steps.get(i) == locator.step()) {
StepInserter inserter = new StepInserter(this);
steps.add(i + offset, inserter);
return inserter;
}
}
throw new NoSuchElementException("Not found: " + locator.step());
}
public void prepareBuild() {
// We need to make a defensive copy as prepareBuild() may trigger modifications
new ArrayList<>(steps).forEach(stepBuilder -> {
Locator.push(stepBuilder, BaseSequenceBuilder.this);
stepBuilder.prepareBuild();
Locator.pop();
});
}
public List buildSteps() {
return steps.stream().map(stepBuilder -> {
Locator.push(stepBuilder, BaseSequenceBuilder.this);
List steps = stepBuilder.build();
Locator.pop();
return steps;
}).flatMap(List::stream).collect(Collectors.toList());
}
public int indexOf(StepBuilder builder) {
return steps.indexOf(builder);
}
public boolean isEmpty() {
return steps.isEmpty();
}
public int size() {
return steps.size();
}
/**
* Simplified interface that works better with lambdas
*/
@FunctionalInterface
public interface SimpleBuilder {
Step build();
}
private static class StepInserter extends BaseSequenceBuilder implements StepBuilder {
private StepInserter(BaseSequenceBuilder parent) {
super(parent);
}
@Override
public List build() {
return buildSteps();
}
}
private static class ProvidedStepBuilder implements StepBuilder {
private final Step step;
public ProvidedStepBuilder(Step step) {
this.step = step;
}
@Override
public List build() {
return Collections.singletonList(step);
}
@Override
public ProvidedStepBuilder copy(Object newParent) {
// This builder is immutable
return this;
}
}
public static class SimpleAdapter implements StepBuilder {
private final SimpleBuilder builder;
public SimpleAdapter(SimpleBuilder builder) {
this.builder = builder;
}
@Override
public List build() {
return Collections.singletonList(builder.build());
}
}
}