Sream_筛选 切片 映射 排序

package com.zhou.java3;

import com.zhou.java2.Employee;
import com.zhou.java2.EmployeeData;
import org.junit.jupiter.api.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Stream;

/**
 * @author upzhou
 * @create 2022-04-04 16:13
 */
public class StreamAPITest1 {

    //1. 筛选与切片
    @Test
    public void test1() {

        List<Employee> list = EmployeeData.getEmployees();

//        filter(Predicate p): 接收 Lambda, 从流中排出某些元素
        Stream<Employee> stream = list.stream();
        //查询工资大于 7000 的
        stream.filter(e -> e.getSalary() > 7000).forEach(System.out::println);

        System.out.println();
//        limit(n): 截断流, 使其元素不超过给定数量
        list.stream().limit(3).forEach(System.out::println);

        System.out.println();
//        skip(n): 跳过元素,返回一个扔掉了前 n 个元素的流.若流中元素不足 n 个,则返回空流
        list.stream().skip(3).forEach(System.out::println);

//        distinct(): 筛选,通过流所生成的元素的 hashCode()和 equals()去除重复元素
        list.add(new Employee(1010, "刘强东", 40, 8000));
        list.add(new Employee(1010, "刘强东", 40, 8000));
        list.add(new Employee(1010, "刘强东", 40, 8000));
        list.add(new Employee(1010, "刘强东", 40, 8000));
        list.add(new Employee(1010, "刘强东", 40, 8000));

//        System.out.println(list);
        list.stream().distinct().forEach(System.out::println);

    }


    // 映射
    @Test
    public void test2() {
//      map(Function): 接收一个函数作为参数，将元素转换成其他形式或提取信息，该函数会被应用到每个元素上，并将其映射成一个新的元素。
        List<String> list = Arrays.asList("aa", "bb", "cc", "dd");
        list.stream().map(str -> str.toUpperCase()).forEach(System.out::println);

//      练习1：获取员工姓名长度大于3的员工的姓名。
        List<Employee> employees = EmployeeData.getEmployees();
        Stream<String> namesStream = employees.stream().map(Employee::getName);
        namesStream.filter(name -> name.length() > 3).forEach(System.out::println);

//        练习 2:
        Stream<Stream<Character>> streamStream = list.stream().map(StreamAPITest1::formStringToStream);
        streamStream.forEach(s -> {
            s.forEach(System.out::println);
        });
        System.out.println();

//      flatMap(Function f): 接收一个函数作为参数，将流中的每个值都换成另一个流，然后把所有流连接成一个流
        Stream<Character> characterStream = list.stream().flatMap(StreamAPITest1::formStringToStream);
        characterStream.forEach(System.out::println);
    }

    //将字符串中的多个字符构成的集合转换为对应的 Stream 实例
    public static Stream<Character> formStringToStream(String str) {
        ArrayList<Character> list = new ArrayList<>();
        for (Character c : str.toCharArray()) {
            list.add(c);
        }
        return list.stream();
    }


    @Test
    public void test3() {
        ArrayList list1 = new ArrayList();
        list1.add(1);
        list1.add(2);
        list1.add(3);

        ArrayList list2 = new ArrayList();
        list2.add(4);
        list2.add(5);
        list2.add(6);

//        list1.add(list2);   //  [1, 2, 3, [4, 5, 6]] (相当于 map)
        list1.addAll(list2);    //  [1, 2, 3, 4, 5, 6]  (相当于 flatMap)
        System.out.println(list1);
    }

    //3. 排序
    @Test
    public void test4() {
//        sorted(): 自然排序
        List<Integer> list = Arrays.asList(12, 43, 65, 34, 87, 0, -98, 7);
        list.stream().sorted().forEach(System.out::println);
        //抛异常. 原因: Employee 没有实现 Comparable 接口
//        List<Employee> employees = EmployeeData.getEmployees();
//        employees.stream().sorted().forEach(System.out::println);

//        sorted(Comparator com): 定制排序
        List<Employee> employees = EmployeeData.getEmployees();
        employees.stream().sorted((e1, e2) -> {

            int ageValue = Integer.compare(e1.getAge(), e2.getAge());
            if (ageValue != 0) {
                return ageValue;
            } else {
                return -Double.compare(e1.getSalary(), e2.getSalary());
            }

        }).forEach(System.out::println);
    }


}