如何调整Spring Data JPA下JPQL查询的输出格式为目标键值对结构

以下是两种常用的实现方式，均可满足你的需求：

方式一：Service层直接转换现有结果（改造成本最低） #

无需修改现有Repository代码，只需要在业务层将返回的List<Object[]>转换为Map<String, Long>即可，示例代码如下：

Service层转换代码 #

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

// 在你的Service类中编写方法
public Map<String, Long> countPetByStatus() {
    List<Object[]> queryResult = petRepository.countByStatus();
    // 流式处理将数组转为键值对
    return queryResult.stream()
            .collect(Collectors.toMap(
                    item -> (String) item[0], // 数组第一个元素为状态名，作为Map的key
                    item -> (Long) item[1]  // 数组第二个元素为统计数量，作为Map的value
            ));
}

Controller层代码 #

直接返回转换后的Map即可，Spring Boot内置的Jackson序列化工具会自动将其转换为你需要的JSON键值对结构：

import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.Map;

@RestController
@RequestMapping("/pet")
public class PetController {

    private final PetService petService;

    // 构造函数注入
    public PetController(PetService petService) {
        this.petService = petService;
    }

    @GetMapping("/status-count")
    public ResponseEntity<Map<String, Long>> getStatusCount() {
        Map<String, Long> result = petService.countPetByStatus();
        return ResponseEntity.ok(result);
    }
}

方式二：修改Repository直接返回结构化结果（避免硬编码数组索引） #

可以使用JPA的NEW map构造语法，直接让查询返回结构化的KV集合，避免数组索引硬编码导致的维护问题：

修改Repository代码 #

@Query("SELECT NEW map(v.status.name as status, count(v) as count) FROM Pet v GROUP BY v.status.name")
List<Map<String, Object>> countByStatus();

Service层转换代码 #

public Map<String, Long> countPetByStatus() {
    List<Map<String, Object>> queryResult = petRepository.countByStatus();
    return queryResult.stream()
            .collect(Collectors.toMap(
                    item -> (String) item.get("status"),
                    item -> (Long) item.get("count")
            ));
}

注意事项 #

• JPA的count()函数返回值为Long类型，如果你需要返回整数类型可以手动强转：((Long)item[1]).intValue()
• 由于你已经使用GROUP BY对状态名分组，所以不会出现重复key的问题，无需额外处理Collectors.toMap的键冲突逻辑

内容的提问来源于stack exchange，提问作者Craig Armstrong