如何使用ModelMapper进行深度继承对象?

时间:2018-05-19 10:06:50

标签: java dto modelmapper

A.java

@Entity
@Getter
@Setter
@Inheritance
@JsonTypeInfo(use=JsonTypeInfo.Id.NAME, inclue=JsonTypeInfo.As.PROPERTY, property="type") 
@JsonSubTypes({
    @JsonSubTypes.Type(value=AA.class,name="aa"),
    @JsonSubTypes.Type(value=AB.class,name="ab"),
})
public abstract class A {
    @Id
    @GeneratedValue
    private Long id;

    //json ignore for getter
    @ManyToOne
    private A parent;

    @OneToMany(mappedBy="parent")
    private List<A> children;
}

AA.java

@Entity
@Getter
@Setter
@DiscriminatorValue("aa")
public class AA extends A{
    private User user;
}

AB.java

@Entity
@Getter
@Setter
@DiscriminatorValue("ab")
public class AB extends A {
    private String name;
}

现在,当我将类 AB 的实例作为JSON返回时,它看起来像这样:

{
     "id": 1,
     "type": "ab",
     "children": [...],
     "name": "ali"
}

由于我想使用自定义User JSON,我想使用ModelMapper。除了 type 之外,所有事情都没问题。

我的DTO:

@Getter
@Setter
public class ADto {
     private Long id;

     private List<ADto> children;
}

@Getter
@Setter
public class AADto extends ADto {
     private UserDto user;
}

@Getter
@Setter
public class ABDto extends ADto {
     private String name;
}

如何使用ModelMapper并获取课程type

1 个答案:

答案 0 :(得分:0)

可能的解决方案:

  1. 您是否尝试在父类中添加属性类型?如果从数据库中检索此字段,这应该足够了。
  2. 一个简单的测试技巧,添加protected String type = this.getClass().getSimpleName().toLowerCase();。请注意,它必须受保护。这是我测试过的代码:

    @Data
    abstract class A {
      protected String type = this.getClass().getSimpleName().toLowerCase();
    }
    
     class AB extends A {}
    
     class AA extends A {}
    
     @Test
     public void test(){
        A a1 = new AB();
        A a2 = new AA();
        assertThat(a1.getType()).isEqualTo("ab");
        assertThat(a2.getType()).isEqualTo("aa");
     }