Java排序List時，若排序元素的屬性可能為null，則可以在Comparator的實作中處理null。
例如下面是List中的元素Employee。
Employee
public class Employee {
private String name;
private Integer age;
// getters and setters
public String toString() {
return String.format("name=%s, age=%d", name, age);
}
}
使用下面的靜態方法compare()來處理null。第一及第二個參數為比較對像，第三個boolean參數決定null要排前面還是排後面。
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
public class Main {
public static void main(String[] args) {
List employeeList = Arrays.asList(
new Employee("mary", 22),
new Employee("john", 16),
new Employee(null, 34),
new Employee("alice", null));
employeeList = employeeList.stream().sorted((o1, o2) -> {
int result = compare(o1.getName(), o2.getName(), true);
if (result != 0) {
return result;
}
result = compare(o1.getAge(), o2.getAge(), false);
return result;
}).collect(Collectors.toList());
employeeList.forEach(System.out::println);
/*
name=null, age=34
name=alice, age=null
name=john, age=16
name=mary, age=22
*/
}
public static <T extends Comparable<T>> int compare(T a, T b, boolean nullFirst) {
if (a == null) {
return (b == null) ? 0 : (nullFirst ? -1 : 1);
} else if (b == null) {
return nullFirst ? 1: -1;
} else {
return a.compareTo(b);
}
}
}
或是用Comparator.nullFirst()或Comparator.nullLast()來達到更lambda的寫法。
employeeList = employeeList.stream()
.sorted(Comparator
.comparing(Employee::getName, Comparator.nullsFirst(Comparator.naturalOrder()))
.thenComparing(Employee::getAge, Comparator.nullsLast(Comparator.naturalOrder())))
.collect(Collectors.toList());
沒有留言:
張貼留言