如何在Spring Data JPA中通过投影返回角色权限Map集合

给定如下数据库表结构与数据，拟执行指定JPQL查询，需要将结果转换为Map<String, List<String>>（键为角色名，值为对应权限列表），如何通过Spring Data JPA投影实现该需求？

数据库表结构与数据 #

create table permissions (id bigint not null, permission varchar(255) not null, role_id bigint not null, primary key (id))
create table roles (id bigint not null, role_name varchar(255) not null, primary key (id))

insert into roles (role_name,id) values ("USER_ADMIN",1)
insert into permissions (permission,role_id,id) values ("write:all",1,10)
insert into permissions (permission,role_id,id) values ("write:key",1,11)
insert into permissions (permission,role_id,id) values ("write:vl",1,12)
insert into permissions (permission,role_id,id) values ("write:ts",1,13)


insert into roles (role_name,id) values ("DB_ADMIN",2)
insert into permissions (permission,role_id,id) values ("red:all",2,20)
insert into permissions (permission,role_id,id) values ("red:key",2,21)
insert into permissions (permission,role_id,id) values ("red:vl",2,22)
insert into permissions (permission,role_id,id) values ("red:ts",2,23)

拟执行的JPQL查询 #

"SELECT r.roleName AS roleName, p.permission AS permissions " +
           "FROM Role r " +
           "LEFT JOIN r.permissions p " +
           "WHERE r.roleName in ('USER_ADMIN', 'DB_ADMIN')"

注：原JPQL中的双引号需改为单引号，符合JPQL语法规范

需求目标 #

将查询结果转换为如下结构的集合：

Map<String, List<String>> rolePermissionsMap = new HashMap<>();

其中键为角色名（如USER_ADMIN），值为该角色对应的所有权限字符串列表。

解决方案：Spring Data JPA投影实现 #

1. 定义投影接口 #

创建一个投影接口，映射JPQL查询返回的字段：

public interface RolePermissionProjection {
    // 对应JPQL中AS roleName的字段
    String getRoleName();
    // 对应JPQL中AS permissions的字段
    String getPermissions();
}

2. 在Repository中定义查询方法 #

在RoleRepository中添加带@Query注解的方法，返回投影接口列表：

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import java.util.List;

public interface RoleRepository extends JpaRepository<Role, Long> {
    @Query("SELECT r.roleName AS roleName, p.permission AS permissions " +
           "FROM Role r " +
           "LEFT JOIN r.permissions p " +
           "WHERE r.roleName IN ('USER_ADMIN', 'DB_ADMIN')")
    List<RolePermissionProjection> findRolePermissions();
}

3. 实体类关联关系确认 #

确保Role和Permission实体类已正确配置关联映射，否则JPQL中的r.permissions无法解析：

// Role实体
@Entity
@Table(name = "roles")
public class Role {
    @Id
    private Long id;
    @Column(name = "role_name")
    private String roleName;
    // 与Permission的一对多关联
    @OneToMany(mappedBy = "role")
    private List<Permission> permissions;
    
    // 构造器、getter、setter省略
}

// Permission实体
@Entity
@Table(name = "permissions")
public class Permission {
    @Id
    private Long id;
    private String permission;
    // 与Role的多对一关联
    @ManyToOne
    @JoinColumn(name = "role_id")
    private Role role;
    
    // 构造器、getter、setter省略
}

4. 转换查询结果为目标Map #

通过Java Stream API将投影列表分组转换为目标Map：

import java.util.Map;
import java.util.stream.Collectors;

// 注入RoleRepository
@Autowired
private RoleRepository roleRepository;

// 执行查询并转换
Map<String, List<String>> rolePermissionsMap = roleRepository.findRolePermissions()
        .stream()
        .collect(Collectors.groupingBy(
                RolePermissionProjection::getRoleName, // 按角色名分组
                Collectors.mapping(
                        RolePermissionProjection::getPermissions, // 提取权限字符串
                        Collectors.toList() // 收集为列表
                )
        ));

说明 #

• 由于JPQL的LEFT JOIN会返回多条数据（每个权限对应一条记录），通过Collectors.groupingBy可以将同一角色的所有权限聚合为一个列表。
• 投影接口仅需定义查询字段对应的getter方法，Spring Data JPA会自动实现该接口的代理类，无需手动编写实现类。

内容的提问来源于stack exchange，提问作者Peter Penzov