(Spring) DTO 에서 Entity 변환 후 테이블 저장 (ModelMapper 활용)

🌟 공부하게 된 계기

QNA 프로젝트를 공부하면서 DTO 를 많이 활용해보려고 하는데, 답변 저장 과정에서 입력받은 데이터를 DTO 형태로 저장하는 것 까지는 저번에 정리한 Entity 를 DTO 형태로 바꾸는 것과 비슷하게 할 수 있었다.

그런데 AnswerRepository 로 save 를 수행하려면 인자가 되는 객체가 Answer 이어야하는데 형태 데이터 객체는 DTO 였다. 물론 귀찮게 Answer 객체를 생성 후 하나하나 담아서 만들어낼 수 있겠지만 분명히 방법이 있겠지라고 생각하였다.

검색창에 "Spring DTO to Entity" 라고 검색을 하니 ModelMapper 를 통한 방식이 나와있었다. 해당 방법은 저번에 Entity to DTO 를 하면서도 궁금했던 방식이라 과정을 정리해보기로 했다.

참고 : https://www.baeldung.com/entity-to-and-from-dto-for-a-java-spring-application

 

1️⃣ ModelMapper 의존성 주입 및 Bean 등록

dependencies {
    // modelMapper 의존성 추가
    implementation group: 'org.modelmapper', name: 'modelmapper', version: '3.1.0'
}

@Configuration // Bean 등록을 위한 설정 파일임을 명시
public class AppConfig {
    @Bean
    public ModelMapper modelMapper() { // ModelMapper Bean 등록
        return new ModelMapper();
    }
}

@Configuration 어노테이션이 있어야 SpringBoot 에서 실행 전에 Bean 탐색이 가능하다.

2️⃣ DTO 생성

public class AnswerDto {
    @Setter
    @Getter
    public static class addAnswer {
        private String content;
        private LocalDateTime createDate;
        private Question question;

        public addAnswer (String content, Question question) { // 생성자
            this.content = content;
            this.question = question;
            this.createDate = LocalDateTime.now();
        }
    }
}

추후 값을 저장하고, 가져오는 과정들이 필요하기 때문에 Getter 와 Setter 는 반드시 있어야한다.

3️⃣ Controller 단

public class AnswerController {
// 			(생략)
    @PostMapping("/create/{id}")
    @ResponseBody
    public String createAnswer(@PathVariable int id, @RequestParam (defaultValue = "") String content) {
//			(생략)
        answerService.createAnswer(id,content);
        return "답변이 등록되었습니다.";
    }
}

4️⃣ Service 단

public class AnswerService {
    private final AnswerRepository answerRepository;
    private final QuestionRepository questionRepository;
    private final ModelMapper modelMapper;
    
    public void createAnswer(int id, String content) {
//			(생략)
        addAnswer addAnswer = new addAnswer(content, question);

        Answer answer = modelMapper.map(addAnswer, Answer.class); // Mapper 사용 부분

        answerRepository.save(answer); // Answer 객체 넘겨줌
    }
}

Controller 로 부터 넘겨받은 값들을 바탕으로 addAnswer DTO 를 생성하고, ModelMapper 를 통해 같은 속성들을 연결시켜 Answer 객체를 생성한 후 Repository 를 통해 값을 저장했다. 만약 ModelMapper 를 쓰지 않았다면 아래와 같이 귀찮은 과정을 거쳐야했을 것이다.

Answer answer = new Answer(addAnswer.getContent, addAnswer.getCreateDate, ... )

가독성도 높아지고, 혹시나하는 실수도 줄여줄 수 있는 

5️⃣ 결과 확인

알맞세 매핑되어 값이 저장된 것을 확인할 수 있다.