[Querydsl] 서브쿼리 사용하기

안녕하세요!
이번 시간에는 Querydsl에서의 Subquery 기본 가이드를 진행합니다.

개인적으로 ORM을 사용하며, 객체지향적으로 엔티티가 구성되어있으면 서브쿼리가 필요한 일은 거의 없다고 생각하는데요.

혹시나 필요한 분들이 계시면 도움이 되셨으면 합니다.

그럼 시작합니다!

모든 코드는 Github에 있으니 참고하세요 :)

1. select Sub Query

첫번째는 select절의 서브쿼리입니다.
도메인에 대한 설명보다는 Querydsl 코드가 좀 더 도움이 되실것 같아 코드로 바로 설명드리겠습니다.

Spring Data Jpa에서 Querydsl 적용을 어떻게 하는지는 이전 포스팅을 참고해보세요 :)

Querydsl의 코드는 아래와 같습니다.
여기서 주의깊게 봐야할 코드는 ExpressionUtils.as() 입니다.

@Override
    public List<StudentCount> findAllStudentCount() {
        return queryFactory
                .select(Projections.fields(StudentCount.class,
                        academy.name.as("academyName"),
                        ExpressionUtils.as(
                                JPAExpressions.select(count(student.id))
                                        .from(student)
                                        .where(student.academy.eq(academy)),
                                "studentCount")
                ))
                .from(academy)
                .fetch();
    }

ExpressionUtils은 Querydsl 내부에서 새로운 Expression을 사용할 수 있도록 지원합니다.
여기서 as를 통해 서브쿼리의 결과물을 alias 시킵니다.

예전에는 JPASubQuery를 통해 만들었지만, 버전업이 되어 현재는 JPAExpressions를 통해 서브쿼리를 생성합니다.

실제로 아래와 같이 테스트 코드를 작성해서

@RunWith(SpringRunner.class)
@SpringBootTest
public class SubQueryTest {

    @Autowired
    private AcademyRepository academyRepository;

    @Autowired
    private StudentRepository studentRepository;

    @After
    public void tearDown() throws Exception {
        studentRepository.deleteAll();
        academyRepository.deleteAll();
    }

    @Test
    public void SELECT절_서브쿼리로_사용된다() {
        //given
        String academyName1 = "name1";
        Academy academy1 = Academy.builder()
                .address("address1")
                .name(academyName1)
                .build();

        academy1.addStudent(Arrays.asList(
                new Student("student1", 1),
                new Student("student2", 2)
        ));

        String academyName2 = "name2";
        Academy academy2 = Academy.builder()
                .address("address2")
                .name(academyName2)
                .build();

        academy2.addStudent(Arrays.asList(
                new Student("student3", 3),
                new Student("student4", 4),
                new Student("student5", 5)
        ));

        academyRepository.saveAll(Arrays.asList(academy1, academy2));

        //when
        List<StudentCount> result = academyRepository.findAllStudentCount();

        //then
        assertThat(result.get(0).getAcademyName(), is(academyName1));
        assertThat(result.get(1).getAcademyName(), is(academyName2));
    }
}

그리고 테스트를 수행해보면!

아주 잘 수행되는 것을 확인할 수 있습니다.
select절의 서브쿼리가 잘 된거겠죠?

2. where Sub Query

두번째로는 where절의 서브쿼리입니다.
상당히 자주 쓰는 방식이라고 생각합니다.

실제로 레거시 코드를 까보면 쿼리매퍼들 (MyBatis 등) 에서 정말 많이들 사용하고 계셨더라구요

사용하는 방법은 다음과 같습니다.

    @Override
    public List<Academy> findAllByStudentId(long studentId) {
        return queryFactory
                .selectFrom(academy)
                .where(academy.id.in(
                        JPAExpressions
                                .select(student.academy.id)
                                .from(student)
                                .where(student.id.eq(studentId))))
                .fetch();
    }

앞서 select와 달리 ExpressionUtils이 필요 없게 되었습니다.
JPAExpressions만으로 서브쿼리를 생성하시면 됩니다.

테스트 코드를 작성해서 확인해보면!

@Test
    public void WHERE절_서브쿼리로_사용된다() {
        //given
        String academyName1 = "name1";
        Academy academy1 = Academy.builder()
                .address("address1")
                .name(academyName1)
                .build();

        academy1.addStudent(Arrays.asList(
                new Student("student1", 1),
                new Student("student2", 2)
        ));

        String academyName2 = "name2";
        Academy academy2 = Academy.builder()
                .address("address2")
                .name(academyName2)
                .build();

        academy2.addStudent(Arrays.asList(
                new Student("student3", 3),
                new Student("student4", 4),
                new Student("student5", 5)
        ));

        academyRepository.saveAll(Arrays.asList(academy1, academy2));

        //when
        List<Academy> result = academyRepository.findAllByStudentId(3);

        //then
        assertThat(result.size(), is(1));
        assertThat(result.get(0).getName(), is(academyName2));
    }

아래와 같이 성공되는 것을 확인할 수 있습니다.