如何使用Java流API收集数据以映射

Question

我有一个问题，希望有人能帮助我提高Java流API的知识。

我的服务通过RabbitMQ以如下数组的形式从另一个服务接收数据：

[ {"deviceId":1,"userId":1},
{"deviceId":2,"userId":1},
{"deviceId":3,"userId":2},
{"deviceId":4,"userId":2},
{"deviceId":5,"userId":1},
{"deviceId":6,"userId":2},
... ]

我需要对这些数据做的是将其收集到用户具有userId的列表和设备id的列表中。因此，从上面的数据来看，它应该是：

User {userId: 1, List<deviceId>: 1,2,5} & User {userId: 2, List<deviceId>: 3,4,6}

我确实有一个可行的解决方案，但它很笨拙

解决方案是：

    private List<User> parseDeviceData(JsonNode jsonNode) {
    List<JsonNode> list = StreamSupport.stream(jsonNode.spliterator(), false)
            .filter(this::validDeviceRecord) // checking for presence of deviceId & userId
            .collect(Collectors.toList());

    Map<Integer, ArrayList<Integer>> map = new HashMap<>();
    for (JsonNode j : list) {
        Integer id = j.get("deviceId").asInt();
        Integer userId = j.get("userId").asInt();
        ArrayList<Integer> l;
        if (map.containsKey(userId)) {
            l = map.get(userId);
        } else {
            l = new ArrayList<>();
        }
        l.add(id);
        map.put(userId, l);
    }

    return map.entrySet().parallelStream()
            .map(entry -> new User(entry.getKey(), entry.getValue()))
            .collect(Collectors.toList());
}

如何对流执行相同的操作？谢谢

Answer 1

您可以将下面的代码作为示例。

List<MyObject> list = new ArrayList<>();
        list.add(new MyObject("1", "1"));
        list.add(new MyObject("2", "2"));
        list.add(new MyObject("3", "1"));
        list.add(new MyObject("4", "2"));
        list.add(new MyObject("5", "3"));
        list.add(new MyObject("6", "1"));
        Map<String, List<String>> items = list.stream().collect(Collectors.groupingBy(MyObject::getUserId, Collectors.mapping(MyObject::getDeviceId,Collectors.toList())));
        System.out.println(items);