JAVA 8 Stream操作List和Map以及复合拼装对象
2023-02-23 本文已影响0人
暴走的狐狸
1 程序员对什么数据类型操作做多?
毋庸置疑,那就是集合类的数据类型。不管是LIST,MAP,SET或者是python的字典。
2 List的相关操作java流操作:
场景一 java8的LIST和map进行按某个条件分组,然后根据特定字段去重,最后统计去重后每组的个数
import java.util.*;
public class GroupByExample {
public static void main(String[] args) {
List<Person> list = new ArrayList<>();
list.add(new Person("John", "Male", 20));
list.add(new Person("Alice", "Female", 18));
list.add(new Person("Bob", "Male", 20));
list.add(new Person("Carol", "Female", 18));
list.add(new Person("David", "Male", 20));
// 根据某个字段分组,并返回Map<key,List<Object>>的数据格式
// Group by gender
Map<String, List<Person>> genderGroup = list.stream().collect(
Collectors.groupingBy(Person::getGender));
// 根据某个字段分组,并返回Map<key,Integer>的计数格式,也就是拿到这个key有多少条聚合的数据。
// Count the number of persons in each group
Map<String, Long> countByGender = list.stream().collect(
Collectors.groupingBy(Person::getGender, Collectors.counting()));
//多个字段进行分组,并返回Map<key,Integer>的计数格式,也就是拿多个字段进行组合分组
// Group by gender and age
Map<String, Map<Integer, List<Person>>> ageGroup = list.stream().collect(
Collectors.groupingBy(Person::getGender,
Collectors.groupingBy(Person::getAge)));
// Count the number of persons in each gender and age group
Map<String, Map<Integer, Long>> countByGenderAndAge = list.stream().collect(
Collectors.groupingBy(Person::getGender,
Collectors.groupingBy(Person::getAge, Collectors.counting())));
//按性别分组,然后根据name去重
// Group by gender and remove duplicates based on name
Map<String, List<Person>> distinctNameByGender = list.stream().collect(
Collectors.groupingBy(Person::getGender,
Collectors.collectingAndThen(
toCollection(() ->
new TreeSet<>(Comparator.comparing(Person::getName))
),ArrayList::new)
));
}
}
POJO对象
class Person {
private String name;
private String gender;
private int age;
public Person(String name, String gender, int age) {
this.name = name;
this.gender = gender;
this.age = age;
}
// 省略setter and getter
}
JAVA LIST 多个字段 group by的时候,我一般喜欢封装一个方法,加一个连接符来处理。比如我有一个对象,叫
ActualSortingLog,当前分拣日志, 我希望根据sortingTime和PipeLine进行分组。那就可以创建一个叫
fetchGroupKey的方法
/**
*
* 根据sortingTime和PipeLine进行分组
* @param actualSortingLog
* @return
*/
private String fetchGroupKey(ActualSortingLog actualSortingLog) {
return actualSortingLog.getSortingTime() +"#"+ user.getPipeLine();
}
这样好处是解耦,也方便扩展,代码也可读。进一步,如果希望对字段做一些处理,再分组,也就简单很多。比如
这边进一步,希望按分钟和pipeLine字段的前3位聚合,同时时间格式变为yyyyMMddHHmm这种,则代码如下:
/**
* 按分钟线加pipeline前三位进行聚合
*
* @param actualSortingInfoDTO
* @return
*/
private String makeGroupKeyMinuteWithDeviceCode(ActualSortingInfoDTO actualSortingInfoDTO) {
DateTimeFormatter dateTimeFormatter = DateTimeFormatter.ofPattern("yyyyMMddHHmm");
return dateTimeFormatter.format(LocalDateTimeUtils.longToLocalTime(actualSortingInfoDTO.getSortTime())) + SEPARATOR +
filterDeviceCode(actualSortingInfoDTO.getPipeline());
}
当然这里我又封装了时间工具LocalDateTimeUtils类和设备编码切割类filterDeviceCode,是因为这些细碎的逻辑在后续的切割和分组中会经常用到。分开也有利于测试和管理。这里就不展开讲。
再来说一说List的去重逻辑,首先是简单去重
@Test
@DisplayName("list去重测试")
void testDuplicate() {
ActualSortingInfoDTO mockData = mockData("001223", "1", 1677037037000L);
ActualSortingInfoDTO mockData2 = mockData("002331", "2", 1677037037000L);
// 模拟一个重复的sortingId,应该会去重
ActualSortingInfoDTO duplicateId = mockData("002331", "2", 1677037037001L);
List<ActualSortingInfoDTO> list = new ArrayList<>();
list.add(mockData);
list.add(mockData2);
list.add(duplicateId);
List<ActualSortingInfoDTO> distinctList = list.stream().collect(Collectors.collectingAndThen(
Collectors.toCollection(simpleTreeSetSupplier()),
ArrayList::new));
log.info("简单去重的数据" + distinctList);
// 复杂逻辑去重,比如我希望根据pipeLine的前三位的值是001来去重
List<ActualSortingInfoDTO> complexDistinctList = list.stream().collect(Collectors.collectingAndThen(
Collectors.toCollection(distinctPipeLine()),ArrayList::new
));
log.info("复杂去重的数据" + complexDistinctList);
// list to map,复杂逻辑去重,比如我希望根据pipeLine的前三位的值是001和sortingId不等于2来去重,应该是保留第一条的数据mockData
Map<String, List<ActualSortingInfoDTO>> distinctMap = ListStreamUtil.group(complexDistinctList,this::makeGroupKeyMinuteWithDeviceCode);
Assertions.assertEquals(
distinctMap.get(makeGroupKeyMinuteWithDeviceCode(mockData)).get(0).getPipeline(),"001223");
Assertions.assertEquals(
distinctMap.get(makeGroupKeyMinuteWithDeviceCode(mockData)).get(0).getSortingId(),"1");
distinctMap.forEach((k, v) -> log.info("分组后:" + k + " " + v));
}
这里也建议对复杂的去重方法进行封装,比如我这边封装了一个方法,叫distinctPipeLine,后续就可以自定义各种去重逻辑了。
自定义去重方法如下:
private Supplier<TreeSet<ActualSortingInfoDTO>> distinctPipeLine() {
return () -> new TreeSet<>(
Comparator.comparing(actualSortingInfoDTO ->
actualSortingInfoDTO.getPipeline().equals("001")
&& !Objects.equals(actualSortingInfoDTO.getSortingId(), "2")));
}