如何在Java Stream收集Map时同步更新Order的Product属性

问题背景 #

我从第三方API获取了一个Order列表：

List<Order> orders = fromApi(orderId);

Order类定义：

class Order{
     Long orderId;
     Product product;
}

接着通过订单ID调用另一个API获取了Product的映射关系：

List<Long> orderIds = orders.stream().map(Order::orderId).collect(Collectors.toList());
Map<Long,Product> userProductMap = fromAnotherApi(orderIds);

Product类定义：

class Product{
        String productDesc;
}

现在需要生成一个以orderId为键、Order对象为值的Map，同时在Stream迭代过程中，用userProductMap中的Product更新对应Order的product属性。当前代码使用Function.identity()直接返回原对象，需要替换为自定义逻辑完成属性更新。

实现代码 #

直接在Collectors.toMap的第二个参数中编写lambda表达式，先完成Order的属性更新，再返回更新后的对象：

Map<Long, Order> userOrderMap = orders.stream()
    .collect(Collectors.toMap(
        Order::orderId, // 提取orderId作为Map的键
        order -> {
            // 从映射中获取对应Product并更新Order
            order.product = userProductMap.get(order.orderId);
            return order; // 返回更新后的Order作为Map的值
        }
    ));

空值安全优化 #

如果userProductMap中可能不存在某些orderId对应的Product，可以添加空值处理避免空指针：

Map<Long, Order> userOrderMap = orders.stream()
    .collect(Collectors.toMap(
        Order::orderId,
        order -> {
            // 若不存在对应Product，可设置默认值或保持原属性
            order.product = Optional.ofNullable(userProductMap.get(order.orderId))
                .orElse(new Product()); // 示例：创建空的Product对象
            return order;
        }
    ));

不修改原对象的版本 #

如果不想修改原orders列表中的对象，可以先创建Order的副本再更新属性（假设Order有拷贝构造函数）：

Map<Long, Order> userOrderMap = orders.stream()
    .collect(Collectors.toMap(
        Order::orderId,
        order -> {
            Order copiedOrder = new Order(order); // 拷贝原Order对象
            copiedOrder.product = userProductMap.get(order.orderId);
            return copiedOrder;
        }
    ));

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