使用Java Stream API將List按自定義分組規則轉換成Map的一個例子

本文完整測試代碼見文末。

測試數據是List里的4個員工對象實例：

根據員工所在的城市進行分組：

結果分成了三組：

第一組的員工在上海：

第二組的員工在成都：

統計每組員工個數：

把員工進行分組，得分大於101分的在一組，小於等於101的在另一組：

分組結果：

package java8;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.function.Consumer;
import java.util.stream.Collectors;

class Employee {
	private String city;
	private String name;
	private int score;
	
	public Employee(String name, String city, int score){
		this.city = city;
		this.name = name;
		this.score = score;
	}
	
	public String getCity(){
		System.out.println("city: " + this.city);
		return this.city;
	}
	
	public String getName() {
		return this.name;
	}
	
	public int getScore() {
		return this.score;
	}
	
	@Override
    public String toString() {
        return String.format("Employee: " + this.name + " city: " + this.city);
    }
}

class Person {
    private String name;
    private int age;
 
    Person(String name, int age) {
 
        this.name = name;
        this.age = age;
    }
 
    @Override
    public String toString() {
        return String.format("Person{name='%s', age=%d}", name, age);
    }
}

// Jerry 2016-01-15 20:51PM ? 多用於extends generic的type，接受所有Object的sub class
public class StreamTest {
	private static void printMap(Map<? extends Object, ? extends Object> map) {
		 for(Entry<? extends Object, ? extends Object> entry:map.entrySet()) {
			    System.out.println("key = " + entry.getKey() + " , Value = " + entry.getValue());
			 }
	}
	
	public static void main(String[] args) {
		ArrayList<Employee> employees = new ArrayList<Employee>();
		employees.add(new Employee("A", "Shanghai",100));
		employees.add(new Employee("B", "Chengdu",101));
		employees.add(new Employee("C", "Shenzhen",102));
		employees.add(new Employee("D", "Chengdu",104));
		
		// group by City
		Map<String, List<Employee>> employeesByCity =
				employees.stream().collect( Collectors.groupingBy(Employee::getCity));
		//  default void forEach(Consumer<? super T> action) {
		for(Map.Entry<String, List<Employee>> entry:employeesByCity.entrySet()) {
		    System.out.println("key= " + entry.getKey() + " , Value = " + entry.getValue());
		    entry.getValue().forEach(System.out::println);
		 }
		 
		 // 2016-01-15 20:37PM 
		 Consumer<Employee> aa = a -> { System.out.println("Employee: " + a.getName() + " : " +  a.getScore()); };
		 List<Employee> chengduEmployee = employeesByCity.get("Chengdu");
		 chengduEmployee.forEach(aa);
		 
		 // test for counting
		 Map<String, Long> employeesByCity2 = 
				 employees.stream().collect( Collectors.groupingBy(Employee::getCity, Collectors.counting()));
		 printMap(employeesByCity2);
		
		 // calculate average score
		 Map<String, Double> employeesByCity3 = 
				 employees.stream().collect( Collectors.groupingBy(Employee::getCity,
						 Collectors.averagingInt(Employee::getScore)));
		 
		 printMap(employeesByCity3);
		/*Stream<Person> people = Stream.of(new Person("Paul", 24), new Person("Mark", 30), new Person("Will", 28));
		Map<Integer, List<String>> peopleByAge = people.collect(groupingBy(p -> p.age, mapping((Person p) -> p.name, toList())));
		System.out.println(peopleByAge);*/
		 
		 /*
		  * 分區是一種特殊的分組，結果 map 至少包含兩個不同的分組——一個true，一個false。
		  * 例如，如果想找出最優秀的員工，你可以將所有雇員分為兩組，一組銷售量大於 N，
		  * 另一組小於 N，使用 partitioningBy 收集器：
		  */
		 System.out.println("partition result");
		 Map<Boolean, List<Employee>> partitioned =
				 employees.stream().collect(Collectors.partitioningBy(e -> e.getScore() > 101));
		 printMap(partitioned);
		 
		 /*
		  * 你也可以將 groupingBy 收集器傳遞給 partitioningBy 收集器來將聯合使用分區和分組。例如，你可以統計每個分區中的每個城市的雇員人數：

		Map<Boolean, Map<String, Long>> result =
		employees.stream().collect(partitioningBy(e -> e.getNumSales() > 150,
		groupingBy(Employee::getCity, counting())));

		這樣會生成一個二級 Map:

		{false={London=1}, true={New York=1, Hong Kong=1, London=1}}
		  */
	}
}

要獲取更多Jerry的原創文章，請關注公眾號"汪子熙":