Java 8 的lambda表達式語法與對Stream流的操作已經出來很久了,之前一直沒有感覺到有特別好用,直到最近自己在工作開發中心血來潮用了Stream來處理集合,才發現它的強大之處。以前需要許多代碼才能完成的集合處理操作,使用Stream和lambda之后,只需要一兩行代碼就能夠輕易完成,而且代碼含義也很明確。話不多說,上例子:
場景舉例:假設我們需要對一個公司的員工對象按照員工職能進行分類,分類之前不知道一共有幾個類別(就這么假設吧。。。)。
這是我們定義的員工類(省略get,set):
public class Employee{
private Integer number;
private String name;
private Integer age;
private Rank rank;
private Gender gender;
public Employee(Integer number, String name, Integer age, Rank rank, Gender gender) {
this.number = number;
this.name = name;
this.age = age;
this.rank = rank;
this.gender = gender;
}
@Override
public String toString() {
return "Employee{" +
"number=" + number +
", name='" + name + '\'' +
", age=" + age +
", rank=" + rank +
'}';
}
public static enum Rank{
MANAGER, PROGRAMMER,TESTER,OPERATOR
}
public static enum Gender{
MALE,FEMALE
}
}
傳統的Java代碼(我就不寫了)一般是先new一個map,key是員工職能,value是一個屬于這個職能的員工的集合list,然后對所有員工進行遍歷,先判斷map中有沒有對應的key,有就加到對應的value的list里面;沒有就先new一個list,然后把當前員工加到list里面,再把k,v加到map里面,如此循環直到結束。而用了Stream之后,我們看看可以怎么簡化:
Map<Employee.Rank, List<Employee>> map = employees.stream().collect(Collectors.groupingBy(Employee::getRank, Collectors.toList()));
只用了一個groupingBy方法就能取代之前代碼的功能,十分強大,效率很高。而且代碼的含義也明確:根據Employee的getRank方法返回的值對其進行分組,每組是一個list,結果放到map中。groupingBy方法的第一個參數是分類依據,第二個參數是對分類結果進行處理的函數。
附上完整示例:
public class MainTest {
public static void main(String[] args) {
List<Employee> employees = new ArrayList<>();
employees.add(new Employee(1, "e1", 30, Employee.Rank.MANAGER, Employee.Gender.MALE));
employees.add(new Employee(2, "e2", 31, Employee.Rank.PROGRAMMER, Employee.Gender.MALE));
employees.add(new Employee(3, "e3", 29, Employee.Rank.PROGRAMMER, Employee.Gender.MALE));
employees.add(new Employee(4, "e4", 24, Employee.Rank.PROGRAMMER, Employee.Gender.MALE));
employees.add(new Employee(5, "e5", 25, Employee.Rank.PROGRAMMER, Employee.Gender.FEMALE));
employees.add(new Employee(6, "e6", 29, Employee.Rank.PROGRAMMER, Employee.Gender.MALE));
employees.add(new Employee(7, "e7", 25, Employee.Rank.TESTER, Employee.Gender.FEMALE));
employees.add(new Employee(8, "e8", 27, Employee.Rank.TESTER, Employee.Gender.MALE));
employees.add(new Employee(9, "e9", 28, Employee.Rank.OPERATOR, Employee.Gender.MALE));
employees.add(new Employee(10, "e10", 26, Employee.Rank.OPERATOR, Employee.Gender.FEMALE));
System.out.println(employees);
// 對員工按職能進行分類
Map<Employee.Rank, List<Employee>> map = employees.stream()
.collect(Collectors.groupingBy(Employee::getRank, Collectors.toList()));
System.out.println(map);
// 對組內員工按年齡排序
map.values().stream().forEach(MainTest::sortList);
System.out.println(map);
}
// 對員工按年齡大小進行排序
private static void sortList(List<Employee> list) {
Collections.sort(list, (bean1, bean2) -> bean1.getAge().compareTo(bean2.getAge()));
}
}
在上面的代碼中,我們也用了stream的forEach方法來進行遍歷,forEach方法的參數是一個對遍歷的每個元素的進行操作的方法,這里我們傳入的是對map.Values的每個list進行集合內部按年齡排序的方法。
運行結果:
所有員工:[Employee{number=1,name='e1',age=30,rank=MANAGER},Employee{number=2,name='e2',age=31,rank=PROGRAMMER},Employee{number=3,name='e3',age=29,rank=PROGRAMMER},Employee{number=4,name='e4',age=24,rank=PROGRAMMER},Employee{number=5,name='e5',age=25,rank=PROGRAMMER},Employee{number=6,name='e6',age=29,rank=PROGRAMMER},Employee{number=7,name='e7',age=25,rank=TESTER},Employee{number=8,name='e8',age=27,rank=TESTER},Employee{number=9,name='e9',age=28,rank=OPERATOR},Employee{number=10,name='e10',age=26,rank=OPERATOR}]
分組之后:{MANAGER=[Employee{number=1,name='e1',age=30,rank=MANAGER}],TESTER=[Employee{number=7,name='e7',age=25,rank=TESTER},Employee{number=8,name='e8',age=27,rank=TESTER}],PROGRAMMER=[Employee{number=2,name='e2',age=31,rank=PROGRAMMER},Employee{number=3,name='e3',age=29,rank=PROGRAMMER},Employee{number=4,name='e4',age=24,rank=PROGRAMMER},Employee{number=5,name='e5',age=25,rank=PROGRAMMER},Employee{number=6,name='e6',age=29,rank=PROGRAMMER}],OPERATOR=[Employee{number=9,name='e9',age=28,rank=OPERATOR},Employee{number=10,name='e10',age=26,rank=OPERATOR}]}
組內年齡排序之后:{MANAGER=[Employee{number=1,name='e1',age=30,rank=MANAGER}],TESTER=[Employee{number=7,name='e7',age=25,rank=TESTER},Employee{number=8,name='e8',age=27,rank=TESTER}],PROGRAMMER=[Employee{number=4,name='e4',age=24,rank=PROGRAMMER},Employee{number=5,name='e5',age=25,rank=PROGRAMMER},Employee{number=3,name='e3',age=29,rank=PROGRAMMER},Employee{number=6,name='e6',age=29,rank=PROGRAMMER},Employee{number=2,name='e2',age=31,rank=PROGRAMMER}],OPERATOR=[Employee{number=10,name='e10',age=26,rank=OPERATOR},Employee{number=9,name='e9',age=28,rank=OPERATOR}]}```