Как использовать два фильтра в потоке для разных преобразований - PullRequest
Новая
       11

Как использовать два фильтра в потоке для разных преобразований

0 голосов
/ 05 ноября 2018

Мне нужно выполнять преобразования только для определенного условия. Я делаю это преобразование: 

// filter 1: less date - group by max date by groupId
        List<Info> listResult = new ArrayList<>(listInfo.stream()
                .filter(info -> info.getDate().getTime() < date.getTime())
                .collect(Collectors.groupingBy(Info::getGroupId, Collectors.collectingAndThen(
                        Collectors.reducing((Info i1, Info i2) -> i1.getDate().getTime() > i2.getDate().getTime() ? i1 : i2),
                        Optional::get))).values());

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

// filter 2: more date - nothing change in list
        List<Info> listMoreByDate = listInfo.stream()
                .filter(info -> info.getDate().getTime() >= date.getTime())
                .collect(Collectors.toList());

Далее, чтобы объединить эти два фильтра - я объединяю два списка: 

listResult.addAll(listMoreByDate);

У меня вопрос: можно ли это сделать в одном потоке? Поскольку фильтр 2 абсолютно бесполезен, он просто возвращает список для этого условия.

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

Мой полный код: 

import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.stream.Collectors;

public class App {
    public static void main(String[] args) throws ParseException {
        Info info1 = new Info(1L, getDateFromStr("2018-02-02T10:00:00"), 3L);
        Info info2 = new Info(2L, getDateFromStr("2018-02-02T12:00:00"), 3L);
        Info info3 = new Info(3L, getDateFromStr("2018-02-05T12:00:00"), 6L);
        Info info4 = new Info(4L, getDateFromStr("2018-02-05T10:00:00"), 6L);

        Date date = getDateFromStr("2018-02-03T10:10:10");

        List<Info> listInfo = new ArrayList<>();
        listInfo.add(info1);
        listInfo.add(info2);
        listInfo.add(info3);
        listInfo.add(info4);

        // filter 1: less date - group by max date by groupId
        List<Info> listResult = new ArrayList<>(listInfo.stream()
                .filter(info -> info.getDate().getTime() < date.getTime())
                .collect(Collectors.groupingBy(Info::getGroupId, Collectors.collectingAndThen(
                        Collectors.reducing((Info i1, Info i2) -> i1.getDate().getTime() > i2.getDate().getTime() ? i1 : i2),
                        Optional::get))).values());

        // filter 2: more date - nothing change in list
        List<Info> listMoreByDate = listInfo.stream()
                .filter(info -> info.getDate().getTime() >= date.getTime())
                .collect(Collectors.toList());

        listResult.addAll(listMoreByDate);

        System.out.println("result: " + listResult);
    }

    private static Date getDateFromStr(String dateStr) throws ParseException {
        return new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss").parse(dateStr);
    }
}

class Info {
    private Long id;
    private Date date;
    private Long groupId;

    public Info(Long id, Date date, Long groupId) {
        this.id = id;
        this.date = date;
        this.groupId = groupId;
    }

    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public Date getDate() {
        return date;
    }

    public void setDate(Date date) {
        this.date = date;
    }

    public Long getGroupId() {
        return groupId;
    }

    public void setGroupId(Long groupId) {
        this.groupId = groupId;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        Info info = (Info) o;
        return Objects.equals(id, info.id) &&
                Objects.equals(date, info.date) &&
                Objects.equals(groupId, info.groupId);
    }

    @Override
    public int hashCode() {

        return Objects.hash(id, date, groupId);
    }

    @Override
    public String toString() {
        final StringBuilder sb = new StringBuilder("Info{");
        sb.append("id=").append(id);
        sb.append(", date=").append(date);
        sb.append(", groupId=").append(groupId);
        sb.append('}');
        return sb.toString();
    }
}

Ответы [ 3 ]

0 голосов
/ 05 ноября 2018

Я не вижу ничего проще, чем 

List<Info> listResult = Stream.concat(
    listInfo.stream()
        .filter(info -> info.getDate().getTime() < date.getTime())
        .collect(Collectors.toMap(Info::getGroupId, Function.identity(),
            BinaryOperator.maxBy(Comparator.comparing(Info::getDate))))
        .values().stream(),
    listInfo.stream()
        .filter(info -> info.getDate().getTime() >= date.getTime())
    )
    .collect(Collectors.toList());

