java8lambda表达式对集合分组并且排序（记一次性能优化案例）

最近遇到个需求是这样的：

写一个类似于通讯录的简单功能，要求是首先按照部门显示列表，然后点击部门下拉显示部门里面的人员信息，且人员信息按照职位排序。

先分析数据结构：

最外层是部门，部门里面是员工，且项目是统一返回数据格式。

{
    "code": 0,
    "data": [
        {
            "name": "部门名称",
            "userData": [
                {
                    "deptName": "部门名称",
                    "info": "部门名称+职位",
                    "phone": "电话号码",
                    "userName": "员工姓名"
                }
            ]
        },
        {
            "name": "部门名称",
            "userData": [
                {
                    "deptName": "部门名称",
                    "info": "部门名称+职位",
                    "phone": "电话号码",
                    "userName": "员工姓名"
                }
            ]
        }
    ],
    "message": "success"
}

数据格式大概就是这样。然后分析写法。正常来说需要用sql将这些数据一次性分组并返回出来，本人没有找到合适的方法。然后打算用最原始的方法，就是先查出所有的部门，再遍历部门查询出部门里面所有的人员，再将数据返回。下面贴上代码。

List<PageData> deptAll = (List<PageData>) dao.findForList("AddressListMapper.querydeptAll", pageData);
        //创建list保存所有部门和部门人员
        List<PageData> deptPersonnelAll = new ArrayList<PageData>();
        //遍历获取到的部门查询相对应得人员信息
/*        for (HashMap<String, Object> dept : deptAll) {
            //获取部门名称
            String deptName = String.valueOf(dept.get("name"));
            //获取部门code
            String deptCode = String.valueOf(dept.get("code"));
            //根据部门code查询部门对应的人员
            PageData pageDatap = new PageData();
            pageDatap.put("deptid", deptCode);
            ArrayList<PageData> personnelList = queryPersonnel(pageDatap);
            //将数据存封装为map并存进list
            PageData deptSingle = new PageData();
            deptSingle.put("name", deptName);
            deptSingle.put("userData", personnelList);
            deptPersonnelAll.add(deptSingle);
        }*/
        deptAll.stream().forEach(dept -> {
            //获取部门名称
            String deptName = String.valueOf(dept.get("name"));
            //获取部门code
            String deptCode = String.valueOf(dept.get("code"));
            //根据部门code查询部门对应的人员
            PageData pageDatap = new PageData();
            pageDatap.put("deptid", deptCode);
            ArrayList<PageData> personnelList = null;
            try {
                personnelList = queryPersonnel(pageDatap);
            } catch (Exception e) {
                e.printStackTrace();
            }
            //将数据存封装为map并存进list
            PageData deptSingle = new PageData();
            deptSingle.put("name", deptName);
            deptSingle.put("userData", personnelList);
            deptPersonnelAll.add(deptSingle);
        });
        return deptPersonnelAll;

上面遍历集合写了两种方式，一种是以前的for循环写法，另一种是用表达式。

但是这种方式会出现一个性能问题，比如说有10000个部门，就需要查询10000次数据库。这样会增大数据库的压力和查询时间。所以这样就需要优化。

优化：

优化的方式就是在数据一次查出所有的人员信息，并将人员信息和部门信息进行关联，这样就能知道每个人所对应的部门。然后用java代码可以对数据进行分组。

贴代码：

public List<Map<String, Object>> getAllPersonnel(PageData pageData) throws Exception {
    //查询类型对应的人员
    List<AddressList> addressList = (List<AddressList>) dao.findForList("AddressListMapper.getAllPersonnel", pageData);
    //lambda表达式对数据进行分组
    Map<String, List<AddressList>> collect = addressList.stream().collect(Collectors.groupingBy(AddressList::getDeptName));
    //处理分组后的数据
    //创建一个list集合存储所有数据
    List<Map<String, Object>> maps = new ArrayList<>();
    //遍历分组后的数据，进行组装
    collect.forEach((k, v) -> {
        HashMap<String, Object> map = new HashMap<>();
        map.put("name", k);
        map.put("userData", v);
        maps.add(map);
    });
    return maps;
}

这边分组是用的表达式分组的，定义一个对象，将数据库查出来的数据封装进去，然后按照属性进行分组。这样就可以避免多次请求数据库。

处理排序：

这样优化完后还有一个排序的问题。

解决办法：

上面代码中groupingBy有三个参数

第一个参数：分组按照什么分类

第二个参数：分组最后用什么容器保存返回值（默认是HashMap::new）

第三个参数：按照第一个参数分类后，对应的分类的结果如何收集。（Collectors.toList）

这边主要问题是数据库里面将数据排序好之后返回到代码里面。代码里面对数据进行分组之后，数据的顺序发生了改变。

map里面的有序集合LinkedHashMap。这里可以使用这个集合作为分组后用来保存返回值的容器。这样就解决了分组后的排序问题。