Java（八）：Stream 流式操作

ゝ一纸荒年。 2024-03-30 12:37 173阅读 0赞

#### 介绍 ####

Java8对集合操作功能的增强，专注于对集合的各种高效、便利、优雅的聚合操作。记录一下Stream流的一些用法。

**获取list某个字段组装新List**

List<Integer> userIdList = userList.stream().map(e -> e.getUserId()).collect(Collectors.toList());

**根据指定字段分组**

Map<String, List<User>> mapByName = list.stream().collect(Collectors.groupingBy(User::getName));

**去重**

List<Integer> numList = Lists.newArrayList(1, 5, 3, 3, 6);
    numList = numList.stream().distinct().collect(Collectors.toList());
    
    // 根据指定字段去重
    list = list.stream().collect(Collectors.collectingAndThen(Collectors.toCollection(() -> new TreeSet<>(Comparator.comparing(User::getId))), ArrayList::new));

**条件过滤**

// 只要包含有“张三”的数据
    list = list.stream().filter(e -> e.getName().equals("张三")).collect(Collectors.toList());

**求和**

// 基本类型
    int sumAge = userList.stream().mapToInt(User::getAge).sum();
    // 其他  -- 若bigDecimal对象为null，可filter()过滤掉空指针
    BigDecimal totalMemberNum = userList.stream().map(User::getMemberNum).reduce(BigDecimal.ZERO, BigDecimal::add);

**最大值/最小值**

Date minDate = userList.stream().map(User::getCreateTime).min(Date::compareTo).get();
    Date maxDate = userList.stream().map(User::getCreateTime).max(Date::compareTo).get();
    User maxUp = userList.stream().max(Comparator.comparingInt(User::getAge)).get();

**差值（新增/删除）**

List<Integer> userIdListNew = Lists.newArrayList(1, 2, 3, 5, 6);
    List<Integer> userIdListOld = Lists.newArrayList(1, 2, 3, 4);
    
    // 删除人员 [4]
    List<Integer> removeUserIdList = userIdListOld.stream().filter(userIdOld -> !userIdListNew.contains(userIdOld)).collect(Collectors.toList());
    
    // 新增人员 [5, 6]
    List<Integer> addUserIdList = userIdListNew.stream().filter(userIdNew -> !userIdListOld.contains(userIdNew)).collect(Collectors.toList());

**分类统计数量**

// 多字段统计 -- ex: 统计相同name下相同age的个数
    Map<String, Map<Integer, Long>> map = list.stream().collect(
                    Collectors.groupingBy(User::getName, Collectors.groupingBy(User::getAge, Collectors.counting()))
            );
    
    // 单字段统计 [LongSummaryStatistics中包含总数、最小值、最大值、平均值等信息]   --  ex: 根据名称去统计
    Map<String, LongSummaryStatistics> map = list.stream()
            .collect(
                    Collectors.groupingBy(User::getName, Collectors.summarizingLong(User::getAge))
            );

**求list重复元素值**

@Test
    public void test02() throws Exception {
        List<Integer> list = Lists.newArrayList(1, 2, 3, 4, 5, 6, 1, 6, 6);
        List<Integer> repeatDataList = list.stream()
            .collect(Collectors.toMap(e -> e, e -> 1, Integer::sum))
            .entrySet().stream()
            .filter(entry -> entry.getValue() > 1)
            .map(Map.Entry::getKey)
            .collect(Collectors.toList());
        System.out.println(repeatDataList); // [1, 6]
        
        
        // 求list对象中某一个字段的重复值
        List<String> repeatValueDataList = dictList
                        .stream().map(e -> e.getValue()).collect(Collectors.toList())
                        .stream().collect(Collectors.toMap(e -> e, e -> 1, Integer::sum))
                        .entrySet().stream()
                        .filter(entry -> entry.getValue() > 1)
                        .map(Map.Entry::getKey)
                        .collect(Collectors.toList());
    }
    
    public <T> List<T> getRepeatDataList(List<T> list) {
        return list.stream()
                // 获得元素出现频率的 Map，键为元素，值为元素出现的次数
                .collect(Collectors.toMap(e -> e, e -> 1, Integer::sum))
                // Set<Entry>转换为Stream<Entry>
                .entrySet().stream()
                // 过滤出元素出现次数大于 1 的 entry
                .filter(entry -> entry.getValue() > 1)
                // 获得 entry 的键（重复元素）对应的 Stream
                .map(Entry::getKey)
                // 转化为 List
                .collect(Collectors.toList());
    }