diff --git a/src/main/java/com/example/jpapractice/controller/StudentController.java b/src/main/java/com/example/jpapractice/controller/StudentController.java index 257dc42..f00123a 100644 --- a/src/main/java/com/example/jpapractice/controller/StudentController.java +++ b/src/main/java/com/example/jpapractice/controller/StudentController.java @@ -1,5 +1,6 @@ package com.example.jpapractice.controller; +import com.example.jpapractice.dto.StudentDto; import com.example.jpapractice.entity.Student; import com.example.jpapractice.service.StudentService; import lombok.RequiredArgsConstructor; @@ -30,7 +31,7 @@ public ResponseEntity createStudent(@RequestBody Student student) { * @return 학생 목록 */ @GetMapping - public ResponseEntity> getAllStudents() { + public ResponseEntity> getAllStudents() { return ResponseEntity.ok(studentService.getAllStudents()); } @@ -40,7 +41,7 @@ public ResponseEntity> getAllStudents() { * @return 학생 정보 */ @GetMapping("/{id}") - public ResponseEntity getStudentById(@PathVariable Long id) { + public ResponseEntity getStudentById(@PathVariable Long id) { return ResponseEntity.ok(studentService.getStudentById(id)); } @@ -51,7 +52,7 @@ public ResponseEntity getStudentById(@PathVariable Long id) { * @return 수정된 학생 정보 */ @PutMapping("/{id}") - public ResponseEntity updateStudent( + public ResponseEntity updateStudent( @PathVariable Long id, @RequestBody Student updatedStudent) { return ResponseEntity.ok(studentService.updateStudent(id, updatedStudent)); diff --git a/src/main/java/com/example/jpapractice/service/StudentService.java b/src/main/java/com/example/jpapractice/service/StudentService.java index 8e4f878..37aa0d6 100644 --- a/src/main/java/com/example/jpapractice/service/StudentService.java +++ b/src/main/java/com/example/jpapractice/service/StudentService.java @@ -55,11 +55,12 @@ public StudentDto getStudentById(Long id) { * @return 수정된 학생 정보 */ @Transactional - public Student updateStudent(Long id, Student updatedStudent) { - Student existingStudent = getStudentById(id); + public StudentDto updateStudent(Long id, Student updatedStudent) { + StudentDto existingStudent = getStudentById(id); existingStudent.setName(updatedStudent.getName()); existingStudent.setAge(updatedStudent.getAge()); - existingStudent.setClassRoom(updatedStudent.getClassRoom()); + existingStudent.setClassRoomName(updatedStudent.getClassRoom().getName()); + existingStudent.setClassRoomId(updatedStudent.getClassRoom().getId()); return existingStudent; }