java 8 流 List<List<Long>>

Question

我有一个数据集定义为List<List<Long>> dataSet，dataSet(List)中的元素有8个子元素，我想用数据集索引0元素组，最后建立一个地图地图>，怎么做？ 旧代码是：

List<List<Long>> dataSet = .....; 
Map<Long, Set<Long>> a = new HashMap<>();
for (List<Long> data : dataSet) {
    Long userId = data.get(0);
    Long targetId = date.get(7);
    if (a.containsKey(userId)) {
        a.get(userId).add(targetId);
    } else {
        Set<Long> ids = new HashSet<>();
        ids.add(targetIds);
        a.put(userId, ids);
    }
}

Answer 1

我已经编写了适合您需求的Collector-interface 的具体实现。

注意：这个收集器也可以并行工作，非常方便

public class MapCollector implements Collector<List<Long>, Map<Long, Set<Long>>, Map<Long, Set<Long>>>{

    @Override
    public Supplier<Map<Long, Set<Long>>> supplier(){
        return HashMap::new;
    }

    @Override
    public BiConsumer<Map<Long, Set<Long>>, List<Long>> accumulator(){
        return ( m, l ) -> {
            Set<Long> longs = m.get(l.get(0));
            if( longs == null ){
                longs = new HashSet<>();
            }
            longs.add(l.get(7));
            m.put(l.get(0), longs);
        };
    }

    @Override
    public BinaryOperator<Map<Long, Set<Long>>> combiner(){
        return ( m1, m2 ) -> {
            m2.forEach(( k, v ) -> {
                Set<Long> longs = m1.get(k);
                if( longs == null ){
                    longs = v;
                } else{
                    longs.addAll(v);
                }
                m1.put(k, longs);
            });
            return m1;
        };
    }

    @Override
    public Function<Map<Long, Set<Long>>, Map<Long, Set<Long>>> finisher(){
        return UnaryOperator.identity();
    }

    @Override
    public Set<Characteristics> characteristics(){
        return EnumSet.of(Characteristics.IDENTITY_FINISH, Characteristics.UNORDERED, Characteristics.CONCURRENT);
    }
}

您可以通过以下方式使用它：Map<Long, Set<Long>> map = dataSet.stream().collect(new MapCollector());

希望这对你有用；）