Соберите сложные объекты в одном лямбда-выражении - PullRequest
Новая
       13

Соберите сложные объекты в одном лямбда-выражении

3 голосов
/ 06 мая 2019

У меня есть список объектов.Сначала мне нужно отсортировать его по типу.Чем по лицу.В конце суммируйте все количества: 

      class Coin{
            String type;
            BigInteger faceValue;
            BigInteger quantity;
...
       }

            List<Coin> coins = new ArrayList<>();
            coins.add(new Coin("USD", 1, 150));
            coins.add(new Coin("USD", 1, 6));
            coins.add(new Coin("USD", 1, 60));
            coins.add(new Coin("USD", 2, 100));
            coins.add(new Coin("USD", 2, 100));
            coins.add(new Coin("CAD", 1, 111));
            coins.add(new Coin("CAD", 1, 222));

Список результатов должен содержать только 3 новых объекта монет: 

Coin("USD", 1 , 216)
Coin("USD", 2 , 200)
Coin("CAD", 1 , 333)

Как это можно записать только в одном лямбда-выражении?

Ответы [ 2 ]

1 голос
/ 06 мая 2019

Вы можете решить это, используя Collectors.toMap как: 

public List<Coin> groupedCoins(List<Coin> coins) {
    return new ArrayList<>(
            coins.stream()
                    .collect(Collectors.toMap(
                            coin -> Arrays.asList(coin.getType(), coin.getFaceValue()), Function.identity(),
                            (coin1, coin2) -> {
                                BigInteger netQ = coin1.getQuantity().add(coin2.getQuantity());
                                return new Coin(coin1.getType(), coin1.getFaceValue(), netQ);
                            }))
                    .values());
}

или более сложную группировку по одному вкладышу и суммируя как: 

public List<Coin> groupedAndSummedCoins(List<Coin> coins) {
    return coins.stream()
            .collect(Collectors.groupingBy(Coin::getType,
                    Collectors.groupingBy(Coin::getFaceValue,
                            Collectors.reducing(BigInteger.ZERO, Coin::getQuantity, BigInteger::add))))
            .entrySet()
            .stream()
            .flatMap(e -> e.getValue().entrySet().stream()
                    .map(a -> new Coin(e.getKey(), a.getKey(), a.getValue())))
            .collect(Collectors.toList());
}
0 голосов
/ 06 мая 2019

Начиная с Java 8:

по типу: 

Map<String, List<Coin>> coinlistGrouped =
    coins.stream().collect(Collectors.groupingBy(w -> w.type));

по номиналу: 

Map<String, List<Coin>> coinlistGrouped =
    coins.stream().collect(Collectors.groupingBy(w -> w.faceValue));
  • Вы можете использовать их оба для группировки по типу, а затем по номиналу
...