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

안녕하세요.

강의 내용과 관련 없는 내용이라 질문을 해야 하나말아야하나 고민하다가

돌려보시는 분들도 이게 정상적으로 결과가 나오지 않을 거 같아서... 질문 드립니다.

올려주신 강의 자료 파일(pdf)의 List<OrderFlatDto>로 만든 부분을 List<OrderQueryDto>로 변환하는 소스는 

아래과 같은데요.

return flats.stream()
        .collect(groupingBy(o -> new OrderQueryDto(o.getOrderId(),
                        o.getName(), o.getOrderDate(), o.getOrderStatus(), o.getAddress()),
                mapping(o -> new OrderItemQueryDto(o.getOrderId(),
                        o.getItemName(), o.getOrderPrice(), o.getCount()), toList())
        )).entrySet().stream()
        .map(e -> new OrderQueryDto(e.getKey().getOrderId(),
                e.getKey().getName(), e.getKey().getOrderDate(), e.getKey().getOrderStatus(),
                e.getKey().getAddress(), e.getValue()))
        .collect(toList());

이렇게 되면 OrderQueryDto가 Key이고, List<OrderItemQueryDto>가 Value로 맵을 만들어 다시 OrderQueryDto를 생성하는 식으로 되는 것 같은데... 저는 결과가 2개의 OrderQueryDto로 합쳐지지않고, 4개의 OrderQueryDto로 나오더라구요.

그래서 아래와 같이 바꿔봤습니다.

return new ArrayList<>(flats.stream()
        .collect(toMap(OrderFlatDto::getOrderId, o -> {
            // Key : OrderId, Value : OrderQueryDto (OrderItemQueryDto 1건 포함)
            OrderQueryDto orderQueryDto = new OrderQueryDto(o.getOrderId(), o.getName(), o.getOrderDate(), o.getOrderStatus(), o.getAddress());
            orderQueryDto.getOrderItems().add(new OrderItemQueryDto(o.getOrderId(), o.getItemName(), o.getOrderPrice(), o.getCount()));
            return orderQueryDto;
        }, (e1, e2) -> {
            e1.getOrderItems().addAll(e2.getOrderItems());
            return e1;
        })).values());

뭐 좀 별로인 부분도 있지만 이렇게 수정하니 원하는 결과 (OrderQueryDto가 2개)가 나왔습니다.

다른 방법도 많고 JPA의 중요한 부분도 아니고 강의 들으시는 분들은 어떻게 출력되어야 하는지 이미 다 알고 계시겠지만...

혹시나 저처럼 다 돌려보고 확인해야 하는 사람에겐 중요할 것 같아서....

제가 잘못한거거나 이미 수정하신 부분이시라면 죄송합니다.