public class Java8StreamTest {
public static class Book{
private String id;
private String name;
public Book(String id, String name) {
this.id = id;
this.name = name;
}
public String getId() {
return id;
}
public void setId(String id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
@Override
public String toString() {
return "Book{" +
"id='" + id + '\'' +
", name='" + name + '\'' +
'}';
}
}
@Test
public void testUnique(){
List books = Lists.newArrayList(new Book("1","1"),new Book("2","2"),new Book("3","3"),new Book("2","2"));
//使用TreeSet去重
List unique1 = books.stream().collect(
collectingAndThen(toCollection(() -> new TreeSet<>(Comparator.comparing(o -> o.getId()))),
ArrayList::new));
System.out.println(unique1);
//使用map去重
List unique2 = books.stream()
.filter(distinctByKey(o -> o.getId()))
.collect(Collectors.toList());
System.out.println(unique2);
}
public static Predicate distinctByKey(Function super T, Object> keyExtractor) {
Map seen = new ConcurrentHashMap<>();
System.out.println("这个函数将应用到每一个item");
return t -> seen.putIfAbsent(keyExtractor.apply(t), Boolean.TRUE) == null;
}
}