新建测试实体类以及测试用例
@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class UserInfo implements Serializable {
private static final long serialVersionUID = 1L;
private String name;
private Integer age;
private String address;
private Double score;
}
private static List<UserInfo> getList(){
List<UserInfo> list = new ArrayList<>(3);
list.add(UserInfo.builder().name("张三").age(18).address("合肥").score(55.61).build());
list.add(UserInfo.builder().name("李四").age(15).address("南京").score(37.89).build());
list.add(UserInfo.builder().name("王五").age(20).address("上海").score(19.0).build());
list.add(UserInfo.builder().name("赵六").age(20).address("深圳").score(27.9).build());
list.add(UserInfo.builder().name("小红").age(16).address("北京").score(67.2).build());
list.add(UserInfo.builder().name("小明").age(17).address("合肥").score(50.4).build());
list.add(UserInfo.builder().name("小明").age(17).address("合肥").score(50.4).build());
return list;
}
- 根据年龄顺序排序
list = list.stream().sorted(Comparator.comparing(UserInfo::getAge, Comparator.nullsLast(Integer::compareTo))).collect(Collectors.toList());
- 根据年龄倒叙排序
list = list.stream().sorted(Comparator.comparing(UserInfo::getAge, Comparator.nullsLast(Integer::compareTo)).reversed()).collect(Collectors.toList());
- 根据字符串年龄字段倒序排序
list = list.stream()
.sorted(Comparator.comparing(item -> Double.parseDouble(item.getAge()), Comparator.nullsFirst(Double::compareTo).reversed()))
.collect(Collectors.toList());
- 根据年龄、分数顺序排序
list = list.stream().sorted(Comparator.comparing(UserInfo::getAge, Comparator.nullsLast(Integer::compareTo)).thenComparing(UserInfo::getScore, Comparator.nullsLast(Double::compareTo))).collect(Collectors.toList());
- JSONArray根据JSONObject中的某个元素进行排序
// 方式一:Collections.sort(list, new Comparator() {})
List<JSONObject> list = JSONArray.parseArray(resultArrays.toJSONString(), JSONObject.class);
Collections.sort(list, new Comparator<JSONObject>() {
//排序字段
private final String FIELD_NAME = "num";
//重写compare方法
@Override
public int compare(JSONObject obj1, JSONObject obj2) {
int val1 = obj1.getIntValue(FIELD_NAME);
int val2 = obj2.getIntValue(FIELD_NAME);
//是降序
return val2 - val1;
}
});
return JSONArray.parseArray(list.toString());
// 方式二:使用JsonArray.sort()
resultArrays.sort(Comparator.comparing(obj -> ((JSONObject) obj).getIntValue("num")).reversed());
- 截取前三个
list = list.stream().limit(3).collect(Collectors.toList());
- 根据地址过滤
list = list.stream().filter(item -> StringUtils.equals("合肥",item.getAddress())).collect(Collectors.toList());
- 简单去重
list = list.stream().filter(item -> StringUtils.equals("合肥",item.getAddress())).distinct().collect(Collectors.toList());
- 根据地址字段去重
list = list.stream().filter(item -> StringUtils.equals("合肥",item.getAddress())).filter(distinctByKey(b -> b.getAddress())).collect(Collectors.toList());
public static <T> Predicate<T> distinctByKey(Function<? super T, Object> keyExtractor) {
Map<Object, Boolean> seen = new ConcurrentHashMap<>();
return t -> seen.putIfAbsent(keyExtractor.apply(t), Boolean.TRUE) == null;
}
- 单独提取某个字段并生成集合并去重
List<String> nameList = list.stream().map(item -> item.getName()).distinct().collect(Collectors.toList());
- list某两个字段组成map,如果key值有重复会报错
Map<String,Integer> map = list.stream().collect(Collectors.toMap(UserInfo::getName, UserInfo::getAge));
- list转map,如果key值有重复取第一个
Map<String,UserInfo> map = list.stream().collect(Collectors.toMap(UserInfo::getName, Function.identity(), (key1, key2) -> key2));
- 根据名称字段分组
Map<String,List<UserInfo>> mapForName = list.stream().collect(Collectors.groupingBy(UserInfo::getName));
- 根据名称、地址多字段分组
Function<UserInfo, List<Object>> compositeKey = wlb -> Arrays.asList(wlb.getName(), wlb.getAddress());
Map<Object, List<UserInfo>> mapForNameAndAdd = list.stream().collect(Collectors.groupingBy(compositeKey, Collectors.toList()));
- 根据年龄字段获取最大值
Integer max = list.stream().map(UserInfo::getAge).max(Integer::compareTo).get();
- 根据年龄字段获取最小值
Integer min = list.stream().map(UserInfo::getAge).min(Integer::compareTo).get();
- 根据年龄字段获取最大值对应的实体类
UserInfo userInfoMax = list.stream().max(Comparator.comparing(UserInfo::getAge)).get();