프록시 객체의 클래스 타입 비교 문제

    public static String getEntityType(BaseEntity entityObj) {
        for (EntityTypeV2 entity : EntityTypeV2.values()) {
            if (entityObj.getClass().isInstance(entity.entityObj)) {
                return entity.entityType;
            }
        }
        throw new BusinessException(HttpResponse.Fail.NOT_FOUND);
    }

enum에서 사용하는 메서드인데 class 타입을 비교해주는 메서드이다.

 

프로젝트에서 이미지를 s3에 저장하거나 db에서 불러올 때 entity의 타입이 필요로 하는데

편의성을 위해 BaseEntity를 통해서 처리를 하는 기능을 만들었었다.

 

이때까지 잘 작동은 했으나 프록시 객체에서 문제가 발생했다.

 

    public Page<ReviewOrderDetailResponse> getUnwrittenReviews(Member member, int page, int size) {
        log.info("[ReviewServiceImpl.getReviews] : 접근");
        Pageable pageable = PageRequest.of(page, size);
        Page<OrderDetail> findAll = orderDetailRepository.findOrderDetailsMember(member, pageable);
        return findAll.map(orderDetail -> ReviewOrderDetailResponse.of(orderDetail, fileService.findOne(orderDetail.getItem())));
    }

위와 같은 코드에서

 

fileService.findOne(orderDetail.getItem())

이부분이 문제다

orderDetail안에 있는 item을 param으로 사용하는데 프록시 객체로 가기에

enum클래스에서 비교가 item이랑 되야하는데 매개변수로 전달된 클래스의 타입은 Item$HibernateProxy$iBj96ep4 와 같이 찍힌다..

 

 

나는 이 문제의 해결은 아래와 같이 했다.

    public static String getEntityType(BaseEntity entityObj) {
        Class<?> classType = getClassType(entityObj);

        for (EntityTypeV2 entity : EntityTypeV2.values()) {
            if (classType.isInstance(entity.entityObj)) {
                return entity.entityType;
            }
        }

        throw new BusinessException(HttpResponse.Fail.NOT_FOUND);
    }

    private static Class<?> getClassType(BaseEntity entityObj) {
        if (entityObj instanceof HibernateProxy) {
            HibernateProxy hProxy = (HibernateProxy) entityObj;
            LazyInitializer initializer = hProxy.getHibernateLazyInitializer();
            return initializer.getPersistentClass();
        }

        return entityObj.getClass();
    }

getClassType 메서드를 만들었고 이를 통해 proxy 타입의 클래스일 경우 실제 클래스 타입을 반환하게 처리하니 정상적으로 작동한다.