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

안녕하세요, 배운 내용을 기반으로 스스로 개발 연습을 하고 있었습니다.

Article이라는 엔티티에 본문에 첨부되어 뷰에서 보일 이미지 파일들과 버튼을 누르면 다운로드가 가능한 첨부 파일들을 동시에 넣고자 했습니다.

이미지 파일과 첨부 파일은 모두 UploadFile이라는 엔티티를 사용하였습니다.

서버에서 실험차 Article 엔티티를 작성해보니, 이미지 파일과 첨부 파일 리스트가 각각 적용 되는 것이 아니라 합쳐졌습니다.

예를 들어 이미지 파일을 2개 등록하고 첨부 파일을 3개 등록하였으면, imageFiles와 attachedFiles List의 사이즈가 각각 2, 3개 인 것이 아니라 똑같이 5, 5개였습니다.

이미지 파일과 첨부 파일이 다른 종류의 파일임에도 불구하고 결국 같은 엔티티를 공유하고 있어서 발생한 문제인걸까요?

현재 생각하고 있는 해결책은 이미지 파일과 첨부 파일을 각각 다른 엔티티로 만들어서 따로 테이블을 관리하면 되겠다고 생각중입니다.

하지만 혹시 이보다 더 좋거나 근본적인 해결책이 있지 않을까 싶어서 질문을 올려봅니다.

두 엔티티의 코드는 다음과 같은데 혹시 다른 코드가 더 필요한 것이라면 올리겠습니다!

// 게시글 엔티티
@Entity
@Getter
public class Article {

    @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
    @Column(name = "article_id")
    private Long id;

    private String writer;
    private String title;
    @Lob
    private String content;

    @OneToMany(mappedBy = "article", cascade = CascadeType.ALL, orphanRemoval = true)
    private List<UploadFile> imageFiles = new ArrayList<>();

    @OneToMany(mappedBy = "article", cascade = CascadeType.ALL, orphanRemoval = true)
    private List<UploadFile> attachedFiles = new ArrayList<>();

    public static Article createArticle(ArticleAddForm articleAddForm, List<UploadFile> imageFiles, List<UploadFile> attachedFiles) {

        Article article = new Article();
        article.writer = articleAddForm.getWriter();
        article.title = articleAddForm.getTitle();
        article.content = articleAddForm.getContent();
        article.imageFiles.addAll(imageFiles);
        article.attachedFiles.addAll(attachedFiles);

        for (UploadFile imageFile : imageFiles) {
            imageFile.setArticle(article);
        }
        for (UploadFile attachedFile : attachedFiles) {
            attachedFile.setArticle(article);
        }

        return article;
    }
}

// 업로드 파일 엔티티 (이미지 파일, 첨부 파일의 엔티티로서 동시에 사용됨)
@Entity
@Getter
public class UploadFile {

    @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
    @Column(name = "upload_file_id")
    private Long id;

    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "article_id")
    private Article article;

    private String originalName;
    private String storeName;

    public UploadFile() {
    }

    public UploadFile(String originalName, String storeName) {
        this.originalName = originalName;
        this.storeName = storeName;
    }

    public void setArticle(Article article) {
        this.article = article;
    }
}