Java 8 Stream API：高效构建含空值键的映射并更新关联对象

import lombok.Data;
import lombok.EqualsAndHashCode;

import java.math.BigDecimal;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.function.Function;
import java.util.stream.Collectors;

@Data
@EqualsAndHashCode(onlyExplicitlyIncluded = true)
static class ProductKey {
    @EqualsAndHashCode.Include
    private Long productCode;
    @EqualsAndHashCode.Include
    private Long productDetailCode; // 假设此字段可能为null，或不用于查找
}

@Data
static class Product {
    @EqualsAndHashCode.Include
    private ProductKey productKey;
    private ProductDetail productDetail; // 初始可能为null
}

@Data
@EqualsAndHashCode(onlyExplicitlyIncluded = true)
static class ProductDetail {
    @EqualsAndHashCode.Include
    private Long productCode;
    private String description;
    private BigDecimal price;
    private String category;
}

// 辅助方法，模拟查找ProductKey，可能返回Optional.empty()
public static Optional<ProductKey> findProductKey(Long productCode, List<ProductKey> productKeys) {
    return productKeys.stream()
            .filter(productKey -> productCode.equals(productKey.getProductCode()))
            // takeWhile(productKey -> productKey != null) 在这里是冗余的，
            // 因为filter已经确保了productKey非null，且Optional.findFirst()会处理空流
            .findFirst();
}

// 辅助方法，模拟将ProductDetail列表转换为Map<Long, ProductDetail>
public static Map<Long, ProductDetail> mapProductCodeToProductDetail(List<ProductDetail> productDetailList) {
    return productDetailList.stream()
            .collect(Collectors.toMap(
                    ProductDetail::getProductCode,
                    Function.identity(),
                    (existing, replacement) -> existing // 处理重复productCode的情况
            ));
}

// 模拟数据初始化
List<ProductKey> productKeyList = List.of(
        new ProductKey() {{ setProductCode(101L); setProductDetailCode(1L); }},
        new ProductKey() {{ setProductCode(102L); setProductDetailCode(2L); }},
        new ProductKey() {{ setProductCode(103L); setProductDetailCode(3L); }}, // 假设103没有对应的ProductDetail
        new ProductKey() {{ setProductCode(101L); setProductDetailCode(4L); }}  // 模拟重复的productCode，但ProductKey不同
);

List<ProductDetail> productDetailRawList = List.of(
        new ProductDetail() {{ setProductCode(101L); setDescription("Detail A"); setPrice(BigDecimal.valueOf(10.0)); }},
        new ProductDetail() {{ setProductCode(102L); setDescription("Detail B"); setPrice(BigDecimal.valueOf(20.0)); }}
);

// 首先，将原始的ProductDetail列表转换为以productCode为键的Map，方便查找
Map<Long, ProductDetail> productDetailMap = mapProductCodeToProductDetail(productDetailRawList);

// 构建 Map<ProductKey, ProductDetail>
Map<ProductKey, ProductDetail> prodDetailByKey = productKeyList.stream()
    // 1. 调用 findProductKey 模拟查找，将每个 ProductKey 转换为 Optional<ProductKey>
    .map(productKey -> findProductKey(productKey.getProductCode(), productKeyList))
    // 2. 过滤掉空的 Optional，只保留成功找到 ProductKey 的元素
    .filter(Optional::isPresent)
    // 3. 从 Optional 中提取实际的 ProductKey 对象
    .map(Optional::get)
    // 4. 使用 Collectors.toMap() 进行收集
    .collect(Collectors.toMap(
        Function.identity(), // keyMapper: ProductKey 本身就是我们想要的键
        productKey -> productDetailMap.get(productKey.getProductCode()), // valueMapper: 根据 ProductKey 的 productCode 从 productDetailMap 中获取 ProductDetail
        (existing, replacement) -> existing // mergeFunction: 处理如果 ProductKey 列表有重复 ProductKey 导致键冲突的情况，这里选择保留旧值
    ));

System.out.println("生成的映射 (prodDetailByKey):");
prodDetailByKey.forEach((key, value) -> System.out.println("  Key: " + key.getProductCode() + ", Value: " + (value != null ? value.getDescription() : "null")));

// 模拟 Product 列表
List<Product> products = List.of(
        new Product() {{ setProductKey(new ProductKey() {{ setProductCode(101L); setProductDetailCode(1L); }}); }},
        new Product() {{ setProductKey(new ProductKey() {{ setProductCode(102L); setProductDetailCode(2L); }}); }},
        new Product() {{ setProductKey(new ProductKey() {{ setProductCode(103L); setProductDetailCode(3L); }}); }} // 103没有对应的ProductDetail
);

System.out.println("\n更新前的 Product 列表:");
products.forEach(p -> System.out.println("  ProductKey: " + p.getProductKey().getProductCode() + ", Detail: " + (p.getProductDetail() != null ? p.getProductDetail().getDescription() : "null")));

// 使用 forEach 循环更新 Product 对象的 ProductDetail
products.forEach(product -> {
    ProductDetail detail = prodDetailByKey.get(product.getProductKey());
    product.setProductDetail(detail);
});

System.out.println("\n更新后的 Product 列表:");
products.forEach(p -> System.out.println("  ProductKey: " + p.getProductKey().getProductCode() + ", Detail: " + (p.getProductDetail() != null ? p.getProductDetail().getDescription() : "null")));