Java 8 Stream从对象映射创建对象

Question

我刚开始通过Java 8流API学习和实现集合。 我有一堂课：

public class Discount {
    int amount;
    String lastMarketingRegion;

    Discount (int amount, String lastMarketingRegion) {
        this.amount = amount;
        this.lastMarketingRegion= lastMarketingRegion;
    }

    public int getAmount() { return amount; }

    public String getLastMarketingRegion() { return lastMarketingRegion; }

    public String toString() {
        return String.format("{%s,\"%s\"}", amount, lastMarketingRegion);
    }
}

我得到以下内容：

Map<String, Discount> prepaid = new HashMap<String, Discount>();
prepaid.put("HAPPY50", new Discount(100, "M1"));
prepaid.put("LUCKY10", new Discount(10, "M2"));
prepaid.put("FIRSTPAY", new Discount(20, "M3"));

Map<String, Discount> otherBills = new HashMap<String, Discount>();
otherBills.put("HAPPY50", new Discount(60, "M4"));
otherBills.put("LUCKY10", new Discount(7, "M5"));
otherBills.put("GOOD", new Discount(20, "M6"));

List<Map<String, Discount>> discList = new ArrayList<Map<String, Discount>>();
discList.add(prepaid);
discList.add(otherBills);

因此，基本上，我有一个针对不同付款方式的所有折扣代码的Discount图列表。

要求是使用sum_of_amount和last_region创建所有付款类型的所有折扣代码的单个地图：

Map<String, Discount> totalDiscounts = 
{LUCKY10={17, "M5"}, FIRSTPAY={20, "M3"}, HAPPY50={160, "M4"}, GOOD={20, "M6"}}

我可以得到：

Map<String, Integer> totalDiscounts = 
    {LUCKY10=17, FIRSTPAY=20, HAPPY50=160, GOOD=20}

通过使用以下代码：

 Map<String, Integer> afterFormatting = discList.stream()
                           .flatMap(m -> m.entrySet().stream())
                           .collect(Collectors.groupingBy(Map.Entry::getKey, Collectors.summingInt(map -> map.getValue().amount)));

但该区域也需要一个Discount对象。

我需要一个Discount对象的集合，其中的金额是相同键的总金额，而区域则来自otherBills。

任何帮助将非常感激。 谢谢。

编辑1-为简单起见，请考虑lastMarketingRegion对于折扣代码具有相同的值。 我也试图通过图表来解释它-

Answer 1

来自评论

当您为LUCKY10拥有“ M2”和“ M5”条目时，为什么会期望“ LUCKY10”-“ M5”？

因为其他账单比预付费有更高的优先级

您可以为此使用Collectors.toMap 。 它的最后一个参数是mergeFunction ，它合并两个在地图中具有相同String键的Discounts。

Map<String, Discount> totalDiscounts = discList.stream()
            .flatMap(m -> m.entrySet().stream())
            .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue,
                    (discount1, discount2) -> new Discount(discount1.getAmount() + discount2.getAmount(),
                            discount2.getLastMarketingRegion())));

由于出列表生成的数据流是有序的，在discount2 折扣会从一个otherBills映射，所以我选择它的区域。

如果您通过添加otherBills和prepaid构造列表，那么它将具有不同的输出。

依靠相遇顺序使这不是一个很好的解决方案 。 （如果您假设我们在处理完第一个地图后处理了第二个地图中的条目，为什么要首先合并它们？）

查看我使用Map.merge 其他答案

Answer 2

如果只有两个地图，则可以使用Map.merge ，而不是使用基于流的解决方案（我的其他答案 ）。

在这里，我们制作了prepaid地图的副本。 然后，我们遍历otherBills地图。 对于每个键

1. 如果映射不存在，则将其添加到地图（结果地图）
2. 如果映射已经存在，我们将构造一个新的Discount对象，其金额为地图中已存在的Discount对象（一个来自prepaid ）和当前Discount对象（一个来自otherBill ）的otherBill 。 它从otherBill映射中获取Discount对象的区域。

---

Map<String, Discount> result = new HashMap<>(prepaid);
otherBills.forEach((k, v) -> result.merge(k, v, (discountFromPrepaid, discountFromOtherBill) ->
        new Discount(discountFromPrepaid.getAmount() + discountFromOtherBill.getAmount(),
                discountFromOtherBill.getLastMarketingRegion())));