java - 如何在处理 Java Streams 时从 Java IDE 请求更多返回类型信息？

Question

考虑以下任务:

Given a comma-separated list of cities (state, city, population) generate a list of states and the most populous city in each state along with the population. The list should be sorted on the name of state.

cities.csv看起来像这样:

New York, New York, 8491079
Los Angeles, California, 3928864
Chicago, Illinois, 2722389
Houston, Texas, 2239558
Philadelphia, Pennsylvania, 1560297
Phoenix, Arizona, 1537058
San Antonio, Texas, 1436697
San Diego, California, 1381069
Dallas, Texas, 1281047
San Jose, California, 1015785
...

预期的输出如下所示:

Alabama, Birmingham [212247]
Alaska, Anchorage [301010]
Arizona, Phoenix [1537058]
Arkansas, Little Rock [197706]
California, Los Angeles [3928864]
...
Texas, Houston [2239558]

我的尝试(经过反复试验并遵循 Java 8 流的食谱样式处理)如下所示:

private static void mostPopulousCityInState() throws IOException {
            Files.lines(Paths.get("cities.csv")).map(City::new)
                    .collect(groupingBy(City::getState, maxBy(Comparator.comparing(City::getPopulation)))) //1
            .entrySet() //2
            .stream()   //3
            .sorted(Comparator.comparing(e -> e.getKey()))
            .forEach(e -> {
                City city = e.getValue().get();
                System.out.println(e.getKey() + ", " + city.getName() + " [" + city.getPopulation() + "]");
            });
}

(为简洁起见省略了类 City，但它具有相当简单的构造函数和 setter/getter )。

这看起来还不错(尽管还可以改进)。但我的问题更多是关于如何从总体上考虑 Java 8 流:

1. 例如，我可以将光标放在 collect 上吗？并做一些键盘快捷键告诉我它返回 Map<String, Optional<City>> ？如果我在此处执行“导航到源”，那么它会将我带到 Stream.java 中的通用定义。显示<R, A> R collect(Collector<? super T, A, R> collector);这在这种情况下是无用的。由于编译器此时知 Prop 体类型，因此期望 IDE 向我显示它们(而不是泛型类型)是否合理？
2. 一般来说，您如何跟踪管道中的返回类型？我目前的问题是我在通用返回类型中迷路了。具体来说，是什么让你清楚 stream在 line //3上面是Stream的 Map.Entry<String, Optional<City>>是吗？
3. 它会随着时间的推移变得更好吗？还是您需要更深入地了解 Java 的类型系统？

Answer 1

你可以用更简单的方式做到这一点:你不需要在 group by 操作之后明确地对 map 进行排序:你可以直接使用自定义 SortedMap与 groupingBy(classifier, mapFactory, downstream) .在这种情况下， map 工厂可以是 TreeMap::new ，这将创建一个 SortedMap按州名升序排列。

另请注意 Files.lines 返回应在 try-with-resources 中关闭的 Stream:

The returned stream encapsulates a Reader. If timely disposal of file system resources is required, the try-with-resources construct should be used to ensure that the stream's close method is invoked after the stream operations are completed.

private static void mostPopulousCityInState() throws IOException {
    try (Stream<String> stream = Files.lines(Paths.get("cities.csv"))) {
        stream.map(City::new)
              .collect(Collectors.groupingBy(
                  City::getState,
                  TreeMap::new,
                  Collectors.maxBy(Comparator.comparing(City::getPopulation))
              ))
              .forEach((k, v) -> {
                  City city = v.get();
                  System.out.println(k + ", " + city.getName() + " [" + city.getPopulation() + "]");
              });
    }
}

通过此更改，您不需要第二个 Stream 管道来对 map 的条目进行排序，这会使您的其他问题变得没有实际意义。

IDE 问题可能是特定的 IntelliJ(不知道)。我使用最新的 Eclipse IDE (Mars.2) 进行了测试，如果我将鼠标悬停在对 collect 的调用上，它正确地告诉我返回类型是 TreeMap<String, Optional<City>> .您说得对，IDE 应该能够显示正确的类型信息。

entrySet().stream(...)足够清楚，它返回 Map.Entry 的 Stream元素。但为了更容易理解，您可以将 map 存储在局部变量中:

Map<String, Optional<City>> map = // result of operation
map.entrySet().stream() // ... etc