Stream API - 需要遍历列表并与 Map 进行比较。将结果写入不同的 Map

Question

公共 class ProductInStockRequest {

private String productId;

private Integer requestedQuantity;

}

我有一个清单

    requestList.add(new ProductInStockRequest("100", 5));
    requestList.add(new ProductInStockRequest("200", 11));
    requestList.add(new ProductInStockRequest("300", 33));
    requestList.add(new ProductInStockRequest("400", 55));

我有一个 Map<String, Integer> productInDbMap = new HashMap<>();

    productInDbMap.put("100", 10);
    productInDbMap.put("200", 10);
    productInDbMap.put("300", 44);
    productInDbMap.put("400", 77);

我创建了 new Map<String, String> responseMap = new HashMap<>();

我需要对 List 中的每个元素进行 go 并检查相关 productId 是否有足够的数量并将结果写入 responseMap

尝试做这样的事情：

    requestList.stream().map(requestedItem -> {
        int quantity = productInDbMap.get(requestedItem.getProductId());

        if (quantity >= requestedItem.getRequestedQuantity()) {
            responseMap.put(requestedItem.getProductId(), "order-able");

        } else {
            int availableQuantity = quantity - requestedItem.getRequestedQuantity();
            String s = String.valueOf(availableQuantity);
            responseMap.put(requestedItem.getProductId(), s);
        }
        return responseMap;
    });

没有运气，请指教

Answer 1

你可以从这样的事情开始：

        Map<String, String> responseMap = requestList.stream().map(requestedItem -> {
            int quantity = productInDbMap.get(requestedItem.getProductId());

            if (quantity >= requestedItem.getRequestedQuantity()) {
                return new AbstractMap.SimpleEntry<>(requestedItem.getProductId(), "order-able");
            } else {
                int availableQuantity = quantity - requestedItem.getRequestedQuantity();
                String s = String.valueOf(availableQuantity);
                return new AbstractMap.SimpleEntry<>(requestedItem.getProductId(), s);
            }
        }).collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

一般来说，永远不要修改 stream 或 lambda 中的任何内容。 在这种情况下编译器将允许它（并且会在修改变量的情况下抱怨），但这始终是一种不好的做法并且不安全。

Answer 2

可直接代收至map：

Map<String,String> responseMap = requestList.stream()
            .collect(Collectors.toMap(ProductInStockRequest::getProductId, requestedItem ->  {
            int quantity = productInDbMap.get(requestedItem.getProductId());

            if (quantity >= requestedItem.getRequestedQuantity()) {
                return"order-able";
            } else {
                int availableQuantity = quantity - requestedItem.getRequestedQuantity();
                return String.valueOf(availableQuantity);
            }
        }));

如果您使用 forEach，您的方法将工作正常：

Map<String,String> responseMap = new HashMap<>()
requestList.stream()
            .forEach(requestedItem ->  {
                int quantity = productInDbMap.get(requestedItem.getProductId());

                if (quantity >= requestedItem.getRequestedQuantity()) {
                    responseMap.put(requestedItem.getProductId(), "order-able");

                } else {
                    int availableQuantity = quantity - requestedItem.getRequestedQuantity();
                    String s = String.valueOf(availableQuantity);
                    responseMap.put(requestedItem.getProductId(), s);
                }
        });