使用 Java Stream API 以声明性方式重写创建和填充 Map 的命令式代码

发布于05月13日

我有一段代码:

Map<String, BillingAccount> billingAccountsMap = new HashMap<>();
for (BillingAccount ba : billingAccounts) {
    if (ba.getBillingAccountIdentifier() != null && ba.getBillingAccountIdentifier().getIdentifier() != null) {
        billingAccountsMap.put(ba.getBillingAccountIdentifier().getIdentifier(), ba);
    }
}

我只想用Java Stream API和collect(Collectors.toMap())以功能性的方式重写它，但我对null个 case 有点困惑.

我正在使用Java11.

推荐答案

您可以使用filter()操作过滤掉可为空的值，然后通过向其中传递内置收集器Collectors.toMap()来应用collect:

Map<String, BillingAccount> billingAccountsById = billingAccounts.stream()
    .filter(account -> Objects.nonNull(account.getBillingAccountIdentifier()))
    .filter(account -> Objects.nonNull(account.getBillingAccountIdentifier().getIdentifier()))
    .collect(Collectors.toMap(
        account -> account.getBillingAccountIdentifier().getIdentifier(), // keyMapper
        Function.identity())); // valueMapper

注意，对于这种方法，每个标识符都必须是唯一的.否则，需要提供mergeFunction作为第三个参数来解析映射到同一个键的值.