인프런 커뮤니티 질문&답변

이번 jpashop 프로젝트에서는 book 엔티티만 등록할  수 있게 했는데 만약에 등록폼에서 책, 앨범, 영화를 선택할 수 있게하여 데이터를 받을 때 dto를 이용한다고 가정해서 총 4종류의 dto(아이템, 책, 영화 ,앨범)를 만들었습니다.

그렇게 하여 아이템 dto가 부모 타입이고 나머지 3개의 dto는 자식으로 dto끼리 상속관계를 맺어주고 컨트롤러에서는 다형성을 이용해 아이템dto로 3가지 데이터를 다 받을 수 있게 하는게 맞나요?

 

@Data
class ItemDTO {...}

@Data
class MovieDTO extends ItemDTO{...}

@Data
class AlbumDTO extends ItemDTO{...}

@Data
class BookDTO extends ItemDTO{...}


// == 컨트롤러 ==//

@Controller
class ItemController {

    @PostMapping
    public String create(@ModelAttribute ItemDTO itemDTO) {
        ...
        return "redirect:/";
    }
}

 

만약에 이 방식이 맞다면 itemDTO 타입으로 넘어온 객체가 타입에 맞는 엔티티에 값을 전달해줘야 하는데. 이 때 컨트롤러의 create 메서드에서 itemDTO 인자에  앨범dto가 들어왔는지, 책dto가 들어왔는지, 영화dto가 들어왔는지 어떻게 구별할까요..?

 

일단 제가 생각한 방식에 대해서 말씀드리겠습니다.

@Data
class ItemDTO {
    private String type;
}

@Data
class MovieDTO extends ItemDTO {
    private String type = "M";

}

@Data
class AlbumDTO extends ItemDTO {
    private String type = "A";
}

@Data
class BookDTO extends ItemDTO {
    private String type = "B";
}

// ==  컨트롤러 == //
@Controller
class ItemController {

    @Autowired ItemService itemService;

    @PostMapping
    public String create(@ModelAttribute ItemDTO itemDTO) {

        itemService.createItemEntity(itemDTO);

        return "redirect:/";

    }
}

// == 서비스  == //
@Service
class ItemService {

    public void createItemEntity(ItemDTO itemDTO) {
        switch (itemDTO.getType()) {
            case "A":
                // 앨범 엔티티 생성
                break;

            case "B":
                // 책 엔티티 생성
                break;

            case "M":
                // 영화 엔티티 생성
                break;
        }

    }
}

itemDTO 객체에 type 필드 값을 만들어서 부모객 타입으로 다뤄도 어느 자식 타입인지 구별할 수 있도록 했습니다.

먼저 등록폼에서 만약에 영화를 선택할 경우  영화dto의 type 필드 값에 'M'을 같이 넘겨줍니다. 

그리고 컨트롤에서는 등록폼으로 받은 itemDTO 객체를 서비스로 보냅니다.

서비스에서는  넘어온 itemDTO 객체의 type 필드 값을 switch문으로 확인해서 "A" 인 경우는 앨범 엔티티를 생성하고 "B"인 경우에는 책 엔티티를 생성하는 방식입니다.

 

실무에서는 어떤 방식으로 구현하는지 궁금합니다.