java 8 - stream,map和count distinct

时间:2015-05-17 02:19:21

标签: java java-8 java-stream

我第一次尝试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())
        );

它有效,但我觉得我作弊,因为我流式传输地图的入口集,而不是对初始流进行操作......必须是更正确的方法,但这样做,但我无法弄明白......

由于

2 个答案:

答案 0 :(得分:22)

您可以执行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)));

答案 1 :(得分:0)

Tagir Valeev的回答是正确的(+1)。这是另一个使用您自己的groupBy下游收集器完全相同的:

    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));
        }
    }));