java 8 - 流，地图和计数不同

java java-8 java-stream

2022-09-02 13:00:33

我第一次尝试使用java 8流...

我有一个对象 Bid，它表示用户对拍卖中某件商品的出价。我有一个出价列表，我想做一个地图，计算用户出价的次数（不同）。

这是我对它的看法：

bids.stream()
         .collect(
             Collectors.groupingBy(
                  bid ->  Bid::getBidderUserId, 
                  mapping(Bid::getAuctionId, Collectors.toSet())
             )
         ).entrySet().stream().collect(Collectors.toMap(
             e-> e.getKey(),e -> e.getValue().size())
        );

它的工作原理，但我觉得我在作弊，因为我流式传输地图的条目集，而不是对初始流进行操作...一定是更正确的方法，但我无法弄清楚...

谢谢

答案 1

您可以执行两次：groupingBy

Map<Integer, Map<Integer, Long>> map = bids.stream().collect(
        groupingBy(Bid::getBidderUserId,
                groupingBy(Bid::getAuctionId, counting())));

这样，您就可以了解每个用户在每次竞价中的出价。因此，内部地图的大小是用户参与的拍卖数量。如果您不需要其他信息，可以执行以下操作：

Map<Integer, Integer> map = bids.stream().collect(
        groupingBy(
                Bid::getBidderUserId,
                collectingAndThen(
                        groupingBy(Bid::getAuctionId, counting()),
                        Map::size)));

这正是您所需要的：将用户映射到用户参与的拍卖数量。

更新：还有类似的解决方案，更接近您的示例：

Map<Integer, Integer> map = bids.stream().collect(
        groupingBy(
                Bid::getBidderUserId,
                collectingAndThen(
                        mapping(Bid::getAuctionId, toSet()),
                        Set::size)));

答案 2

Tagir Valeev的答案是正确的（+1）。下面是一个附加的，它使用您自己的下游收集器为组By执行完全相同的操作：

    Map<Integer, Long> map = bids.stream().collect(
               Collectors.groupingBy(Bid::getBidderUserId, 
                                     new Collector<Bid, Set<Integer>, Long>() {

        @Override
        public Supplier<Set<Integer>> supplier() {
            return HashSet::new;
        }

        @Override
        public BiConsumer<Set<Integer>, Bid> accumulator() {
            return (s, b) -> s.add(b.getAuctionId());
        }

        @Override
        public BinaryOperator<Set<Integer>> combiner() {
            return (s1, s2) -> {
                s1.addAll(s2);
                return s1;
            };
        }

        @Override
        public Function<Set<Integer>, Long> finisher() {
            return (s) -> Long.valueOf(s.size());
        }

        @Override
        public Set<java.util.stream.Collector.Characteristics> characteristics() {
            return Collections.unmodifiableSet(EnumSet.of(Collector.Characteristics.UNORDERED, Collector.Characteristics.IDENTITY_FINISH));
        }
    }));