使用Java 8 Stream API将List<A>转换为Map<String,List<A>>

Question

使用Java 8 Stream API将List<A>转换为Map<String,List<A>>

3

我有以下代码：

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class Test {

    /**
     * @param args
     */
    public static void main(String[] args) 
    {
        Test t=new Test();
        t.test();
    }

    class A
    {
        int id;
        B b;

        public int getId() {
            return id;
        }
        public void setId(int id) {
            this.id = id;
        }
        public B getB() {
            return b;
        }
        public void setB(B b) {
            this.b = b;
        }

    }

    class B
    {
        int id;
        String unique;

        public String getUnique() {
            return unique;
        }
        public void setUnique(String unique) {
            this.unique = unique;
        }
        public int getId() {
            return id;
        }
        public void setId(int id) {
            this.id = id;
        }
    }

    public void test()
    {
        B b1=new B();
        b1.setId(9999);
        b1.setUnique("123456789");

        B b2=new B();
        b2.setId(7777);
        b1.setUnique("123456789");

        B b3=new B();
        b3.setId(8888);
        b3.setUnique("987654321");


        A a1=new A();
        a1.setId(1);;
        a1.setB(b1);

        A a2=new A();
        a2.setId(2);;
        a2.setB(b1);


        A a3=new A();
        a3.setId(3);;
        a3.setB(b2);

        A a4=new A();
        a4.setId(4);;
        a4.setB(b2);

        A a5=new A();
        a5.setId(5);;
        a5.setB(b3);

        A a6=new A();
        a6.setId(6);;
        a6.setB(b3);

        List<A> list=new ArrayList<A>();
        list.add(a1);
        list.add(a2);
        list.add(a3);
        list.add(a4);
        list.add(a5);
        list.add(a6);



        Map<B,List<A>> map=list.stream()
                               .collect(Collectors.groupingBy(A::getB));

        //required output
        //Map<String, List<A>>

    }
}

在这个例子中，我有一个名为class A的类，它引用了class B。 class B有一个名为unique的字段。作为源，我得到了列表（a1到a6），它们引用了Class B的对象（b1、b2和b3）。对象b1和b2包含相同的唯一值。我想使用Java 8 Stream API输出一个Map>，使得如果被A引用的B对象的唯一值相同，则结果列表应该包含Class A对象的合并列表。例如，以下是输出：

Map<String,List<A>> output = [
                                 "123456789" : [a1,a2,a3,a4],
                                 "987654321" : [a5,a6]
                             ]

- Prerak Tiwari

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- jrtapsell · Accepted Answer

修正了这行代码（原本是b1，但应该是b2）：

b2.setUnique("123456789");

正确的映射代码是这样的：

Map<String,List<A>> map=list.stream()
    .collect(Collectors.groupingBy(p -> p.getB().unique));

输出结果（将A用{ID}表示）如下：

{123456789=[{1}, {2}, {3}, {4}], 987654321=[{5}, {6}]}