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

╰半夏微凉° 2024-04-17 06:29 65阅读 0赞

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

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

先分析数据结构：

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

{
        "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。这里可以使用这个集合作为分组后用来保存返回值的容器。这样就解决了分组后的排序问题。