处理 Set<Foo> 的元素并使用流创建 Set<Bar>

Question

我有一个 Set<String> 的 "hostname:port" 对，我想从中创建一个 Set<InetSocketAddress>。我试过了：

Set<InetSocketAddress> ISAAddresses = StrAddresses
    .stream().map(addr -> new InetSocketAddress(
        addr.split(":")[0],
        Integer.parseInt(addr.split(":")[1])));

但这会在 IntelliJ 中产生以下错误：

不兼容的类型。必需 Set<InetSocketAddress> 但“地图”是 推断为Stream<R>：不存在类型变量 R 的实例，所以 Stream<R> 符合 Set<InetSocketAddress>

我使用地图和 lambda 的方式一定有问题。

Answer 1

Stream#map 函数不返回 Map。它将流中的当前元素转换（映射）到其他元素。因此，它使用 given transformation function 从 Stream<X> 生成 Stream<Y>，该函数采用 X 并输出 Y。

StrAddresses.stream()                           // String
    .map(addr -> new InetSocketAddress(
        addr.split(":")[0],
        Integer.parseInt(addr.split(":")[1]))); // InetSocketAddress

您以 Stream<String> 开头并以 Stream<InetSocketAddress> 结尾。

引用其documentation：

返回一个由将给定函数应用到该流的元素的结果组成的流。

---

如果您想将该流转换为Set，您需要使用Stream#collect 方法，如下所示：

StrAddresses.stream()
    .map(addr -> new InetSocketAddress(
        addr.split(":")[0],
        Integer.parseInt(addr.split(":")[1])))
    .collect(Collectors.toSet());

实用方法Collectors.toSet() 返回一个收集器，用于优化优化的Set。例如，如果您明确想要HashSet，您可以使用它：

.collect(Collectors.toCollection(HashSet::new));

来自其documentation：

对此流的元素执行可变归约操作。可变缩减是其中缩减的值是可变结果容器，例如ArrayList [...]

---

作为一个小提示，您目前每次拆分同一个元素两次：

addr.split(":")[0],                     // First
Integer.parseInt(addr.split(":")[1])))  // Second

您可以通过记住之前的值来保存额外的split 过程。在这种情况下，这可以通过使用第二个Stream#map 调用来优雅地完成。首先我们从Stream<String> 转换为Stream<String[]> 再转换为Stream<InetSocketAddress>：

StrAddresses.stream()                                 // String
    .map(addr -> addr.split(":"))                     // String[]
    .map(addrData -> new InetSocketAddress(
        addrData[0], Integer.parseInt(addrData[1])))  // InetSocketAddress
    .collect(Collectors.toSet());

注意Stream#map 是一个惰性操作。这意味着一旦您调用该方法，Java 将不会转换整个 Stream 从 A 到 B。它将等到像Stream#collect 这样的非惰性（最终确定）操作出现，然后遍历Stream 并应用每个惰性操作 element-wise。因此，您可以根据需要添加任意数量的Stream#map 调用，而不会在整个Stream 上产生额外循环。

Answer 2

您需要collect映射到Set后返回的InetSocketAddress地址中的Stream。这可以作为 -

Set<InetSocketAddress> ISAAddresses = StrAddresses.stream()
                  .map(addr -> new InetSocketAddress(addr.split(":")[0], Integer.parseInt(addr.split(":")[1])))
                  .collect(Collectors.toSet());