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

안녕하세요 상속관계 엔티티를 다루는 방법에 대해 궁금한 점 있어 질문 드립니다

상속관계의 엔티티가 존재하고, 자식 엔티티마다 서비스 로직이 다를 수 있을 때 어떻게 분기처리하는지 알고 싶어 질문 드립니다

@Entity
@NoArgsConstructor(access = AccessLevel.PROTECTED)
@Getter
@Inheritance(strategy = InheritanceType.SINGLE_TABLE)
@DiscriminatorColumn(name = "base_type")
public abstract class BaseEntity {

    @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    private String title;

    private String description;

    @Column(name = "base_type", updatable = false, insertable = false, length = 30)
    private String baseType;

    protected BaseEntity(Long id, String title, String description) {
        this.id = id;
        this.title = title;
        this.description = description;
    }
}

@Entity
@DiscriminatorValue(value = "FIRST")
@NoArgsConstructor(access = AccessLevel.PROTECTED)
@Getter
public class FirstEntity extends BaseEntity{

    private int like;

    @Builder
    public FirstEntity(Long id, String title, String description, int like) {
        super(id, title, description);
        this.like = like;
    }
}        

@Entity
@DiscriminatorValue(value = "SECOND")
@NoArgsConstructor(access = AccessLevel.PROTECTED)
@Getter
public class SecondEntity extends BaseEntity {

    private String redirectUrl;

    @Builder
    public SecondEntity(Long id, String title, String description, String redirectUrl) {
        super(id, title, description);
        this.redirectUrl = redirectUrl;
    }
}

이러한 상속구조 엔티티가 있고, 두 자식의 서비스 로직이 다를 수 있을 때
컨트롤러에서 분기처리를 어떻게 하는지 알고 싶습니다

@RestController
@RequiredArgsConstructor
@RequestMapping("/v1/test")
public class TestController {

    private final TestService testService;

    @GetMapping
    public void findEntity(@RequestParam String type) {

        if(type.equalsIgnoreCase("first")) {

        } else if(type.equalsIgnoreCase("second")) {
            
        }
        //...
        //...
        //...
    }
}

만약 이러한 구조로 간다면, 자식클래스가 늘어났을 때 또 if문을 추가해야해서 효율적이지 못할 것 같아서요!

지금 생각한 것은 상태 패턴을 이용하면 어떨까 하는데, 이게 좋은지 잘 모르겠습니다..

public interface BaseContext {

    boolean support(String type);

    void executeFind();
}

이렇게 인터페이스를 만들어주고

@Component
public class FirstContext implements BaseContext{

    private static final String TYPE = "FIRST";
    private final FirstService firstService;

    public FirstContext(FirstService firstService) {
        this.firstService = firstService;
    }

    @Override
    public boolean support(String type) {
        return TYPE.equalsIgnoreCase(type);
    }

    @Override
    public void executeFind() {
        firstService.find();
    }
}

자식클래스의 상태를 구현하는 클래스를 만들어주고

@RestController
@RequiredArgsConstructor
@RequestMapping("/v1/test")
public class TestController {
    private final List<BaseContext> contexts;

    @GetMapping
    public void findEntity(@RequestParam String type) {

        BaseContext findContext = contexts.stream()
                .filter(context -> context.support(type))
                .findAny()
                .orElseThrow(() -> new IllegalStateException("Error"));
        
        findContext.executeFind();
    }
}

컨트롤러에서는 이렇게 받아주면

자식 엔티티가 늘어나도 if-else를 하지 않아도 된다고 생각했습니다

다만, 자식 엔티티가 추가될 때 마다 Context를 추가해야 한다는 점과

Context가 Service를 바라보고있어 Service가 수정되면 Context와 Service 2가지 모두 수정해야한다는 점이 마음에 걸립니다

보통 상속관계의 엔티티를 다룰 때 어떤식으로 분기처리를 하나요??

강사님이 아니시더라도 다른 수강생분들의 의견도 달아주시면 좋겠습니다!!