fun.mike.flapjack.pipeline.lab.TransformBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of flapjack-lab Show documentation

The newest version!

package fun.mike.flapjack.pipeline.lab;

import java.util.LinkedList;
import java.util.List;
import java.util.function.Function;
import java.util.function.Predicate;

import fun.mike.record.alpha.Record;

public class TransformBuilder {
    private final List operations;

    public TransformBuilder() {
        operations = new LinkedList<>();
    }

    // Map
    public TransformBuilder map(Function mapper) {
        return map(null, null, mapper);
    }

    public TransformBuilder map(String id, Function mapper) {
        return map(id, null, mapper);
    }

    public TransformBuilder map(String id, String description, Function mapper) {
        MapOperation operation = new MapOperation(id, description, mapper);
        operations.add(operation);
        return this;
    }

    // Filter
    public TransformBuilder filter(Predicate predicate) {
        return filter(null, null, predicate);
    }

    public TransformBuilder filter(String id, Predicate predicate) {
        return filter(id, null, predicate);
    }

    public TransformBuilder filter(String id, String description, Predicate predicate) {
        FilterOperation operation = new FilterOperation(id, description, predicate);
        operations.add(operation);
        return this;
    }

    public Transform build() {
        return new GenericTransform(operations);
    }

    public TransformResult run(Record record) {
        return build().run(record);
    }
}