Inclusive takeWhile() for Streams

Question

I want to know if there is a way to add the last element of the stream that was tested against the condition of the method takeWhile(). I believe I want to achieve something

Answer 1

I add an horrible approach that might be overkill for your (or any) use case, but just for fun.

public static void main(String[] args) {
    List source = List.of(1, 3, 2, 5, 4, 6);
    Iterator iterator = source.iterator();
    AtomicBoolean proceed = new AtomicBoolean(true);

    Stream
        .generate(() -> {
            if (!proceed.get() || !iterator.hasNext()) {
                return null;
            }
            int value = iterator.next();
            System.out.println("generate: " + value);
            proceed.set(value < 5);
            return value;
        })
        .takeWhile(Objects::nonNull)
        .forEach(bar -> System.out.println("forEach: " + bar));

}

The output will be:

generate: 1
forEach: 1
generate: 3
forEach: 3
generate: 2
forEach: 2
generate: 5
forEach: 5

Probably the worst thing about this approach is that it gives generate() a responsibility (checking if there are more) that it does not belong with it.