這麼簡單，還不會使用java8 stream流的map()方法嗎？

良工說技術發表於2022-12-01

一、前言

在日常的開發工作中經常碰到要處理list中資料的問題，比如從資料庫中查出了很多學生，由於一些原因需要在記憶體中找出這些學生中的所有姓名，或者把名為“王五”的語文成績暫時修改為“100”，這些問題怎麼處理吶，之前我們想到的是遍歷每個元素，然後取出來放到另外一個集合中，在java8中對集合可以進行流式操作使上面的處理更簡潔。今天來看下map()方法的使用。

二、概述

stram流式處理中有map方法，先看下其定義，該方法在java.util.stream.Stream類中，

可以看到map()方法接收一個函式式介面引數，入參有一個T，返回一個Stream流，這個流是R泛型。主要有以下幾點注意，

入參是一個流中的元素；
出參是一個流，且流中是新元素；

用圖表示就是下面的樣子，原始流中有四個圓形圖案，呼叫map()函式後變成了四個五角星圖案，這裡的圓形圖案和五角星圖案是一一對應的，也就是原始流中有多少個元素，呼叫map()方法後依舊有多少個元素，唯一改變的是流中的元素型別。

好了，下面看下怎麼使用map()函式。

三、詳述

以Student為例，

Student.java

package com.example.log.stream.entity;
import java.io.Serializable;
/**
 * 學生類
 * @date 2022/11/30 20:40
 */
public class Student implements Serializable {
    /**
     * 姓名
     */
    private String name;
    /**
     * 班級
     */
    private String schoolClass;

    /**
     *語文成績
     */
    private Integer chineseScore;
    /**
     * 數學成績
     */
    private Integer MathScore;
    
    //省略get、set方法
}

有一個初始化資料的類，用來初始化一個List

Data.java

package com.example.log.stream.test;

import com.example.log.stream.entity.Student;

import java.util.ArrayList;
import java.util.List;

/**
 * @date 2022/11/30 22:43
 */
public class Data {
    public static List<Student> initData(){
        List<Student> students= new ArrayList<>();
        Student s1=new Student();
        s1.setName("王五");
        s1.setSchoolClass("一年級");
        s1.setChineseScore(100);
        s1.setMathScore(100);
        students.add(s1);

        Student s2=new Student();
        s2.setName("李四");
        s2.setSchoolClass("一年級");
        s2.setChineseScore(100);
        s2.setMathScore(100);
        students.add(s2);

        Student s3=new Student();
        s3.setName("李思");
        s3.setSchoolClass("二年級");
        s3.setChineseScore(100);
        s3.setMathScore(100);
        students.add(s3);

        Student s4=new Student();
        s4.setName("王五");
        s4.setSchoolClass("三年級");
        s4.setChineseScore(100);
        s4.setMathScore(100);
        students.add(s4);

        Student s5=new Student();
        s5.setName("趙三");
        s5.setSchoolClass("一年級");
        s5.setChineseScore(100);
        s5.setMathScore(100);
        students.add(s5);
        return students;
    }
}

下面根據上面的資料，進行下面的操作。

2.1、找出所有的學生姓名

TestMap.java

package com.example.log.stream.test;

import com.example.log.stream.entity.Student;

import java.util.ArrayList;
import java.util.List;

/**
 * 測試map方法
 * @date 2022/11/30 21:25
 */
public class TestMap {
    public static void main(String[] args) {
        List<Student> students=Data.initData();
        students.stream().map(student -> student.getName()).forEach(System.out::println);
    }

}

這裡使用了map()方法，入參是Student，出參是以String為泛型的流，最後使用forEach進行了列印，看下結果

可以看到剛好把所有的學生姓名均列印出來了。如果想把這些學生姓名放到一個List中怎麼辦？

2.1.1、將學生姓名放到list中

package com.example.log.stream.test;

import com.example.log.stream.entity.Student;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 測試map方法
 *
 * @author wangcj5
 * @date 2022/11/30 21:25
 */
public class TestMap {
    public static void main(String[] args) {
        List<Student> students=Data.initData();
//        students.stream().map(student -> student.getName()).forEach(System.out::println);
        //將所有的學生姓名放到list中
        List<String> studentNames=students.stream().map(student -> student.getName()).collect(Collectors.toList());
     for (String studentName:studentNames) {
      System.out.println(studentName);
     }
    }
}

結果如下，

同樣的找出所有的班級，找出所有學生的成績都可類比上面的，可以看到列印出的學生姓名或班級是有重複資料，這個要怎麼解決。我們知道在最後是把資料放到list中，為了解決重複的問題可以把資料放到set中，利用set的去重功能，

//將學生姓名放到Set中，可以實現去重功能
Set<String> studentNames=students.stream().map(student -> student.getName()).collect(Collectors.toSet());

結果不再列印，有興趣的可以自己試試。

2.2、將姓名為“趙三”的語文成績置為90

現在有這樣一個需求，要把姓名為“趙三”學生的語文成績置為90，看下利用map()函式怎麼做？

TestMap2.java

package com.example.log.stream.test;

import com.example.log.stream.entity.Student;

import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * 測試map方法
 * @date 2022/11/30 21:25
 */
public class TestMap2 {
    public static void main(String[] args) {
        List<Student> students = Data.initData();

        List<Student> studentNames = students.stream().map(student -> {
            if ("趙三".equals(student.getName())) {
                student.setChineseScore(90);
            }
            return student;
        }).collect(Collectors.toList());

        for (Student studentName : studentNames) {
            System.out.println(studentName);
        }
        //列印老的list
        System.out.println("改之前的資料----------------");
        for (Student studentName : students) {
            System.out.println(studentName);
        }
    }
}

看下列印的結果，