Java 8在包含字符串列表的对象内进行流分组

Question

Map<Tag, List<String>> tagPersonListMap = new HashMap<>();
for (Person person: personList) {
    List<Tag> tags = person.getTags();
    String id = person.getId();
    for(Tag tag: tags) {
        tagPersonListMap.computeIfAbsent(tag, k ->new ArrayList<>()).add(id);
    }
}

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

class Scratch {
    public static void main(String[] args) {
        Person person1 = new Person(Arrays.asList(Tag.EMPLOYEE, Tag.UNKNOWN), "123");
        Person person2 = new Person(Arrays.asList(Tag.UNKNOWN, Tag.VISITOR), "345");
        Person person3 = new Person(Arrays.asList(Tag.VISITOR), "678");
        Person person4 = new Person(Arrays.asList(Tag.EMPLOYEE, Tag.UNKNOWN), "910");
        List<Person> personList = Arrays.asList(
            person1, person2,
            person3, person4
        );

        Map<Tag, List<String>> tagPersonListMap = new HashMap<>();
        for (Person person: personList) {
            List<Tag> tags = person.getTags();
            String id = person.getId();
            for(Tag tag: tags) {
                tagPersonListMap.computeIfAbsent(tag, k ->new ArrayList<>()).add(id);
            }
        }

        System.out.println("Final result: " + tagPersonListMap);
    }
}

enum Tag {
    EMPLOYEE("employee"),
    VISITOR("visitor"),
    UNKNOWN("unknown");
    private final String mValue;
    private Tag(String value) {
        this.mValue = value;
    }
}

class Person {
    List<Tag> tags;
    String id;

    Person (List<Tag> tag, String i) {
        tags = tag;
        id = i;
    }

    List<Tag> getTags() {
        return tags;
    }

    String getId() {
        return id;
    }
}

预期产出

Final result: {EMPLOYEE=[123, 910], UNKNOWN=[123, 345, 910], VISITOR=[345, 678]}

Answer 1

你可以这样做。为了实现这一点，我基本上将标记列表和id转换为(tag tag,String id)的map.entry对象。所以如果有四个标签和一个Id，那就是四个条目。然后，我简单地按标记将它们分组，其值为列表 。

Map<Tag, List<String>> result = personList.stream()
        .flatMap(per -> per.getTags().stream()
                .map(tag -> Map.entry(tag, per.getId())))
                .collect(Collectors.groupingBy(Entry::getKey,
                        Collectors.mapping(Entry::getValue,
                                Collectors.toList())));

result.entrySet().forEach(System.out::println);

印刷品

UNKNOWN=[123, 345, 910]
EMPLOYEE=[123, 910]
VISITOR=[345, 678]

做了这件事，我仍然会坚持你所拥有的。有时，使用溪流就像用大锤钉钉子。