具有多个键的Java 8分组

编程问答 2022-06-07

问题描述

我正在寻找一些帮助，以将具有多个键的列表中的对象列表进行分组。

基本上，我有一个用户列表，其中包含他们的订单列表，我希望能够使用UserName和Address作为键将它们分组在一起。

示例数据：

<UserList>
        <User>
            <Username>user123</Username>
            <Address>London</Address>
            <TransactionList>
                <TransactionList>
                    <OrderNumber>1</OrderNumber>
                    <Cost>3683446.6600</Cost>
                </TransactionList>
            </TransactionList>
        </User>
              <User>
            <Username>user123</Username>
            <Address>London</Address>
            <TransactionList>
                <TransactionList>
                    <OrderNumber>3</OrderNumber>
                    <Cost>500</Cost>
                </TransactionList>
            </TransactionList>
        </User>
               <User>
            <Username>user12356</Username>
            <Address>Manchester</Address>
            <TransactionList>
                <TransactionList>
                    <OrderNumber>6</OrderNumber>
                    <Cost>90000</Cost>
                </TransactionList>
            </TransactionList>
        </User>
              <User>
            <Username>user12356</Username>
            <Address>Manchester</Address>
            <TransactionList>
                <TransactionList>
                    <OrderNumber>10</OrderNumber>
                    <Cost>100</Cost>
                </TransactionList>
            </TransactionList>
        </User>
    </UserList>

我要这样订购，以便每个用户只有一个实例，并且其相关交易将根据用户名和地址分组在一个列表中：

<UserList>
        <User>
            <Username>user123</Username>
            <Address>London</Address>
            <TransactionList>
                <TransactionList>
                    <OrderNumber>1</OrderNumber>
                    <Cost>3683446.6600</Cost>
                </TransactionList>
                <TransactionList>
                    <OrderNumber>3</OrderNumber>
                    <Cost>500</Cost>
                </TransactionList>
            </TransactionList>
        </User>
         <User>
            <Username>user12356</Username>
            <Address>Manchester</Address>
            <TransactionList>
                <TransactionList>
                    <OrderNumber>6</OrderNumber>
                    <Cost>90000</Cost>
                </TransactionList>
                 <TransactionList>
                    <OrderNumber>10</OrderNumber>
                    <Cost>100</Cost>
                </TransactionList>
            </TransactionList>
        </User>
    </UserList>

我尝试使用地图来做到这一点：

Map<String,Map<String,List<User>>> map;

map = userLists.getUserList().stream()
                .collect(Collectors.groupingBy(User::getUserName,Collectors.groupingBy(User::getAddress)));

这只是我要执行的操作，但我想知道是否有更好的方法使用multikey Map或类似的方法，然后迭代并在密钥匹配的情况下列出交易列表。

预先感谢您的帮助。

解决方法

您的方法没有错，第一步是正确的：先按userName然后按address对用户进行分组。就我个人而言，我将其反转，因为更有可能有更多的用户使用相同的地址，但是顺序对于获得核心结果并不重要。

我注意到期望的输出看起来像List<User>，具有共同特征（userName和address）的用户减少，并且串联了transactionList列表。尽管带有复合键（例如${userName}_${address}）的地图可能会有所帮助，但我还是选择List<User>，它与预期输出类似。

因此，第二步是迭代所有这些条目，并将List<User>中的Map缩减为一个用户。每个 inner 条目都将与一个用户相关联（因为userName和address都如此）。 For-each远远胜于此任务。你去了

Map<String,Map<String,List<User>>> map = userLists.stream()
     .collect(Collectors.groupingBy(
            User::getUserName,Collectors.groupingBy(User::getAddress)));

List<User> list = new ArrayList<>();                 // stored output

map.forEach((userName,groupedByAddress) -> {        // for each 'userName'
    groupedByAddress.forEach((address,users) -> {   // ... and each 'address'
        User userToAdd = new User();                 // ...... create an 'User'
        userToAdd.setUserName(userName);             // ...... with the 'userName'  
        userToAdd.setAddress(address);               // ...... with the 'address'
        users.forEach(user -> {                      // ...... and of each the user's
            userToAdd.getTransactionList()           // .......... concatenate
                .addAll(user.getTransactionList());  // .......... all the transactions
        });
    });
});

Stream API非常适合分组以获得中间结果，但是，对于这样的进一步减少，它将非常笨拙。

,

您要寻找的实际上是基于常见用户的名称和地址的合并功能。如果您查看Collectors.toMap API，它会为您提供类似的功能以及Map的键和值选择。

Collectors.toMap(
        u -> Arrays.asList(u.getUserName(),u.getAddress()),Function.identity(),User::mergeUsers
));

合并在哪里看起来就像您期望的那样

static User mergeUsers(User user1,User user2) {
    List<Transaction> overall = new ArrayList<>(user1.getTransactionList());
    overall.addAll(user2.getTransactionList());
    return new User(user1.getUserName(),user1.getAddress(),overall);
}

并且由于您仅在寻找此Map的值，因此具有完整解决方案的输出将是：

Collection<User> users = userLists.getUserList().stream()
        .collect(Collectors.toMap(
                u -> Arrays.asList(u.getUserName(),User::mergeUsers
        )).values();
collectorshashmaphashmapjavajavajava-streammultikey