jdk8 StreamAPI 的使用

本文将以UmsMenu对象为例来介绍Stream API的常用操作。UmsMenu是一个菜单对象，具有树形结构，对象定义如下

Stream对象的创建

Stream对象分为两种，一种串行的流对象，一种并行的流对象。

// menuList指所有菜单列表
// 为集合创建串行流对象
Stream<UmsMenu> stream = menuList.stream();
// 为集合创建并行流对象
Stream<UmsMenu> parallelStream = menuList.parallelStream();

filter

对Stream中的元素进行过滤操作，当设置条件返回true时返回相应元素。

@Test
public void filterTest(){
    //filter操作：获取所有一级菜单
    List<UmsMenu> oneLevelList = menuList.stream()
    .filter(menu -> menu.getParentId() == 0L)
    .collect(Collectors.toList());
    LOGGER.info("filter操作：{}",oneLevelList);
}

map

对Stream中的元素进行转换处理后获取，比如可以将UmsMenu对象转换成Long对象。我们经常会有这样的需求：需要把某些对象的id提取出来，然后根据这些id去查询其他对象，这时可以使用此方法。

@Test
public void mapTest(){
    //map操作：获取所有菜单的id
    List<Long> idList = menuList.stream()
    .map(menu -> menu.getId())
    .collect(Collectors.toList());
    LOGGER.info("map操作：{}",idList);
}

limit

从Stream中获取指定数量的元素。

@Test
public void limitTest(){
    //limit操作：获取前5个菜单
    List<UmsMenu> firstFiveList = menuList.stream()
            .limit(5)
            .collect(Collectors.toList());
    LOGGER.info("limit操作：{}",firstFiveList);
}

count

仅获取Stream中元素的个数。

@Test
public void countTest(){
    //count操作：获取所有一级菜单的个数
    long count = menuList.stream()
    .filter(menu -> menu.getParentId() == 0L)
    .count();
    LOGGER.info("count操作：{}",count);
}

sorted

对Stream中元素按指定规则进行排序。

@Test
public void sortedTest(){
    //sorted操作：将所有菜单按照sort字段进行排序
    List<UmsMenu> sortedList = menuList.stream()
            .sorted((menu1,menu2)->{return menu2.getSort().compareTo(menu1.getSort());})
            .collect(Collectors.toList());
    LOGGER.info("sorted操作：{}",sortedList);
}

skip

跳过指定个数的Stream中元素，获取后面的元素。

@Test
public void skipTest(){
    //skip操作：跳过前5个元素，返回后面的
    List<UmsMenu> skipList = menuList.stream()
            .skip(5)
            .collect(Collectors.toList());
    LOGGER.info("skip操作：{}",skipList);
}

用collect方法将List转成map

有时候我们需要反复对List中的对象根据id进行查询，我们可以先把该List转换为以id为key的map结构，然后再通过map.get(id)来获取对象，这样比较方便。

@Test
public void collect2mapTest(){
    //collect转map操作：将菜单列表以id为key，以菜单对象为值转换成map
    Map<Long, UmsMenu> menuMap = menuList.stream()
            .collect(Collectors.toMap(menu -> menu.getId(), menu -> menu));
    LOGGER.info("collect转map操作：{}",menuMap);
}

应用

我们经常会有返回树形结构数据的需求。比如这里的菜单，具有一级菜单、二级菜单、三级菜单这样的结构。如果我们要返回一个集合，包含菜单以及菜单下面嵌套的子菜单，使用Stream API可以很方便的解决这个问题。

注意：这里我们的菜单上下级之间以parentId来关联，parentId是指上一级菜单的id，顶级菜单的parentId为0。

定义获取树形结构的方法

我们先过滤出parentId为0的顶级菜单，然后给每个顶级菜单设置其子级菜单，covertMenuNode方法的主要用途就是从所有菜单中找出相应子级菜单。

@Override
public List<UmsMenuNode> treeList() {
    List<UmsMenu> menuList = menuMapper.selectByExample(new UmsMenuExample());
    List<UmsMenuNode> result = menuList.stream()
    .filter(menu -> menu.getParentId().equals(0L))
    .map(menu -> covertMenuNode(menu, menuList)).collect(Collectors.toList());
    return result;
}

为每个菜单设置子级菜单

这里我们使用filter操作来过滤出每个菜单的子级菜单，由于子级菜单下面可能还会有子级菜单，这里我们使用递归来解决。但是递归操作什么时候停止，这里把递归调用方法放到了map操作中去，当没有子级菜单时filter下的map操作便不会再执行，从而停止递归。

/**
 * 将UmsMenu转化为UmsMenuNode并设置children属性
 */
private UmsMenuNode covertMenuNode(UmsMenu menu, List<UmsMenu> menuList) {
    UmsMenuNode node = new UmsMenuNode();
    BeanUtils.copyProperties(menu, node);
    List<UmsMenuNode> children = menuList.stream()
            .filter(subMenu -> subMenu.getParentId().equals(menu.getId()))
            .map(subMenu -> covertMenuNode(subMenu, menuList)).collect(Collectors.toList());
    node.setChildren(children);
    return node;
}