[Java] 쿼리는 웬만하면 적게! 그리고 데이터 컬렉션 처리 리팩토링 과정 (Map, stream API, 연산메서드)

JPA로 개발 중에 레포에서 가져오는데 쿼리를 적게 날리고 싶었다.
일단 다 가져온 다음에 분리작업을 해서 dto 에 넣는게 더 효율적일 것 같다는 생각에 리팩토링을 진행함.

코드 비교

조건을 걸어서 해당 조건에 충족하는 데이터만 가져오려고 했다.
근데 다 가져와야하니까 3번 다 날렸다.

조건절 넣어서 쿼리 3번 날리던 기존 방법

public ResponseSchoolDto viewSchool(Long userId) {

        // schoolType (HIGH, UNI, GRAD) 에 따라 리스트를 다르게 보낸다.
        List<School> highSchools = schoolRepository.findByUserUserIdAndSchoolType(userId, SchoolType.HIGH);
        List<School> uniSchools = schoolRepository.findByUserUserIdAndSchoolType(userId, SchoolType.UNI);
        List<School> gradSchools = schoolRepository.findByUserUserIdAndSchoolType(userId, SchoolType.GRAD);
        
        List<SchoolDto> highSchoolDtos = highSchools.stream()
                .map(SchoolDto::from)
                .collect(Collectors.toList());
        List<SchoolDto> uniSchoolDtos = uniSchools.stream()
                .map(SchoolDto::from)
                .collect(Collectors.toList());
        List<SchoolDto> gradSchoolDtos = gradSchools.stream()
                .map(SchoolDto::from)
                .collect(Collectors.toList());

        return ResponseSchoolDto.builder()
                .HIGH(highSchoolDtos)
                .UNI(uniSchoolDtos)
                .GRAD(gradSchoolDtos)
                .build();

}

filter() 추가

일단 다 가져온 School 리스트에서 schoolType을 가져온 후 그 아이가 내가 원하는 애면 들어가도록 해줬다.

public ResponseSchoolDto viewSchool(Long userId) {

        // 한 번에 다 받아와서 filter로 구분하기
        List<School> allSchools = schoolRepository.findByUserUserId(userId);

        List<SchoolDto> highSchoolDtos = allSchools.stream()
                .filter(school -> school.getSchoolType() == SchoolType.HIGH)
                .map(SchoolDto::from)
                .collect(Collectors.toList());
        List<SchoolDto> uniSchoolDtos = allSchools.stream()
                .filter(school -> school.getSchoolType() == SchoolType.UNI)
                .map(SchoolDto::from)
                .collect(Collectors.toList());
        List<SchoolDto> gradSchoolDtos = allSchools.stream()
                .filter(school -> school.getSchoolType() == SchoolType.GRAD)
                .map(SchoolDto::from)
                .collect(Collectors.toList());

        return ResponseSchoolDto.builder()
                .HIGH(highSchoolDtos)
                .UNI(uniSchoolDtos)
                .GRAD(gradSchoolDtos)
                .build();
    }

결과

쿼리 한번으로 가져온 후 필터 처리해서 같은 결과가 나옴을 확인할 수 있따!.
쉽게 보자면 db에서 분기처리를 해서 가져오느냐, 가지고 와서 java에서 처리하느냐 차이인 듯하다.
이렇게 다 ~ 가져와야 하는 경우에는 쿼리를 3번 보내기보다는 java에서 처리하는 게 낫다고 판단해서 이렇게 리팩토링을 했다.

팀원들과 의논해 본 결과 한번에 가져올 수 있는 쿼리는 이렇게 가져오는게 당연히 맞다고 결론이 났다.

+추가로 코드 리팩토링

filter도 필요없이 코드를 짜보겠다.

위에서는 필터로 나누면서 얘가 여기면 여기! 이렇게 짰는데 <<이게 아니라
하나씩 돌면서 얘는 여기네 <<이렇게 찾게 하고 싶었다.

List<SchoolDto> highSchoolDtos = allSchools.stream()
                .filter(school -> school.getSchoolType() == SchoolType.HIGH)
                .map(SchoolDto::from)
                .collect(Collectors.toList());
List<SchoolDto> uniSchoolDtos = allSchools.stream()
                .filter(school -> school.getSchoolType() == SchoolType.UNI)
                .map(SchoolDto::from)
                .collect(Collectors.toList());
List<SchoolDto> gradSchoolDtos = allSchools.stream()
                .filter(school -> school.getSchoolType() == SchoolType.GRAD)
                .map(SchoolDto::from)
                .collect(Collectors.toList());

위에서 코드를 이렇게 짰는데 고민하다 GPT의 도움을 빌려봤다.
근데 개쩌는 코드가 나왔다.

무슨 컬렉션을 쓰면 되게요???

두근두근 ...

- Map 을 쓰자 !
- List.collect(Collectors.groupingBy()) 라는 것이 있단다.

// 이렇게 하면 스쿨타입별로 리스트가 각각 존재할 수 있게 된다...
Map<SchoolType, List<SchoolDto>> schoolDtoMap = allSchools.stream()
                .map(SchoolDto::from)
                .collect(Collectors.groupingBy(schoolDto -> SchoolType.valueOf(schoolDto.getSchoolType())));

// 그리고 리턴해줄때 get해서 꺼내가면됨. 미쳐따
return ResponseSchoolDto.builder()
                .HIGH(schoolDtoMap.get(SchoolType.HIGH))
                .UNI(schoolDtoMap.get(SchoolType.UNI))
                .GRAD(schoolDtoMap.get(SchoolType.GRAD))
                .build();

현재 SchoolType 이 Enum 타입이라 그렇지 일반 변수였으면 훨씬 더 쉽게 빼냈을 것이다.
Collectors 메서드들을 잘 살피면 순환돌 때 꽤나 유용한 것이 많겠다.

결론

이번 포스팅에서는
Collection 클래스의 stream 변환부터
Stream API의 filter, map, collect 그리고
Collectors 의 groupingBy 까지 사용하면서
응답값을 처리하는 과정에 대해 알아보았다.

Collectors 클래스 이거 대단한 자식이다.
컬렉션을 잘 활용할 줄 알아야겠다. 허구한 날 List만 쓰다니 .. Map이나 Set도 자주 기억해주자!