How do I split a stream to handle elements in different ways?

user_1613383 0
Viewed 560

I'm using code like this to process a file consisting of multiple lines:

BufferedReader reader = ...
reader.lines().forEach(Same common Action)

This works fine as long as every line needs to be handled the same way. But sometimes there may be several different actions.

For example, let's say the first line is a header and the other lines are content. For the first line I want to perform Action1, and for the others I want Action2.

In Java 7 style I'd do something like this:

String line;
boolean first = true;
while ( (line = reader.readLine()) != null) {
    if (first) {
        action1(line);
        first = false;
    } else {
        action2(line);
    }
}

But that's complex and ugly, and it is not using streams at all. How can I do that in an idiomatic way using Java 8 streams?

java-stream java-8 java
3 Answers
Spliterator<String> sp = reader.lines().spliterator();    

sp.tryAdvance(YourConsumer)
sp.forEachRemaning(DifferentConsumer)
user_10593720

Don’t make your life so hard:

String header = reader.readLine();
if(header != null) {
    action1(header);
    reader.lines().forEach(line -> action2(line));
}

or if you have already existing actions implementing Consumer<String>:

String header = reader.readLine();
if(header != null) {
    action1.accept(header);
    reader.lines().forEach(action2);
}
user_27114880

You can try this with AtomicBoolean

AtomicBoolean first = new AtomicBoolean(true);
reader.lines()
      .forEach(s -> first.getAndSet(false) ? FirstConsumer : SecondConsumer);
user_87664190
Related