在spring boot get方法中为响应json添加字段

Question

我有一个将数据从存储库提供给rest控制器的服务：

@Service
public class TaskServiceImpl implements TaskService {
    @Autowired
    private TaskRepository taskRepository;

    @Override
    public List<Task> getAllTasks() {
        return taskRepository.findAll();
    }
}

和rest控制器:@RestController@RequestMapping（“/tasks”）public class TaskController{@Autowired private TaskService TaskService；

    @GetMapping
    public List<Task> getAllTasks() {
        return taskService.getAllTasks();
    }
}

我的任务是不仅返回所有任务，而且返回两个字段Two-Todo tasks count和ready tasks count。我知道如何从数据库中找到这个计数。但是，将这些字段添加到response JSON中的正确方法是什么呢？响应json必须如下所示：

{
  [
    {
      "createTime": null,
      "updateTime": null,
      "id": "3fa85f64-5717-4562-b3fc-2c963f66afa6",
      "title": "todo-task",
      "description": "blabla",
      "priority": "HIGH",
      "done": false,
    },
    {
      "createTime": null,
      "updateTime": null,
      "id": "3fa85f64-5717-4562-b3fc-2c963f66afa6",
      "title": "done-task",
      "description": "blabla",
      "priority": "HIGH",
      "done": true,
    }
  ],
  todoCount: 1,
  doneCount: 1
}

Answer 1

可以通过创建要由控制器返回的新模型并设置todoCount和doneCount值来实现：

 @GetMapping
    public TasksModel getAllTasks() {
// get todoCount and doneCount values
          TasksModel tasksModel = new TasksModel();
          tasksModel.setTaskModelList(taskService.getAllTasks())
          tasksModel.setTodoCount(todoCount);
          tasksModel.setDoneCount(doneCount); 

        return taskModel;
    }

而TasksModel为：

class TasksModel  {
     
     List<Task> taskModelList;
     int todoCount;
     int doneCount;

//getter
//setter

}