Java 中的 Lambda List 轉 Map 的多種方法詳解

2022-07-06 14:03:54

故事背景

我們平時在專案中經常會遇到 List 轉 Map 的情況，但是傳統的方式又顯得太臃腫，於是就想到 Lambda 神器，今天我們就來看看都有哪幾種轉換方式（List -> Map）

公共程式碼

// Person 實體類
@Data
class Person {
    private String uuid;
    private String name;
    private String gender;
    private int age;
 
    public Person(String name, String gender, int age) {
        this.uuid = UUID.randomUUID().toString();
        this.name = name;
        this.gender = gender;
        this.age = age;
    }
}
// List 集合
List<Person> persons = new ArrayList<>();
persons.add(new Person("張三", "男", 27));
persons.add(new Person("李四", "男", 14));
persons.add(new Person("王五", "女", 17));
persons.add(new Person("趙六", "女", 34));

方式一（partitioningBy 分兩組）

partitioningBy要求傳入一個Predicate，會按照滿足條件和不滿足條件分成兩組，得到的結果是Map<Boolean, List<T>>結構，比如我們按是否未成年分成兩組

Map<Boolean, List<Person>> personsByAge = persons.stream()
	.collect(Collectors.partitioningBy(p -> p.getAge() > 18));
System.out.println(JSON.toJSONString(personsByAge));
 
// 輸出
{
	false: [{
		"age": 14,
		"gender": "男",
		"name": "李四",
		"uuid": "9fc3be98-f676-42a4-9f02-ebdab328103a"
	}, {
		"age": 17,
		"gender": "女",
		"name": "王五",
		"uuid": "3621044d-25a1-4946-a765-57b074f63f26"
	}],
	true: [{
		"age": 27,
		"gender": "男",
		"name": "張三",
		"uuid": "3f87ec59-29a1-4137-b95b-ae755f0e06ca"
	}, {
		"age": 34,
		"gender": "女",
		"name": "趙六",
		"uuid": "04ed8e9f-545b-49f5-a28b-ce0cccd15663"
	}]
}

方式二（groupingBy 分多組）

比如按照性別進行分組，得到的是Map<String, List<T>>結構

Map<String, List<Person>> personByGender = persons.stream()
	.collect(Collectors.groupingBy(Person::getGender));
System.out.println(JSON.toJSONString(personByGender));
 
// 輸出
{
	"女": [{
		"age": 17,
		"gender": "女",
		"name": "王五",
		"uuid": "feb8ca82-789f-445e-9e85-c14aa1d70546"
	}, {
		"age": 34,
		"gender": "女",
		"name": "趙六",
		"uuid": "6402b5ec-03cd-45d1-aa6d-7134509ca670"
	}],
	"男": [{
		"age": 27,
		"gender": "男",
		"name": "張三",
		"uuid": "e2c5ec58-5767-4807-8470-56a016dbc5eb"
	}, {
		"age": 14,
		"gender": "男",
		"name": "李四",
		"uuid": "d10aad57-038b-4ff8-8b36-86045d657c5a"
	}]
}

方式三（toMap 自定義<Key, Value>）

前面介紹的partitioningBy和groupingBy返回Map的value部分都是List<T>結構的，有時我們需要value是物件的一個屬性，比如我們想構造一個uuid到name的對映，以方便通過uuid快速獲取人員的名字

Map<String, String> uuidNameMap = persons.stream()
	.collect(Collectors.toMap(Person::getUuid, Person::getName));
System.out.println(JSON.toJSONString(uuidNameMap));
 
// 輸出
{
	"7a021022-fa62-4f57-bf33-873b8e030cc3": "王五",
	"e0bad9e6-2c3c-417e-9d27-3b321312421a": "張三",
	"895b0f95-b4fd-481e-ba6c-33f0b636e6cf": "李四",
	"fcb6f403-8489-4853-98c5-6f41341165ba": "趙六"
}

實際情況有可能同一個key會對應多個value，就有可能拋Duplicate key異常。這時可以傳入第三個引數決定重複時如何選擇，比如我們想構造<name, uuid>的對映，但是考慮可能有重名的可能，就可以這麼做（Tips：這裡(p1, p2) -> p1表示如果重複則取前者）

Map<String, String> nameUuidMap = persons.stream()
	.collect(Collectors.toMap(Person::getName, Person::getUuid, (p1, p2) -> p1));
System.out.println(JSON.toJSONString(nameUuidMap));

附加：如何呈現 <Person::getName, Person> 這種資料結構呢？

Map<String, Person> namePersonMap = persons.stream()
	.collect(Collectors.toMap(Person::getName, v -> v, (p1, p2) -> p1));
System.out.println(JSON.toJSONString(namePersonMap));

到此這篇關於Java 中的 Lambda List 轉 Map 幾種方式的文章就介紹到這了,更多相關Java Lambda轉 Map內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！