Lambda表达式之Stream/1

小鱼儿 2022-05-14 09:04 224阅读 0赞

从 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]