java stream distinct()如何按一个或多个指定对象字段进行去重
作者：陈賝
这篇文章主要介绍了java stream distinct()如何按一个或多个指定对象字段进行去重问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
写在前面
本来以为stream应该有类似这种语法，想这样写的，可惜事与愿违，需要开发者换种思路。
List<ProjectInfoVo> acceptances = vo.stream().filter(distinct(b -> b.getProjectId())).collect(Collectors.toList());
自带的 distinct 似乎只能将所有字段都相同的数据去重，若不是还请大佬们指导。
List<ProjectInfoVo> acceptances = vo.stream().distinct().collect(Collectors.toList());
方法一：自带方法
Comparator.comparing(p -> p.get***())
- 单个字段
String sql = "";
List<ProjectInfoVo> vo = jdbcTemplate.query(sql, new BeanPropertyRowMapper<>(ProjectInfoVo.class));
ArrayList<ProjectInfoVo> collect = vo.stream().collect(Collectors.collectingAndThen(
Collectors.toCollection(() -> new TreeSet<>(
Comparator.comparing(p -> p.getProjectId()))), ArrayList::new));
- 多个字段
String sql = "";
List<ProjectInfoVo> vo = jdbcTemplate.query(sql, new BeanPropertyRowMapper<>(ProjectInfoVo.class));
ArrayList<ProjectInfoVo> collect = vo.stream().collect(Collectors.collectingAndThen(
Collectors.toCollection(() -> new TreeSet<>(
Comparator.comparing(p -> p.getProjectId()+";"+p.getMember()))), ArrayList::new));
方法二：自定义方法
Comparator.comparing(p -> p.get***())
1.自定义方法类——distinctByKey
public class StreamUtils {
/**
* 按某个指定字段去重
*
* @param keyExtractor 需要去重的字段
* @return java.util.function.Predicate<T>
* @author 陈賝
* @date 2022/9/2 13:49
*/
public static <T> Predicate<T> distinctByKey(Function<? super T, ?> keyExtractor) {
Map<Object, Boolean> seen = new ConcurrentHashMap<>();
return t -> seen.putIfAbsent(keyExtractor.apply(t), Boolean.TRUE) == null;
}
}
2.使用示例
- 单个字段
List<ProjectInfoVo> acceptances = vo.stream() .filter(StreamUtils.distinctByKey(b -> b.getProjectId())) .collect(Collectors.toList());
- 多个字段
List<ProjectInfoVo> acceptances = vo.stream() .filter(StreamUtils.distinctByKey(b -> b.getProjectId())) .filter(StreamUtils.distinctByKey(b -> b.getMember())) .collect(Collectors.toList());
方法三：Stream API
1.Collectors.toMap
- 单个字段
private static List<CompanyVo> mergeAndDistinct(List<CompanyVo> companyVoList) {
// return companyVoList.stream().collect(Collectors.collectingAndThen(
// Collectors.toCollection(() -> new TreeSet<>(
// Comparator.comparing(CompanyVo::getId))), ArrayList::new));
return new ArrayList<>(companyVoList.stream()
.collect(Collectors.toMap(
CompanyVo::getId, // 以 id 作为 key
companyVo -> companyVo, // 保留 CompanyVo 对象
(existing, replacement) -> existing // 如果有重复的 id，保留已有的对象
)).values());
}
- 多个字段
private static List<CompanyVo> mergeAndDistinct(List<CompanyVo> companyVoList) {
return new ArrayList<>(companyVoList.stream()
.collect(Collectors.toMap(
companyVo -> companyVo.getId() + "-" + companyVo.getCode(), // 使用 id 和 code 作为 key
companyVo -> companyVo, // 保留 CompanyVo 对象
(existing, replacement) -> existing // 如果有重复的 id，保留已有的对象
)).values());
}
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。