Java数据结构转换：从项目-用户映射到用户-项目映射

1. 问题描述与目标

在实际开发中，我们经常需要对数据结构进行转换以适应不同的业务逻辑或查询需求。假设我们有一个map>类型的数据，其中键（string）代表项目名称，值（list）代表与该项目关联的用户列表。例如：

Map> projectToUsersMap = new HashMap<>();
projectToUsersMap.put("projectA", Arrays.asList(new User(1,"Bob"), new User(2,"John"), new User(3,"Mo")));
projectToUsersMap.put("projectB", Arrays.asList(new User(2,"John"), new User(3,"Mo")));
projectToUsersMap.put("projectC", Arrays.asList(new User(3,"Mo")));

我们的目标是将这个映射转换为Map>类型，其中键（User）代表一个独立的用户，值（List）代表该用户所参与的所有项目名称列表。期望的输出效果如下：

Bob  = [projectA]
John = [projectA, projectB]
Mo   = [projectA, projectB, projectC]

这种转换在需要根据用户查询其参与项目列表的场景中非常有用。

2. 核心实现思路

实现这一转换的核心思路是遍历原始映射中的每一个条目（Entry），然后针对每个条目，再遍历其值（即用户列表）。对于遍历到的每一个用户，我们需要将其作为新映射的键，并将当前项目的名称添加到该用户对应的项目列表中。

具体步骤如下：

立即学习“Java免费学习笔记（深入）”；

1. 创建一个新的Map>来存储转换后的结果。
2. 遍历原始projectToUsersMap的每一个Map.Entry>。
   • String是当前的项目名称。
   • List是当前项目下的用户列表。
3. 对于每个Map.Entry，获取其键（项目名称）和值（用户列表）。
4. 遍历当前项目下的每一个User对象。
   • 在新映射中，检查当前User是否已存在作为键。
   • 如果不存在，则使用putIfAbsent(user, new ArrayList())方法将该用户添加为键，并为其创建一个新的空项目列表。
   • 无论用户是否已存在，都通过get(user).add(projectName)方法将当前项目名称添加到该用户对应的项目列表中。

3. Java代码示例

以下是实现上述转换的完整Java代码示例：

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

public class UserProjectMapper {

    // 定义User记录类，自动生成equals(), hashCode(), toString()
    public static record User(int id, String name) {}

    public static void main(String[] args) {
        // 原始数据：项目到用户列表的映射
        Map> projectToUsersMap = new HashMap<>();
        projectToUsersMap.put("projectA", Arrays.asList(new User(1,"Bob"), new User(2,"John"), new User(3,"Mo")));
        projectToUsersMap.put("projectB", Arrays.asList(new User(2,"John"), new User(3,"Mo")));
        projectToUsersMap.put("projectC", Arrays.asList(new User(3,"Mo")));

        // 用于存储转换结果的新映射：用户到项目列表的映射
        Map> userToProjectsMap = new HashMap<>();

        // 遍历原始映射的每个条目
        for (Map.Entry> entry : projectToUsersMap.entrySet()) {
            String projectName = entry.getKey(); // 获取当前项目名称
            List usersInProject = entry.getValue(); // 获取当前项目下的用户列表

            // 遍历当前项目下的每个用户
            for (User user : usersInProject) {
                // 如果用户在新映射中不存在，则为其创建一个新的项目列表
                userToProjectsMap.putIfAbsent(user, new ArrayList<>());
                // 将当前项目名称添加到该用户对应的项目列表中
                userToProjectsMap.get(user).add(projectName);
            }
        }

        // 打印转换后的结果
        System.out.println("转换后的映射 (User -> List):");
        userToProjectsMap.forEach((user, projects) -> 
            System.out.println(user.name() + " = " + projects)
        );
        // 打印完整的Map对象
        System.out.println("\n完整的Map对象表示:");
        System.out.println(userToProjectsMap);
    }
}

4. 代码解析

1. public static record User(int id, String name) {}:

   • 这里使用了Java 16+ 引入的 record 类型来定义 User。record 是一种紧凑的类声明，它自动为我们生成了构造函数、访问器方法（id()和name()）、equals()、hashCode()和toString()方法。
   • 重要性：当自定义对象（如User）作为Map的键时，正确实现equals()和hashCode()方法至关重要。HashMap依赖这两个方法来判断两个键是否相等以及如何存储和查找键。record类型自动处理了这一点，确保了具有相同id和name的User对象被视为同一个键。如果使用普通的class，则需要手动重写这些方法。

2. Map> userToProjectsMap = new HashMap();:

   • 初始化一个空的HashMap，用于存储最终的用户到项目列表的映射。

3. 外层循环 for (Map.Entry> entry : projectToUsersMap.entrySet()):

   • 遍历原始projectToUsersMap的键值对。entry.getKey()获取项目名称（String），entry.getValue()获取该项目对应的用户列表（List）。

4. 内层循环 for (User user : usersInProject):

   • 遍历当前项目下的每一个User对象。

5. userToProjectsMap.putIfAbsent(user, new ArrayList());:

   • 这是实现逻辑的关键一步。putIfAbsent(K key, V value)方法会检查userToProjectsMap中是否已经存在user这个键。
     • 如果不存在，它会将user作为键，并将一个新创建的ArrayList作为值放入映射中。
     • 如果已经存在，它不会做任何操作，保留原有的值（即该用户已有的项目列表）。
   • 这个方法确保了每个用户只对应一个项目列表，并且在第一次遇到某个用户时为其初始化列表。

6. userToProjectsMap.get(user).add(projectName);:

   • 获取当前用户user对应的项目列表。由于putIfAbsent的保证，我们知道这里获取到的列表一定是非null的。
   • 将当前的项目名称projectName添加到该用户的项目列表中。

5. 运行结果

执行上述代码，将得到以下输出：

转换后的映射 (User -> List):
Bob = [projectA]
John = [projectA, projectB]
Mo = [projectA, projectB, projectC]

完整的Map对象表示:
{User[id=1, name=Bob]=[projectA], User[id=2, name=John]=[projectB, projectA], User[id=3, name=Mo]=[projectB, projectA, projectC]}

这完美地实现了将项目到用户的映射转换为用户到项目的映射，每个用户都关联了其参与的所有项目。

6. 总结

本教程展示了如何通过迭代和条件插入（putIfAbsent）的方式，将一个复杂的嵌套Map数据结构高效地转换为另一种满足特定查询需求的数据结构。理解并正确运用Map的equals()和hashCode()方法（或使用record类型自动处理）是处理自定义对象作为Map键的关键。这种数据重构技术在处理多对多关系的数据时非常实用，有助于简化后续的数据查询和业务逻辑实现。