Java带计数器的嵌套for循环逻辑转换为Streams实现问题求助

问题核心原因 #

你原来的写法将counter作为全局变量，会出现多线程冲突、多key处理时计数混乱的问题，也不符合Stream无副作用的设计规范，无法对齐原逻辑每个key对应独立计数器的要求。

最优解决方案 #

直接通过遍历List下标获取计数器，不需要额外维护计数变量，逻辑和原实现完全对齐，性能更高也更安全：

private static void after(Map<String, List<String>> map) {
    map.forEach((headerKey, valueList) -> {
        int listSize = valueList.size();
        if (listSize == 1) {
            System.out.println("Response header: " + headerKey + ": " + valueList.get(0));
            return;
        }
        // 下标从0开始，+1即可得到从1计数的序号
        IntStream.range(0, listSize)
                .forEach(idx -> System.out.printf("Response header: %s[%d]: %s%n",
                        headerKey, idx + 1, valueList.get(idx)));
    });
}

兼容你原有写法的改造方案 #

如果你要保留拆分的output、print方法结构，可以给每个key创建独立的原子计数器，避免全局变量冲突：

private static long after(Map<String, List<String>> map) {
    return map.keySet().stream()
            .peek(key -> output(key, map))
            .count();
}

private static void output(String key, Map<String, List<String>> map) {
    List<String> valueList = map.get(key);
    // 每个key对应独立的线程安全计数器，作用域仅在当前key的处理流程
    AtomicInteger counter = new AtomicInteger(1);
    valueList.stream().forEach(value -> print(value, key, valueList, counter));
}

private static void print(String value, String key, List<String> strings, AtomicInteger counter) {
    if (strings.size() != 1) {
        System.out.println("Response header: " + key + "[" + counter.getAndIncrement() + "]: " + value);
    } else {
        System.out.println("Response header: " + key + ": " + value);
    }
}

注意事项 #

• 不要用普通int作为全局计数器，并行流或者多线程调用时计数会完全错乱
• 纯打印的逻辑不需要用map转空字符串再count，直接用forEach处理即可，count返回的处理数量如果不需要可以把方法返回值改为void

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