在日常的開發(fā)工作中經(jīng)常碰到要處理 List 中數(shù)據(jù)的問題，比如從一個對象集合中獲得對象中的一個屬性的集合。之前我們想到的是遍歷每個元素，然后取出來放到另外一個集合中，比較繁瑣；在 Java8 之后，對集合可以進(jìn)行 Stream 操作，使上面的處理更簡潔。

概述

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

可以看到 map() 方法接收一個函數(shù)式接口參數(shù)，入?yún)⒂幸粋€ T ，返回一個 Stream 流，這個流是 R 泛型。主要有以下幾點(diǎn)注意，

入?yún)⑹且粋€流中的元素；
出參是一個流，且流中是新元素；

用圖表示就是下面的樣子，原始流中有四個圓形圖案，調(diào)用 map() 函數(shù)后變成了四個五角星圖案，這里的圓形圖案和五角星圖案是一一對應(yīng)的，也就是原始流中有多少個元素，調(diào)用 map() 方法后依舊有多少個元素，唯一改變的是流中的元素類型。

示例

我們先創(chuàng)建一個實(shí)體類 Student.java

@Data
public?class?Student?implements?Serializable?{
????/**
?????*?姓名
?????*/
????private?String?name;
????/**
?????*?班級
?????*/
????private?String?schoolClass;
????/**
?????*語文成績
?????*/
????private?Integer?chineseScore;
????/**
?????*?數(shù)學(xué)成績
?????*/
????private?Integer?mathScore;
}

再創(chuàng)建一個初始化數(shù)據(jù)的類：

public?class?DataFactory?{
????public?static?List<Student>?initData(){
????????List<Student>?list=?new?ArrayList<>();
????????Student?s1=new?Student();
????????s1.setName("湯霖");
????????s1.setSchoolClass("小一班");
????????s1.setChineseScore(87);
????????s1.setMathScore(95);
????????list.add(s1);
?
????????Student?s2=new?Student();
????????s2.setName("李牧唐");
????????s2.setSchoolClass("小二班");
????????s2.setChineseScore(98);
????????s2.setMathScore(97);
????????list.add(s2);
?
????????Student?s3=new?Student();
????????s3.setName("顧北辰");
????????s3.setSchoolClass("二年級");
????????s3.setChineseScore(89);
????????s3.setMathScore(94);
????????list.add(s3);
?
????????Student?s4=new?Student();
????????s4.setName("言言");
????????s4.setSchoolClass("小二班");
????????s4.setChineseScore(100);
????????s4.setMathScore(90);
????????list.add(s4);
?
????????Student?s5=new?Student();
????????s5.setName("大運(yùn)");
????????s5.setSchoolClass("小三班");
????????s5.setChineseScore(96);
????????s5.setMathScore(100);
????????list.add(s5);
????????return?list;
????}
}

找出所有的學(xué)生姓名

public?static?void?main(String[]?args)?{
??? DataFactory.initData().stream()
????????.map(student?->?student.getName())
????????.forEach(System.out::println);
}

這里使用了 map() 方法，入?yún)⑹?Student，出參是以 String 為泛型的流，最后使用 forEach 進(jìn)行了打印，看下結(jié)果

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

將學(xué)生姓名放到list中

public?static?void?main(String[]?args)?{
????//將所有的學(xué)生姓名放到list中
????List<String>?studentNames = DataFactory.initData().stream()
????????.map(student?->?student.getName())
????????.collect(Collectors.toList());
????for?(String?studentName : studentNames)?{
????????System.out.println(studentName);
????}
}

同樣的找出所有的班級，找出所有學(xué)生的成績都可類比上面的，可以看到打印出的學(xué)生姓名或班級是有重復(fù)數(shù)據(jù)，這個要怎么解決。我們知道在最后是把數(shù)據(jù)放到 List 中，為了解決重復(fù)的問題可以把數(shù)據(jù)放到 Set 中，利用 Set 的去重功能:

//將學(xué)生姓名放到Set中，可以實(shí)現(xiàn)去重功能
Set<String>?studentNames = DataFactory.initData().stream()
????.map(student?->?student.getName())
????.collect(Collectors.toSet());

結(jié)果不再打印，有興趣的可以自己試試。

將姓名為“湯霖”的語文成績置為90

現(xiàn)在有這樣一個需求，要把姓名為湯霖學(xué)生的語文成績置為90，看下利用 map() 函數(shù)怎么做？

public?static?void?main(String[]?args)?{
????List<Student>?students?=?DataFactory.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("-----改之前的數(shù)據(jù)-----");
????for?(Student?studentName?:?students)?{
????????System.out.println(studentName);
????}
}

輸出結(jié)果：

從上面的結(jié)果中可以看出，湯霖的語文成績的確被改為了90，這點(diǎn)是沒錯的。再看原數(shù)據(jù)也被改為了90，上面提到過 map() 方法生成的是一個新流，原始流中的數(shù)據(jù)也被修改了，因?yàn)橄旅孢@行代碼是在原始流中元素的基礎(chǔ)上修改的引用類型中的數(shù)據(jù)，導(dǎo)致的原始流中的數(shù)據(jù)也會改。