首页 > 解决方案 > 如何使用 Java 模型映射器将具有惰性属性的实体转换为 DTO

问题描述

我将 Spring JPA 与 hibernate、WebFlux 一起使用,但遇到了问题。我试图从数据库中获取一些实体,但是当我尝试将我的实体转换为 DTO 时发生延迟加载异常。我理解它为什么会发生,我可以通过为每个实体图创建 DTO 来避免它(只获取急切的属性),但这似乎有点矫枉过正。我也尝试使用setPropertyCondition()and跳过惰性属性Hibernate.isInitialized(),但是在模型映射器检查属性条件之前引发了异常。我试图在我的映射中返回实体对象,但我找不到杰克逊 Hibernate5Module 使用的正确配置。所以,我问你我应该如何使用@EntityGraph我的存储库以在控制器中没有错误,理想情况下延迟加载的属性应该为空。

我通过在 DTO 映射过程中跳过惰性属性解决了惰性初始化异常的问题。我为它创建了自己的值读取器并将其应用于 java 模型映射器实例。

public class CustomValueReader implements ValueReader<Object> {
    @Override
    public Object get(Object source, String memberName) {
        try {
            Field field = source.getClass().getDeclaredField(memberName);
            field.setAccessible(true);
            return field.get(source);
        }
        catch(Exception e) {
            return null;
        }
    }

    @Override
    public Member<Object> getMember(Object source, String memberName) {
        final Object value = get(source, memberName);
        Class<?> memberType = value != null ? value.getClass() : Object.class;
        return new Member<Object>(memberType) {
            @Override
            public Object getOrigin() {
                return null;
            }

            @Override
            public Object get(Object source, String memberName) {
                return CustomValueReader.this.get(source, memberName);
            }
        };
    }

    @Override
    @SuppressWarnings({ "unchecked", "rawtypes" })
    public Collection<String> memberNames(Object source) {
        List<String> properties = new ArrayList<>();
        for (Field field : source.getClass().getDeclaredFields()) {
            field.setAccessible(true);//needed for accessing field using reflection
            boolean valid = true;
            Object fieldValue = null;
             try{
                 fieldValue = field.get(source);
             }
             catch(Exception e) {
                 valid = false;
             }
            boolean isInited = valid && Hibernate.isInitialized(fieldValue);
             if (isInited){
                 properties.add(field.getName());
             }
        }
        return properties;
    }

    @Override
    public String toString() {
        return "Costum";
    }
}

将其应用于映射器

mapper.getConfiguration().addValueReader(new CustomValueReader());

我还注意到实体图的使用使所有属性急切或懒惰(取决于 EntityGraph.EntityGraphType)忽略属性节点。我的班级定义看起来像

@NamedEntityGraphs({
        @NamedEntityGraph(name = ClinicVisit.CLINIC_VISIT_ENTITY_GRAPH_WITH_PATIENT_AND_DEVICES,
                attributeNodes = @NamedAttributeNode(value = "patient",subgraph = "CVEG_DEVICES"),
                subgraphs = @NamedSubgraph(name = "CVEG_DEVICES", attributeNodes = @NamedAttributeNode("notificationDevices"))
        ),
        @NamedEntityGraph(name = ClinicVisit.CLINIC_VISIT_ENTITY_GRAPH_WITH_PATIENT,
                attributeNodes = @NamedAttributeNode(value = "patient")
        )
})


@Entity
@Table(name="clinic_visit_request")
@Getter
@Setter
public class ClinicVisit {

    public final static String CLINIC_VISIT_ENTITY_GRAPH_WITH_PATIENT_AND_DEVICES = "clinic_visit_request_patient_devices";
    public final static String CLINIC_VISIT_ENTITY_GRAPH_WITH_PATIENT = "clinic_visit_request_patient";

    @Id
    @GeneratedValue
    protected Long id;

    @Column(name = "visit_date")
    protected Date visitDate;

    @Column(name = "update_date")
    protected Date updateDate;

    protected Boolean informed;

    @Column( columnDefinition = "boolean default false", nullable = false)
    protected Boolean handled = false;

    @ManyToOne()
    protected ClinicUser doctor;

    @ManyToOne()
    protected ClinicUser patient;

    @ManyToOne()
    protected ClinicUser confirmer;
}

我的存储库方法定义:

@EntityGraph(value=ClinicVisit.CLINIC_VISIT_ENTITY_GRAPH_WITH_PATIENT,type= EntityGraph.EntityGraphType.FETCH)
   List<ClinicVisit> findAll();

一切正常,这是我的误解。我有三个字段(医生、患者、确认者)并在其中保存相同的用户,所以当我为患者添加 NamedAttributeNode 时,我预计只会加载患者,但由于休眠很聪明,所以加载了其他字段。

我明白为什么 Hibernate5Module 不起作用。我使用 WebFlux,所以 JacksonMapper 的配置应该如下所示

@Configuration
public class AppConfig {

@Bean
Jackson2ObjectMapperBuilderCustomizer jackson2ObjectMapperBuilderCustomizer(){
    return jacksonObjectMapperBuilder -> jacksonObjectMapperBuilder
            .featuresToEnable(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS)
            .propertyNamingStrategy(PropertyNamingStrategy.SNAKE_CASE)
            .modules(new Hibernate5Module());
}


@Bean
Jackson2JsonEncoder jackson2JsonEncoder(ObjectMapper mapper){
    return new Jackson2JsonEncoder(mapper);
}

@Bean
Jackson2JsonDecoder jackson2JsonDecoder(ObjectMapper mapper){
    return new Jackson2JsonDecoder(mapper);
}

@Bean
WebFluxConfigurer webFluxConfigurer(Jackson2JsonEncoder encoder, Jackson2JsonDecoder decoder) {
    return new WebFluxConfigurer() {
        @Override
        public void configureHttpMessageCodecs(ServerCodecConfigurer configurer) {
            configurer.defaultCodecs().jackson2JsonEncoder(encoder);
            configurer.defaultCodecs().jackson2JsonDecoder(decoder);
        }
    };
}

}

标签: javaspringhibernatelazy-loadingmodelmapper

解决方案


设置 Hibernate5Module 就像创建一个 bean 一样简单:

@Bean
public com.fasterxml.jackson.databind.Module datatypeHibernateModule() {
    return new Hibernate5Module();
}

https://stackoverflow.com/a/37840526/10609423

与 Spring Boot 2.1.2 完美配合。如果可以的话,我会发表评论。


推荐阅读