Bootstrap

使用stream流对两个集合的数据通过一个字段进行匹配,将一个集合中的数据插入到另一个集合中!

在前两天的工作中遇到这样一一个场景:我查询出来了两组List集合的数据,需要将一个集合中得对象的数据赋值给另一个集合对象对应的字段,两个对象中有一个共同的id字段,首相我想到的用for循环加上if判断的方式来完成,但是总觉得这样写不够优雅;然候又想到了通过stream流的方式来实现,而且在复杂的应用场景下或者在数据量大的情况下使用stream流会有更高的效率,因为stream是多线程并行操作的.当然在业务简单而且数据量也不大的情况下还是不同的for循环加if判断效率高一些,比如我接下来的测试数据.

定义两个类,有个Student类和一个Name类,最终目的就是将Name中的姓名赋值给Student类中的name字段.

public class Student {

    private int id;

    private String name;

    private int age;

    public Student(int id, String name, int age) {
        this.id = id;
        this.name = name;
        this.age = age;
    }
}
public class Name {

    private int id;

    private String name;

    public Name(int id, String name) {
        this.id = id;
        this.name = name;
    }
}

创建两个集合进并封装数据

    /**
     * 封装Student集合
     */
    private static List<Student> getStudentList() {
        List<Student> students = new ArrayList<>();
        students.add(new Student(1, null, 18));
        students.add(new Student(2, null, 20));
        students.add(new Student(3, null, 23));
        students.add(new Student(4, null, 20));
        students.add(new Student(5, null, 17));
        students.add(new Student(6, null, 15));

        return students;
    }


    /**
     * 封装Name集合
     */
    private static List<Name> getNameList() {
        List<Name> names = new ArrayList<>();
        names.add(new Name(1, "张三"));
        names.add(new Name(2, "李四"));
        names.add(new Name(3, "王五"));
        names.add(new Name(4, "赵六"));
        names.add(new Name(5, "田七"));
        names.add(new Name(6, "马八"));

        return names;
    }

 然后下面是使用stream流的核心代码

   /**
     * 使用流对象的方法插入数据
     */
    private static List<Student> methodA(List<Student> students, List<Name> names) {
        List<Student> studentList = students.stream().map(student -> {
            return names.stream().filter(name -> {        //条件判断
                return student.getId() == name.getId();
            }).map(name -> {
                student.setName(name.getName());
                return student;                            //返回的结果
            }).collect(Collectors.toList());
        }).flatMap(List::stream).collect(Collectors.toList());//设置返回结果类型

        return studentList;
    }

输出结果:可以看到结果是正常的,id都匹配上了对应的name

[Student{id=1, name='张三', age=18}, Student{id=2, name='李四', age=20}, Student{id=3, name='王五', age=23}, Student{id=4, name='赵六', age=20}, Student{id=5, name='田七', age=17}, Student{id=6, name='马八', age=15}]

为了测试一下这种简单场景下两种方法的效率,于是将for循环加if判断的也加上了

    /**
     * 循环遍历插入数据
     */
    private static List<Student> methodB(List<Student> students, List<Name> names) {
        for (Student student : students) {
            for (Name name : names) {
                if (student.getId() == name.getId()) {
                    student.setName(name.getName());
                }
            }
        }
        return students;
    }
public static void main(String[] args) {
        List<Student> students = getStudentList();
        List<Name> names = getNameList();
        Date date1 = new Date();
        long startTime = date1.getTime();
        //使用stream流方式
        List<Student> studentList = methodA(students, names);
        //使用for循环方式
        //List<Student> studentList = methodB(students, names);
        Date date2 = new Date();
        long endTime = date2.getTime();
        System.out.println("所用时长: " + (endTime - startTime));
        System.out.println("输出结果: " + studentList);
    }

以下是两种方式的对比,可以看出来对比还是很明显的,在简单的场景下使用for循环的效率要更高一些.

 

 在数据量大的情况下还是stream更给力一些,可以参考这位博主的测试:

https://www.xyhtml5.com/5302.htmlicon-default.png?t=L892https://www.xyhtml5.com/5302.html

;