Применить поток, чтобы отфильтровать все элементы, удовлетворяющие условию, кроме одного

Question

У меня есть следующий класс:

public class Offer {

    private final OfferType type;
    private final BigDecimal price;

    // constructor, getters and setters
}

и тип enum:

public enum OfferType {
    STANDARD, BONUS;
}

Мой пример использования: при наличии списка предложений в качестве входных данных я хочу отфильтровать все стандартные, кроме самого дешевого. Так что для следующих входных данных

List<Offer> offers = Arrays.asList(new Offer(OfferType.STANDARD, BigDecimal.valueOf(10.0)),
            new Offer(OfferType.STANDARD, BigDecimal.valueOf(20.0)),
            new Offer(OfferType.STANDARD, BigDecimal.valueOf(30.0)),
            new Offer(OfferType.BONUS, BigDecimal.valueOf(5.0)),
            new Offer(OfferType.BONUS, BigDecimal.valueOf(5.0)));

Ожидаю следующий результат

[Offer [type=STANDARD, price=10.0], Offer [type=BONUS, price=5.0], Offer [type=BONUS, price=5.0]]

Существует ли однострочный оператор (с использованием потоков или любой сторонней библиотеки), который позволяет это сделать?

Answer 1

Вот два потока, которые:

1. Группировка по типу предложения
2. преобразует предложения каждой группы в поток
3. выбирает standard предложения, сортирует их и ограничивает 1 элементом (мин. По цене)
4. Объединяет два потока

Код выглядит так:

List<Offer> result = offers.stream()
        .collect(Collectors.groupingBy(Offer::getType))
        .entrySet()
        .stream()
        .flatMap(entry -> entry.getKey() == OfferType.STANDARD ? 
                            entry.getValue().stream()
                            .sorted(Comparator.comparing(Offer::getPrice))
                            .limit(1)
                          :  entry.getValue().stream())
        .collect(Collectors.toList());

Answer 2

Не с одной потоковой операцией, хотя:

List<Offer> some = offers.stream()
                         .filter(x -> x.getType() != OfferType.STANDARD)
                         .collect(Collectors.toCollection(ArrayList::new));

offers.stream()
      .filter(x -> x.getType() == OfferType.STANDARD)
      .min(Comparator.comparing(Offer::getPrice))
      .ifPresent(some::add);

Если вы обнаружите, что делаете это много, возможно, вы можете запустить собственный коллектор:

 public static Collector<Offer, ?, List<Offer>> minCollector() {
    class Acc {

        Offer min = null;
        List<Offer> result = new ArrayList<>();

        void add(Offer offer) {
            if (offer.getType() == OfferType.STANDARD) {
                if (min == null) {
                    min = offer;
                } else {
                    min = offer.getPrice()
                               .compareTo(min.getPrice()) > 0 ? min : offer;
                }
            } else {
                result.add(offer);
            }
        }

        Acc combine(Acc another) {
            this.min = reduceMin(this.min, another.min);
            result.addAll(another.result);
            return this;
        }

        List<Offer> finisher() {
            result.add(min);
            return result;
        }

        private Offer reduceMin(Offer left, Offer right) {
            return Collections.min(Arrays.asList(left, right),
                                   Comparator.nullsLast(Comparator.comparing(Offer::getPrice)));
        }
    }

    return Collector.of(Acc::new, Acc::add, Acc::combine, Acc::finisher);
}

И использование будет:

List<Offer> result = offers.stream()
                           .collect(minCollector());

Answer 3

Существует ли однострочный оператор (с использованием потоков или сторонних библиотека) что позволяет сделать это?

Делай вещи в два раза, это было бы более читабельно.

1) Рассчитайте самую низкую цену для предложений типа Standard:

Optional<Offer> minPriceOffer = 
offers.stream()
      .filter(o -> o.getType() == OfferType.STANDARD)
      .min(Comparator.comparing(Offer::getPrice));

2) Исключить Standard предложений с этой ценой в собранном списке:

List<Offer> offersFiltered = 
offers.stream()
      .filter(o -> {  
               if (o.getType() == OfferType.STANDARD                         
                    && !o.getPrice().equals(minPriceOffer.get().getPrice())) 
                  return false;
               // else
               return true;
             }
       )
      .collect(toList();

Answer 4

List<Offer> result = offers.stream().filter(e -> OfferType.BONUS == e.getType()).collect(toList());
offers.stream().filter(e -> OfferType.STANDARD == e.getType()).findAny().ifPresent(result::add);