开发记录:关于Java Stream,涉及遍历、分组以及list转map、list字段提取
简介和特点
Java 8 API添加了一个新的抽象称为流Stream,可以让你以一种声明的方式处理数据。
这种风格将要处理的元素集合看作一种流, 流在管道中传输, 并且可以在管道的节点上进行处理, 比如筛选, 排序,聚合等。
元素流在管道中经过中间操作(intermediate operation)的处理,最后由最终操作(terminal operation)得到前面处理的结果。
上述流程转换为Java代码为:
List<Integer> transactionsIds =
widgets.stream()
.filter(b -> b.getColor() == RED)
.sorted((x,y) -> x.getWeight() - y.getWeight())
.mapToInt(Widget::getWeight)
.sum();
什么是Stream?
Stream(流)是一个来自数据源的元素队列并支持聚合操作
- 元素是特定类型的对象,形成一个队列。 Java中的Stream并不会存储元素,而是按需计算。
- 数据源 流的来源。 可以是集合,数组,I/O channel, 产生器generator 等。
- 聚合操作 类似SQL语句一样的操作, 比如filter, map, reduce, find, match, sorted等。
和以前的Collection操作不同, Stream操作还有两个基础的特征:
- Pipelining: 中间操作都会返回流对象本身。 这样多个操作可以串联成一个管道, 如同流式风格(fluent style)。 这样做可以对操作进行优化, 比如延迟执行(laziness)和短路( short-circuiting)。
- 内部迭代: 以前对集合遍历都是通过Iterator或者For-Each的方式, 显式的在集合外部进行迭代, 这叫做外部迭代。 Stream提供了内部迭代的方式, 通过访问者模式(Visitor)实现。
常见的方法
-
forEach():迭代流中的每个数据
-
map():映射每个元素到对应的结果
-
Collectors类:实现了很多归约操作,例如将流转换成集合和聚合元素。Collectors 可用于返回列表或字符串。通常在 .collect(Collectors.方法)
-
filter():设置的条件过滤出元素
案例
案例一:根据医院编号,查询医院所有科室列表(并封装)
collect分组
- 实体类:
DepartmentVo
@Data
@ApiModel(description = "Department")
public class DepartmentVo {
@ApiModelProperty(value = "科室编号")
private String depcode;
@ApiModelProperty(value = "科室名称")
private String depname;
@ApiModelProperty(value = "下级节点")
private List<DepartmentVo> children;
}
Department
@Data
@ApiModel(description = "Department")
@Document("Department")
public class Department extends BaseMongoEntity {
private static final long serialVersionUID = 1L;
@ApiModelProperty(value = "医院编号")
@Indexed //普通索引
private String hoscode;
@ApiModelProperty(value = "科室编号")
@Indexed(unique = true) //唯一索引
private String depcode;
@ApiModelProperty(value = "科室名称")
private String depname;
@ApiModelProperty(value = "科室描述")
private String intro;
@ApiModelProperty(value = "大科室编号")
private String bigcode;
@ApiModelProperty(value = "大科室名称")
private String bigname;
}
-
stream流的功能代码
List
departmentList:某指定医院的所有科室列表
使用collect方法,对departmentList中所有Department对象进行分组,按照大科室的code分组,即Department中的bigcode字段。相同bigcode分在一个list中,key则是bigcode
//根据大科室编号 bigcode 分组,每个value是一个大科室list,其中成员是所对应的小科室对象
Map<String, List<Department>> deparmentMap = departmentList.stream().collect(Collectors.groupingBy(Department::getBigcode));
返回的数据,一部分:
{
a4e171f4cf9b6816acdfb9ae62c414d7=[Department(hoscode=1000_0, depcode=200040878, depname=多发性硬化专科门诊, intro=多发性硬化专科门诊, bigcode=a4e171f ...), Department(...), Department(...) ....], 0551a547cc19d3d09f2e57bd2931b7d0=[Department ......], ...
}
- 完整代码
//根据医院编号,查询医院所有科室列表(并封装)
@Override
public List<DepartmentVo> findDeptTree(String hoscode) {
//###########################################################
//创建list集合,用于最终数据封装
List<DepartmentVo> result = new ArrayList<>(); // DepartmentVo实体类,与数据库Department表建立映射关系
//根据医院编号,查询医院所有科室信息
Department departmentQuery = new Department();
departmentQuery.setHoscode(hoscode); // 设置科室表中的医院编号
Example example = Example.of(departmentQuery); // 传入实体
//所有科室列表 departmentList
List<Department> departmentList = departmentRepository.findAll(example); // 得到医院编号字段为指定id的所有科室
//###########################################################
//根据大科室编号 bigcode 分组,每个value是一个大科室list,其中成员是所对应的小科室对象
Map<String, List<Department>> deparmentMap =
departmentList.stream().collect(Collectors.groupingBy(Department::getBigcode));
System.out.println(deparmentMap);
//###########################################################
//遍历map集合 deparmentMap,遍历每一个大科室:对每一个大科室重新进行封装,包含其下级子科室
for(Map.Entry<String,List<Department>> entry : deparmentMap.entrySet()) {
//大科室编号
String bigcode = entry.getKey(); // 即 key
//大科室编号对应的所有下级科室数据,已经建立好bigcode对应下级科室的列表
List<Department> deparment1List = entry.getValue();
//封装大科室
DepartmentVo departmentVo1 = new DepartmentVo();
departmentVo1.setDepcode(bigcode);
departmentVo1.setDepname(deparment1List.get(0).getBigname()); // 每一个下级科室,都有对应的大科室名称,随便选一个子科室获取即可
//封装小科室,将数据库中的科室数据重新封装成DepartmentVo类型,方便之后前端使用
List<DepartmentVo> children = new ArrayList<>();
for(Department department: deparment1List) {
DepartmentVo departmentVo2 = new DepartmentVo();
departmentVo2.setDepcode(department.getDepcode());
departmentVo2.setDepname(department.getDepname());
//封装到list集合
children.add(departmentVo2);
}
//把小科室list集合放到大科室children里面
departmentVo1.setChildren(children);
//放到最终result里面
result.add(departmentVo1);
}
//返回
return result;
}
案例二:获取list集合,使用遍历,重新对集合中的元素进行封装
foreach
// 创建example对象
Example<Hospital> example = Example.of(hospital, matcher);
// 调用方法实现查询,从MongoDB中,获取对象
Page<Hospital> pages = hospitalRepository.findAll(example, pageable);
// 获取查询list集合,遍历进行医院等级封装
pages.getContent().stream().forEach(item -> { // HosType 医院等级,如三甲
this.setHospitalHosType(item); // item 传入的是 Hospital 对象
});
案例三:将List类型转换为Map类型
Collectors.toMap
List<BookingScheduleRuleVo> scheduleVoList = aggregateResult.getMappedResults(); // List
// List转Map,以workdate为key,value是BookingScheduleRuleVo类型的对象
Map<Date, BookingScheduleRuleVo> scheduleVoMap = new HashMap<>();
if(!CollectionUtils.isEmpty(scheduleVoList)) {
scheduleVoMap = scheduleVoList.stream().
collect(
Collectors.toMap(BookingScheduleRuleVo::getWorkDate,
BookingScheduleRuleVo -> BookingScheduleRuleVo));
}
案例四:从List
stream().map().collect(Collectors.toList())
// List<OrderCountVo> orderCountVoList
//获取x需要数据 ,将OrderCountVo中的date过滤,并形成日期列表
List<String> dateList = orderCountVoList.stream().map(OrderCountVo::getReserveDate).collect(Collectors.toList());
//获取y需要数据,过滤OrderCountVo中的count,并形成数量列表
List<Integer> countList =orderCountVoList.stream().map(OrderCountVo::getCount).collect(Collectors.toList());
等效于:用一个for循环,遍历orderCountVoList,并将每个item的getReserveDate()和getCount()的值,添加到各自的list中。