如何在Java Stream API中对流的首个元素做差异化映射？

当然可以实现！不过因为Stream API本身是无状态设计，没法直接识别"首个元素"，所以我们需要借助小技巧来追踪元素位置，区分首个元素和其他元素。下面给你几种实用方案：

方案一：用AtomicInteger追踪索引（通用型，支持并行流） #

如果已经有了map(Foo::getModifiedFoo)之后的流，或者需要兼容并行流场景，这个方案最稳妥。我们用原子类来计数，保证线程安全：

AtomicInteger elementIndex = new AtomicInteger(0);
Stream<Bar> barStream = fooList.stream()
    .map(Foo::getModifiedFoo)
    .map(modifiedFoo -> {
        // 第一个元素时index为0，getAndIncrement返回0，isNotFirst为false
        boolean isNotFirst = elementIndex.getAndIncrement() != 0;
        return new Bar(modifiedFoo, isNotFirst);
    });

原理很简单：AtomicInteger的getAndIncrement()是原子操作，第一个元素处理时返回初始值0，后续元素返回递增后的数值，以此判断是否为首个元素。这个方案不管集合是不是随机访问类型，也不管是顺序流还是并行流都能正常工作。

方案二：用IntStream直接遍历索引（顺序流+随机访问集合首选） #

如果你的集合是ArrayList这类支持快速随机访问的类型，而且只需要顺序流处理，直接用索引遍历会更简洁直观：

Stream<Bar> barStream = IntStream.range(0, fooList.size())
    .mapToObj(index -> {
        Foo modifiedFoo = fooList.get(index).getModifiedFoo();
        // 直接判断索引是否为0，设置Bar的第二个参数
        return new Bar(modifiedFoo, index != 0);
    });

这个方案不需要额外的原子类，代码可读性更强，但要注意：如果是LinkedList这类非随机访问集合，get(index)的时间复杂度是O(n)，整体性能会下降，所以只适合随机访问集合的场景。

针对已有ModifiedFoo流的场景 #

如果你已经提前生成了Stream<ModifiedFoo>（也就是已经执行完map(Foo::getModifiedFoo)），那只能用方案一的变种，因为此时流已经丢失了原集合的索引信息：

// 假设这是你已经构建好的ModifiedFoo流
Stream<ModifiedFoo> modifiedFooStream = fooList.stream().map(Foo::getModifiedFoo);

AtomicInteger elementIndex = new AtomicInteger(0);
Stream<Bar> barStream = modifiedFooStream
    .map(modifiedFoo -> new Bar(modifiedFoo, elementIndex.getAndIncrement() != 0));

注意事项 #

• 不要用普通的int变量替代AtomicInteger：lambda表达式要求捕获的变量是final或有效final，而且并行流场景下普通int会有线程安全问题，原子类是唯一安全的选择。
• 并行流场景下，方案二的IntStream方式并不适合，因为fooList.get(index)在并行时可能会有线程安全问题（如果集合本身不是线程安全的），而方案一的原子类方式是线程安全的。

内容的提问来源于stack exchange，提问作者ankitkpd