使用java8的StreamAPI对集合计算进行代码重构

方法: 查询出所有部门成员中年龄大于30的员工姓名

部门对象:

员工对象:

模拟数据:

    private static List<Dept> list=new ArrayList<Dept>();

    private static List<Employee> listEmpl01=new ArrayList<Employee>();

    private static List<Employee> listEmpl02=new ArrayList<Employee>();

    static{//初始化数据

        Employee employee1=new Employee(1,"张三",25,1);

        Employee employee2=new Employee(1,"李四",32,1);

        Employee employee3=new Employee(1,"王五",25,1);

        Employee employee4=new Employee(1,"赵六",34,2);

        Employee employee5=new Employee(1,"周七",43,2);

        Employee employee6=new Employee(1,"小明",26,2);

        Employee employee7=new Employee(1,"大熊",22,2);

        listEmpl01.add(employee1);

        listEmpl01.add(employee2);

        listEmpl01.add(employee3);

        listEmpl02.add(employee4);

        listEmpl02.add(employee5);

        listEmpl02.add(employee6);

        listEmpl02.add(employee7);

        Dept dept01=new Dept(1,"研发部",listEmpl01);

        Dept dept02=new Dept(2,"人力资源部",listEmpl02);

    }

　　ok了,如果不使用stream,那么就要使用双重for循环来做了:

    public static void main(String[] args) {

        List<String> result = getNameByAge(list);

        System.out.println(result);

    }

    //原始查询方法

    public static List<String> getNameByAge(List<Dept> list){

        List<String> result=new ArrayList<String>();

        for (Dept dept:list) {

            List<Employee> employees = dept.getEmployees();

            for (Employee employee:employees){

                if(employee.getAge()>30){

                    result.add(employee.getName());

                }

            }

        }

        return result;

    }

　　这也是之前非常普片通用的做法,这种做法其实非常繁琐,可以先去掉for循环:

    //for循环重构

    public static List<String> getNameByAgeOne(List<Dept> list){

        List<String> result=new ArrayList<String>();

        list.forEach(dept->{

            dept.getEmployees().forEach(employee->{

                if(employee.getAge()>30){

                    result.add(employee.getName());

                }

            });

        });

        return result;

    }

　　然后接着重构,上面这种明显和没重构之前区别不大:

    //stream重构

    public static List<String> getNameByAgeTwo(List<Dept> list){

        List<String> result=new ArrayList<String>();

        list.stream().forEach(dept->{

            dept.getEmployees().stream().filter(employee -> employee.getAge()>30)//使用流过滤

                    .map(employee->employee.getName()) //挑选出name

                    .forEach(name->result.add(name));

        });

        return result;

    }

　　这里就使用了stream流,先将部门集合转换为流,再进行迭代,取出每一个员工集合,再使用流进行过滤,取name等操作,

但是流中是可以合并流的,即我们完全没必要使用foreach中将每一个员工集合转换成流了,直接使用合并流,而且流可以转换成集合

,这也就意味着可以直接返回流转成的集合了:

    //stream再重构

    public static List<String> getNameByAgeFinal(List<Dept> list){

        return list.stream().flatMap(dept -> dept.getEmployees().stream())

                .filter(employee ->employee.getAge()>30 )

                .map(employee -> employee.getName())

                .collect(Collectors.toList());

    }

}

　　直接将流collect(Collectors.toList());返回非常方便,使用flatMap将每个部门的员工集合流合并得到所有员工的流,就可以去掉for循环了,之后再进行操作,这样重构就完美了!!!

最后结果:

效果都是一样的,但是代码看着高大上多了!stream确实非常强大方便!!

巴特西

使用java8的StreamAPI对集合计算进行代码重构

最新文章

热门文章