使用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的原創文章，請關注公衆號"汪子熙":