так как эти две операции принципиально различны. Создание Map на первом шаге неизбежно, так как оно будет использоваться для идентификации предметов с равным свойством getGroupId.

Тем не менее, вы должны рассмотреть возможность перехода с Date на java.time API.

0 голосов
/ 05 ноября 2018

Другой подход (еще более многословный по определению, но гораздо менее многословный при использовании сайта) - создать пользовательский Collector: 

List<Info> listResult = listInfo.stream().collect(dateThresholdCollector(date));

где 

private static Collector<Info, ?, List<Info>> dateThresholdCollector(Date date) {
    return Collector.of(
            () -> new ThresholdInfoAccumulator(date), ThresholdInfoAccumulator::accept,
            ThresholdInfoAccumulator::combine, ThresholdInfoAccumulator::addedInfos
    );
}

и 

class ThresholdInfoAccumulator {

    private final Date date;
    private final List<Info> addedInfos = new ArrayList<>();

    ThresholdInfoAccumulator(Date date) {
        this.date = date;
    }

    List<Info> addedInfos() {
        return addedInfos;
    }

    ThresholdInfoAccumulator accept(Info newInfo) {
        if (canAdd(newInfo)) {
            addedInfos.add(newInfo);
        }
        return this;
    }

    boolean canAdd(Info newInfo) {
        if (newInfo.getDate().compareTo(date) < 0) { // lower date - max date by groupId
            return addedInfos.removeIf(addedInfo -> isEarlierDateInSameGroup(addedInfo, newInfo));
        }
        return true; // greater or equal date - no change
    }

    private boolean isEarlierDateInSameGroup(Info addedInfo, Info newInfo) {
        return addedInfo.getGroupId().equals(newInfo.getGroupId())
                && addedInfo.getDate().compareTo(newInfo.getDate()) < 0;
    }

    ThresholdInfoAccumulator combine(ThresholdInfoAccumulator other) {
        other.addedInfos().forEach(this::accept);
        return this;
    }
}

Примечание: это не будет настолько эффективно, если у вас есть огромное количество групп / информации, потому что он не группирует по getGroupId (он повторяет весь список для каждого Info, который будет добавлен ).

0 голосов
/ 05 ноября 2018

Да, вы можете объединить два условия, используя partitioningBy сборщик следующим образом: 

 List<Info> resultSet = 
      listInfo.stream()
              .collect(collectingAndThen(partitioningBy(info -> info.getDate().getTime() < date.getTime()),
                     map -> Stream.concat(map.get(true)
                            .stream()
                            .collect(toMap(Info::getGroupId,
                                    Function.identity(),
                                    (Info i1, Info i2) -> i1.getDate().getTime() > i2.getDate().getTime() ? i1 : i2))
                            .values().stream(), map.get(false).stream())
                            .collect(Collectors.toCollection(ArrayList::new))));

Это, по существу, использует коллектор partitioningBy для организации элементов таким образом, что все элементы, отвечающие критериям info.getDate().getTime() < date.getTime(), также как и там, где оно ложно, т. Е. Где info -> info.getDate().getTime() >= date.getTime() истинно в Map<Boolean, List<T>>.

Далее, мы используем коллектор collectingAndThen, чтобы применить завершающую функцию к Map<Boolean, List<T>>, возвращаемому коллектором partitioningBy, в этом случае мы объединяем результат применения логики: 

.collect(Collectors.groupingBy(Info::getGroupId, 
          Collectors.collectingAndThen(Collectors.reducing((Info i1, Info i2) -> i1.getDate().getTime() > i2.getDate().getTime() ? i1 : i2),
           Optional::get))))
.values();

который я упростил до: 

.collect(toMap(Info::getGroupId, Function.identity(), (Info i1, Info i2) -> i1.getDate().getTime() > i2.getDate().getTime() ? i1 : i2)))
.values();

с возвращенными элементами, где info.getDate().getTime() < date.getTime() вернул false (map.get(false).stream()).

Наконец, мы собираем результат в реализацию ArrayList с помощью коллектора toCollection.

...