欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-04-28 15:38:28
...

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

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

先分析数据结构:

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

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