Java8 Stream 流操作

Stream 流可以说是 Java8 新特性中用起来最爽的一个功能了。

有了它，从此操作集合告别繁琐的 for 循环。

以下列举的功能都很实用：

1、查询集合中符合条件的数据集合。

2、判断集合中是否含有特定条件的数据，并返回第一个。

3、按条件排序并获取前N个记录。

4、根据条件获取集合，并计算集合中的有效数据。

5、集合分组。

package com.example.stream;

import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;

import java.math.BigDecimal;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.stream.Collectors;

/**
 * Description: Java8 Stream 流
 * Author JC
 * Date 2022/1/19
 */
@SpringBootTest(classes = DemoApplicationTests.class)
class DemoApplicationTests {


    @Test
    void contextLoads() {

        List<Person> personList = initList();

        //查询人员列表中小于20岁的人员
        List<Person> personListBelow20 = personList.stream().filter(s -> s.getAge() < 20).collect(Collectors.toList());
        System.out.println("小于20岁的人为:");
        personListBelow20.forEach(person -> System.out.println(person));

        //查询是否含有薪水超过30000的人信息 有则返回第一个人数据
        Optional<Person> personOptional = personList.stream().filter(s -> s.getAge() > 30).findFirst();
        if (personOptional.isPresent()) {
            Person p = personOptional.get();
            System.out.println("含有薪水大于30000的人: " + p);
        }

        //按年龄排序，并查询前3个人信息
        List<Person> personListSort = 
                personList.stream().limit(3).sorted(Comparator.comparing(Person::getAge)).collect(Collectors.toList());
        System.out.println("按年龄排序，并查询前3个人信息为:");
        personListSort.forEach(person -> System.out.println(person));

        //按出生日期倒叙排序
        List<Person> personListSortBirthday = 
                personList.stream().sorted(Comparator.comparing(Person::getBirthday).reversed()).collect(Collectors.toList());
        System.out.println("按出生日期倒叙排序个人信息为: ");
        personListSortBirthday.forEach(person -> System.out.println(person));

        //查询薪水大于10000的人的薪资总额
        BigDecimal salaryOver1WTotal = 
                personList.stream().filter(s -> s.getSalary().compareTo(new BigDecimal("10000.00")) > 0)
                        .map(Person::getSalary).reduce(BigDecimal.ZERO, BigDecimal::add);
        System.out.println("薪水大于10000的人的薪资总额为: " + salaryOver1WTotal);

        //以性别分组
        Map<String, List<Person>> personGroups = personList.stream().collect(Collectors.groupingBy(Person::getSex));
        //遍历Map集合
        System.out.println("按性别分组遍历人员信息为:");
        personGroups.forEach((sex, persons) -> {
            System.out.format("sex %s: %s\n", sex, persons);
        });


    }

    private List<Person> initList() {

        List<Person> personList = new ArrayList<>();
        Person p1 = new Person(1, "赵一", "男", 26, parseDate("1995-10-11"), new BigDecimal("30000.00"));
        Person p2 = new Person(2, "钱二", "女", 29, parseDate("1992-3-21"), new BigDecimal("10000.00"));
        Person p3 = new Person(3, "张三", "男", 18, parseDate("2003-1-18"), new BigDecimal("7000.00"));
        Person p4 = new Person(4, "李四", "女", 22, parseDate("1999-7-8"), new BigDecimal("9000.00"));
        Person p5 = new Person(5, "王五", "男", 35, parseDate("1988-1-1"), new BigDecimal("50000.00"));
        personList.add(p1);
        personList.add(p2);
        personList.add(p3);
        personList.add(p4);
        personList.add(p5);
        return personList;
    }

    private Date parseDate(String dateStr) {
        try {
            return new SimpleDateFormat("yyyy-MM-dd").parse(dateStr);
        } catch (ParseException e) {
            return null;
        }
    }


}

Person类如下，需要引入lombox

package com.example.stream;

import lombok.Data;
import java.math.BigDecimal;
import java.util.Date;

/**
 * Description:
 * Author JC
 * Date 2022/1/19
 */
@Data
public class Person {

    private int id;
    private String name;
    private String sex;
    private int age;
    private Date birthday;
    private BigDecimal salary;

    public Person(int id,String name,String sex,int age,Date birthday,BigDecimal salary){
        this.id =id;
        this.name = name;
        this.sex = sex;
        this.age = age;
        this.birthday =birthday;
        this.salary = salary;

    }

}