Давайте использовать пользовательский коллекционер, с которым мы передаем все материалы извне. Коллекционер получит рейтинг и конструктор от вызывающего. Мы хотели бы назвать это так:
public List<Person> rank(List<Person> people) {
return people
.stream()
.sorted(Comparator.<Person>comparingDouble(x -> x.profit).reversed())
.collect(new IntegerRankingCollector<>(
Comparator.comparingDouble(p -> p.profit), // how to differentiate rankings
p -> p.rank, // where to get rank for an element which was already ranked
(p, rank) -> new Person(rank, p.profit) // how to create an element from another element values and a rank
));
}
Этот коллектор может быть реализован как Collectors.toList()
, но с помощью метода аккумулятора, который:
- получает звание предыдущего элемента
- увеличивает его, если ток Ранг элемента должен быть отличным от предшествующий элемент ранг
- создает элемент в новом ранге
Вот как это выглядит, и он должен работать для упорядоченных потоков:
public class IntegerRankingCollector<T> implements Collector<T, List<T>, List<T>> {
...
public IntegerRankingCollector(Comparator<? super T> comparator, Function<T, Integer> ranker, BiFunction<T, Integer, T> creator) {
this.comparator = comparator;
this.ranker = ranker;
this.creator = creator;
}
@Override
public BiConsumer<List<T>, T> accumulator() {
return (list, current) -> {
ArrayList<T> right = new ArrayList<>();
right.add(creator.apply(current, 1));
combiner().apply(list, right);
};
}
@Override
public BinaryOperator<List<T>> combiner() {
return (left, right) -> {
int rankAdjustment = getRankAdjustment(left, right);
for (T t : right)
left.add(creator.apply(t, rankAdjustment + ranker.apply(t)));
return left;
};
}
private int getRankAdjustment(List<T> left, List<T> right) {
Optional<T> lastElementOnTheLeft = optGet(left, left.size() - 1);
Optional<T> firstElementOnTheRight = optGet(right, 0);
if (!lastElementOnTheLeft.isPresent() || !firstElementOnTheRight.isPresent())
return 0;
else if (comparator.compare(firstElementOnTheRight.get(), lastElementOnTheLeft.get()) == 0)
return ranker.apply(lastElementOnTheLeft.get()) - 1;
else
return ranker.apply(lastElementOnTheLeft.get());
}
private Optional<T> optGet(List<T> list, int index) {
if (list == null || list.isEmpty())
return Optional.empty();
else
return Optional.of(list.get(index));
}
...
}
Для полноты, это полный код класса. Я скопировал остальное с Collectors.toList
:
public class IntegerRankingCollector<T> implements Collector<T, List<T>, List<T>> {
private static final Set<Characteristics> CHARACTERISTICSS = Collections.unmodifiableSet(EnumSet.of(Characteristics.IDENTITY_FINISH));
private Comparator<? super T> comparator;
private BiFunction<T, Integer, T> creator;
private Function<T, Integer> ranker;
public IntegerRankingCollector(Comparator<? super T> comparator, Function<T, Integer> ranker, BiFunction<T, Integer, T> creator) {
this.comparator = comparator;
this.ranker = ranker;
this.creator = creator;
}
@Override
public BiConsumer<List<T>, T> accumulator() {
return (list, current) -> {
ArrayList<T> right = new ArrayList<>();
right.add(creator.apply(current, 1));
combiner().apply(list, right);
};
}
@Override
public BinaryOperator<List<T>> combiner() {
return (left, right) -> {
int rankAdjustment = getRankAdjustment(left, right);
for (T t : right)
left.add(creator.apply(t, rankAdjustment + ranker.apply(t)));
return left;
};
}
private int getRankAdjustment(List<T> left, List<T> right) {
Optional<T> lastElementOnTheLeft = optGet(left, left.size() - 1);
Optional<T> firstElementOnTheRight = optGet(right, 0);
if (!lastElementOnTheLeft.isPresent() || !firstElementOnTheRight.isPresent())
return 0;
else if (comparator.compare(firstElementOnTheRight.get(), lastElementOnTheLeft.get()) == 0)
return ranker.apply(lastElementOnTheLeft.get()) - 1;
else
return ranker.apply(lastElementOnTheLeft.get());
}
private Optional<T> optGet(List<T> list, int index) {
if (list == null || list.isEmpty())
return Optional.empty();
else
return Optional.of(list.get(index));
}
@Override
public Supplier<List<T>> supplier() {
return ArrayList::new;
}
@Override
public Function<List<T>, List<T>> finisher() {
return l -> l;
}
@Override
public Set<Characteristics> characteristics() {
return CHARACTERISTICSS;
}
}
Вы можете просто сделать это, используя 'Comparator'. Подробнее о 'Comparator & Comparable' в Java: https://docs.oracle.com/javase/7/docs/api/java/util/Comparator.html – user2004685
@ user2004685 с помощью' Comparator' Я могу, например, отсортировать свою коллекцию но как я могу установить ранги на свои элементы? –
Ваш пример для меня не имеет смысла. Ваш единственный конструктор параметров устанавливает ранг в '0', так как объекты' Person' волшебным образом получают ранг, когда вы их сортируете? – azurefrog