如何用Java 8 Stream替代循环提取priceCode生成String列表？

问题说明 #

原本用foreach循环从ItemPriceCode列表里提取priceCode字符串生成String列表，想换成Java 8 Stream实现，但试的代码没效果，需要写出和原循环功能一致的Stream代码。

原循环代码 #

List<ItemPriceCode> itemPriceCodes = item.getItemPriceCodes();
List<String> priceCodeList = new ArrayList<String>();                              
for (ItemPriceCode ipc : itemPriceCodes) {
    // 从列表中提取字符串值
    priceCodeList.add(ipc.getPriceCode());
}

尝试的错误Stream代码 #

itemPriceCodes.stream().map(n -> String.valueOf(n)).collect(Collectors.toList());

问题出在哪 #

你写的Stream代码里，map(n -> String.valueOf(n))是把整个ItemPriceCode对象转成字符串（默认调用对象的toString()），根本没提取priceCode属性，这和原循环逻辑完全不一样，当然达不到预期效果。

正确的Stream写法 #

直接在map里调用ItemPriceCode的getPriceCode()方法就行，用方法引用更简洁：

List<String> priceCodeList = itemPriceCodes.stream()
    .map(ItemPriceCode::getPriceCode) // 等价于lambda表达式ipc -> ipc.getPriceCode()
    .collect(Collectors.toList());

替换后的完整函数 #

把原循环替换成上面的Stream代码后，完整函数如下：

private Item getItemManufacturerPriceCodes(Item item) {
    List<ItemPriceCode> itemPriceCodes = item.getItemPriceCodes();
    
    // Stream替代原foreach循环提取priceCode列表
    List<String> priceCodeList = itemPriceCodes.stream()
        .map(ItemPriceCode::getPriceCode)
        .collect(Collectors.toList());
    
    // 执行查询
    List<ManufacturerPriceCodes> mpc = manufacturerPriceCodesRepository
        .findByManufacturerIDAndPriceCodeInAndRecordDeleted(item.getManufacturerID(), priceCodeList, NOT_DELETED);
    
    // 转成Map
    Map<String, ManufacturerPriceCodes> ipcToMFPNameMap = mpc.stream()
        .collect(Collectors.toMap(ManufacturerPriceCodes::getPriceCode, Function.identity()));
    
    // 给每个ItemPriceCode设置值（原代码逻辑存疑，这里仅换成Stream的forEach写法）
    itemPriceCodes.forEach(ipcs -> {
        // 原代码设置的是Map的类名，这大概率不是你想要的，应该是根据priceCode取对应的ManufacturerPriceCodes对象？
        // 比如：ManufacturerPriceCodes mpcObj = ipcToMFPNameMap.get(ipcs.getPriceCode());
        // ipcs.setManufacturerPriceCode(mpcObj != null ? mpcObj.get某个属性() : null);
        ipcs.setManufacturerPriceCode(ipcToMFPNameMap.getClass().getName());
    });
    
    // 删除标记为已删除的记录
    item.getItemPriceCodes().removeIf(ipcs -> DELETED.equals(ipcs.getRecordDeleted()));
    
    return item;      
}

提个醒：原代码里ipcs.setManufacturerPriceCode(ipcToMFPNameMap.getClass().getName())这行逻辑可能有问题，它设置的是Map的类名字符串，而不是根据priceCode匹配到的ManufacturerPriceCodes相关数据，如果你的需求是设置对应的对象或其属性，得调整这部分逻辑，比如通过ipcToMFPNameMap.get(ipcs.getPriceCode())拿到对应的值再处理。

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