fun.mike.flapjack.pipeline.lab.GroupOutputChannel 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

There is a newer version: 0.0.15

package fun.mike.flapjack.pipeline.lab;

import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.function.Function;

import fun.mike.record.alpha.Record;

public class GroupOutputChannel implements OutputChannel>> {
    private final Function groupBy;
    private final Map> values;

    public GroupOutputChannel(Function groupBy) {
        this.values = new HashMap<>();
        this.groupBy = groupBy;
    }

    @Override
    public Optional put(int number, String line, Record value) {
        try {
            G group = groupBy.apply(value);
            if (values.containsKey(group)) {
                values.get(group).add(value);
            } else {
                List groupValues = new LinkedList<>();
                groupValues.add(value);
                values.put(group, groupValues);
            }
            return Optional.empty();
        } catch (Exception ex) {
            return Optional.of(OutputPipelineError.build(number, line, value, ex));
        }
    }

    @Override
    public Map> getValue() {
        return values;
    }

    @Override
    public void close() {
    }
}