собирать дочерние объекты в хэш-карту parentObjectId, используя потоки Java 8 - PullRequest
Новая
       71

собирать дочерние объекты в хэш-карту parentObjectId, используя потоки Java 8

2 голосов
/ 12 ноября 2019 
public class StreamTest {

    public static void main(String[] args){
        parentObject parentObject1 = new parentObject();
        parentObject1.setId(1);
        ArrayList<ChildObject> childObjectList = new ArrayList<ChildObject>();
        parentObject1.setChildObjectList(childObjectList);

        ChildObject childObject1 = new ChildObject();
        childObject1.setId(1);
        childObject1.setName("ce1");

        ChildObject childObject2 = new ChildObject();
        childObject2.setId(2);
        childObject2.setName("ce2");

        ChildObject childObject3 = new ChildObject();
        childObject3.setId(3);
        childObject3.setName("ce3");

        childObjectList.add(childObject1);
        childObjectList.add(childObject2);
        childObjectList.add(childObject3);

        parentObject parentObject2 = new parentObject();
        ArrayList<parentObject> parentObjectList = new ArrayList<parentObject>();
        parentObjectList.add(parentObject1);
        parentObjectList.add(parentObject2);

        parentObject2.setId(2);
        ArrayList<ChildObject> childObjectList2 = new ArrayList<ChildObject>();
        parentObject2.setChildObjectList(childObjectList2);

        ChildObject childObject4 = new ChildObject();
        childObject4.setId(4);
        childObject4.setName("ce4");

        ChildObject childObject5 = new ChildObject();
        childObject5.setId(5);
        childObject5.setName("ce5");

        ChildObject childObject6 = new ChildObject();
        childObject6.setId(6);
        childObject6.setName("ce6");

        childObjectList2.add(childObject4);
        childObjectList2.add(childObject5);
        childObjectList2.add(childObject6);


       //parentObjectID as key and childObjectList as values
        HashMap<Integer, Set<String>> parentIDchildNameMap = new HashMap<Integer, Set<String>>();
        parentObjectList.forEach(
                parentObject -> parentIDchildNameMap.put(parentObject.getId(), parentObject.getChildObjectList().stream()
                                                                                .filter(childObject -> childObject.getId()!=0)
                                                                                .map(childObject -> childObject.getName())
                                                                                .collect(toSet())));

        Set<Integer> outset = parentIDchildNameMap.keySet();

        for(Integer va : outset){
            System.out.println("key :"+ va + " value : " +parentIDchildNameMap.get(va));
        }
    }
}

У меня есть parentObject, в котором есть список дочерних объектов, я хочу создать хэш-карту с идентификатором родительского объекта в качестве ключа и списком имен childObject в качестве значений. Я написал вышеприведенное выражение для его создания. Я создаю хэш-карту и помещаю значения сам. Это работает, но есть ли лучший способ сделать это? Могу ли я напрямую собрать их на карте, используя метод сбора?

1 Ответ

1 голос
/ 12 ноября 2019

Да, вы можете использовать Collectors.toMap, чтобы получить желаемый результат напрямую. Это может быть достигнуто как: 

Map<Integer, Set<String>> parentIdChildNameMap = parentObjectList.stream()
        .collect(Collectors.toMap(ParentObject::getId, parent -> parent.getChildObjectList().stream()
                .filter(childObject -> childObject.getId() != 0)
                .map(ChildObject::getName)
                .collect(Collectors.toSet()), (a, b) -> b));
...