Lambda表达式之Stream/1

从 Java8 开始引入了 Stream，通过它能够实现对集合进行函数式编程，提供了许多更强大的针对集合的操作。
Set 和 List 可以通过stream()方法获取Stream实例，数组可以通过 Arrays.stream(array) 获取 Stream 实例。

1. 准备测试数据

实体类

@Data
public class PetStore {
    private String name;
    private Integer num;

    @Override
    public String toString() {
        return "PetStore{" +
                "name='" + name + '\'' +
                ", num=" + num +
                '}';
    }

    public PetStore(String name, Integer num){
        this.name = name;
        this.num = num;
    }
}

实体类集合

//PetStore可以看做是一家宠物店里的宠物笼子
PetStore dog1 = new PetStore("dog",5);
PetStore dog2 = new PetStore("dog",2);
PetStore cat1 = new PetStore("cat",3);
PetStore pig1 = new PetStore("pig",2);
PetStore pig2 = new PetStore("pig",7);

List<PetStore> psList = new ArrayList<>();
psList.add(pig1);
psList.add(dog2);
psList.add(cat1);
psList.add(pig2);
psList.add(dog1);

2. 分组操作

对集合 psList 中的数据进行分组；其中 k 是分组标识（宠物名）；v 是 List<PetStore> ，宠物名称相同的 PetStore 对象放在同一个集合中。

psList.stream().collect(Collectors.groupingBy(PetStore::getName)).forEach(
    (k,v)->{
        System.out.println(k);    //打印
        for(PetStore ps : v) {
            System.out.println(ps.toString());
        }
        System.out.println("=========");
    }
);

/** 打印的结果 **

cat
PetStore{name='cat', num=3}
=========
dog
PetStore{name='dog', num=2}
PetStore{name='dog', num=5}
=========
pig
PetStore{name='pig', num=2}
PetStore{name='pig', num=7}
=========

* */

3. 计数

统计不同分组集合中的对象个数。

Map<String, Long> count = psList.stream()
    .collect(Collectors.groupingBy(PetStore::getName, Collectors.counting()));

System.out.println(count);
//{cat=1, dog=2, pig=2}

4. 累加求和

统计不同分组集合中的所有对象指定字段（数值类型）的和。

Map<String, Integer> sum = psList.stream()
    .collect(Collectors.groupingBy(
        PetStore::getName,
        Collectors.summingInt(PetStore::getNum)
    )
);

System.out.println(sum);
//{cat=3, dog=7, pig=9}

5. 以逗号分隔的数字字符串转换成 List<Integer>

String ids = "1,3,5,6,7";

List<Integer> storeIds = Arrays.stream(ids.split(","))
    .map(s -> Integer.parseInt(s.trim()))
    .collect(Collectors.toList()
);

System.out.println(storeIds.toString());
// [1, 3, 5, 6, 7]