问题

我有一堆方法基本上做同样的事情:根据不同方法返回的值选择类的顶部N实例,所有返回双值.

例如,对于实现以下接口的类的对象:

 interface A {
    Double getSalary();
    Double getAge();
    Double getHeight();
}
 

我想选择每个方法返回的值最高的N对象.

现在我有3种方法:

 List<A> getTopItemsBySalary(List<A> elements);
List<A> getTopItemsByAge(List<A> elements);
List<A> getTopItemsByHeight(List<A> elements);
 

这个机构:

 List<A> getTopItemsBySalary(List<A> elements, int n) {
    return elements.stream()
              .filter(a -> a.getSalary() != null)                 
              .sorted(Comparator.comparingDouble(A::getSalary).reversed())
              .limit(n)
              .collect(Collectors.toList());
}
 

如何传递方法,只有一个方法?

  最佳答案

您可以使用FunctionA转换为Double,如:

 List<A> getTopItems(List<A> elements, Function<A, Double> mapper, int n) {
    return elements.stream()
              .filter(a -> null != mapper.apply(a))                 
              .sorted(Comparator.<A>comparingDouble(a -> mapper.apply(a))
                      .reversed())
              .limit(n)
              .collect(Collectors.toList());
}
 

您可以使用以下方法调用它:

 List<A> top10BySalary = getTopItems(list, A::getSalary, 10);
List<A> top10ByAge = getTopItems(list, A::getAge, 10);
 

如果您的getter总是返回非non-null,那么ToDoubleFunction是一个更好的类型(但如果您的Double返回值可能为null,它将不起作用):

 List<A> getTopItems(List<A> elements, ToDoubleFunction<A> mapper, int n) {
    return elements.stream()
              .sorted(Comparator.comparingDouble(mapper).reversed())
              .limit(n)
              .collect(Collectors.toList());
}
 

  相同标签的其他问题

java