快捷導(dǎo)航

Java中List常用操作比for循環(huán)更優(yōu)雅的寫法示例

更新時間：2021年11月03日 15:16:02 作者：Mr泰迪

List是Java中比較常用的集合類,關(guān)于List接口有很多實現(xiàn)類,下面這篇文章主要給大家介紹了關(guān)于Java中List常用操作比for循環(huán)更優(yōu)雅的寫法,需要的朋友可以參考下

引言

使用JDK1.8之后，大部分list的操作都可以使用lambda表達式去寫，可以讓代碼更簡潔，開發(fā)更迅速。以下是我在工作中常用的lambda表達式對list的常用操作，喜歡建議收藏。

以用戶表為例，用戶實體代碼如下：

public class User {
    private Integer id; //id

    private String name; //姓名

    private Integer age; //年齡

    private Integer departId; //所屬部門id
}

List<User> list = new ArrayList<>();

簡單遍歷

使用lambda表達式之前，如果需要遍歷list時，一般使用增強for循環(huán)，代碼如下：

List<User> list = new ArrayList<>();
for (User u:list) {
    System.out.println(u.toString());
}

使用lambda表達式之后，可以縮短為一行代碼：

list.forEach(u-> System.out.println(u.toString()));

篩選符合某屬性條件的List集合

以篩選年齡在15-17之間的用戶為例，for循環(huán)寫法為：

List<User> users = new ArrayList<>();
for (User u : list) {
	if (u.getAge() >= 15 && u.getAge() <= 17) {
		users.add(u);
	}
}

使用lambda表達式寫法為：

List<User> users = list.stream()
                           .filter(u -> u.getAge() >= 15 && u.getAge() <= 17)
                           .collect(Collectors.toList());

獲取某屬性返回新的List集合

以獲取id為例，項目中有時候可能會需要根據(jù)用戶id的List進行查詢或者批量更新操作，這時候就需要用戶id的List集合，for循環(huán)寫法為：

List<Integer> ids = new ArrayList<>();
for (User u:list) {
	ids.add(u.getId());
}

lambda表達式寫法為：

List<User> users = list.stream()
                           .filter(u -> u.getAge() >= 15 && u.getAge() <= 17)
                           .collect(Collectors.toList());

獲取以某屬性為key，其他屬性或者對應(yīng)對象為value的Map集合

以用戶id為key（有時可能需要以用戶編號為key），以id對應(yīng)的user作為value構(gòu)建Map集合,for循環(huán)寫法為：

Map<Integer,User> userMap = new HashMap<>();
for (User u:list) {
	if (!userMap.containsKey(u.getId())){
		userMap.put(u.getId(),u);
	}
}

lambda表達式寫法為：

Map<Integer,User> map = list.stream()
                                     .collect(Collectors.toMap(User::getId, 
                                     Function.identity(),
                                     (m1,m2)->m1));

Function.identity()返回一個輸出跟輸入一樣的Lambda表達式對象，等價于形如t -> t形式的Lambda表達式。
(m1,m2)-> m1此處的意思是當轉(zhuǎn)換map過程中如果list中有兩個相同id的對象，則map中存放的是第一個對象，此處可以根據(jù)項目需要自己寫。

以某個屬性進行分組的Map集合

以部門id為例，有時需要根據(jù)部門分組，篩選出不同部門下的人員，如果使用for循環(huán)寫法為：

Map<Integer,List<User>> departGroupMap = new HashMap<>();
for (User u:list) {
 if (departGroupMap.containsKey(u.getDepartId())){
            departGroupMap.get(u.getDepartId()).add(u);
 }else {
            List<User> users1 = new ArrayList<>();
            users1.add(u);
            departGroupMap.put(u.getDepartId(),users1);
 }
}

使用lambda表達式寫法為：

Map<Integer,List<User>> departGroupMap = list.stream()
                                                                 .collect(Collectors
                                                                 .groupingBy(User::getDepartId));

其他情況

可以根據(jù)需要結(jié)合stream()進行多個操作，比如篩選出年齡在15-17歲的用戶，并且根據(jù)部門進行分組分組，如果使用for循環(huán)，代碼如下：

Map<Integer,List<User>> departGroupMap = new HashMap<>();
for (User u:list) {
	if (u.getAge() >= 15 && u.getAge() <= 17) {
		if (departGroupMap.containsKey(u.getDepartId())){
			departGroupMap.get(u.getDepartId()).add(u);
		}else {
			List<User> users1 = new ArrayList<>();
			users1.add(u);
			departGroupMap.put(u.getDepartId(),users1);
		}
	}
}

使用lambda表達式，代碼如下：

Map<Integer,List<User>> departGroupMap = list.stream()
                                                               .filter(u->u.getAge() >= 15 && u.getAge() <= 17)
                                                               .collect(Collectors.groupingBy(User::getDepartId));