0

我有 2 节课A，B其中都有一家公司。公司A获得的信息比B（Company有一段id时间没有CNCompany）的公司多一点。我想使用 orika 映射所有字段。如果一个对象的所有字段都为空，我希望该对象为空！

我试图用单元测试来表达这一点。必须做些什么才能让它运行？

public class A {

    private Company company;

    public Company getCompany() {
        return company;
    }

    public void setCompany(Company company) {
        this.company = company;
    }

}

public class B {

    private CNCompany company;

    public CNCompany getCompany() {
        return company;
    }

    public void setCompany(CNCompany company) {
        this.company = company;
    }

}

public class Company {

    private Id id;
    private AccountId accountId;

    public Id getId() {
        return id;
    }

    public void setId(Id id) {
        this.id = id;
    }

    public AccountId getAccountId() {
        return accountId;
    }

    public void setAccountId(AccountId accountId) {
        this.accountId = accountId;
    }

}

public class CNCompany {

    private AccountId accountId

    public AccountId getAccountId() {
        return accountId;
    }

    public void setAccountId(AccountId accountId) {
        this.accountId = accountId;
    }

}

public class MyMapper extends ConfigurableMapper {
    @Override
    protected void configure(MapperFactory factory) {
        factory.classMap(A.class, B.class) //
                .mapNulls(false).mapNullsInReverse(false) //
                .byDefault() //
                .register();
    }
}

@Test
public void testMap() throws Exception {
    A a = new A();

    Company company = new Company();
    Id id = new Id();
    id.setValue("1");
    company.setId(id);
    a.setCompany(company);

    MyMapper myMapper = new MyMapper();
    B outcome = myMapper.map(a, B.class);

    assertThat(outcome.getCompany(), is(nullValue()));
}
4

1 回答 1

0

如果我理解正确，如果对象仅包含空值，您想getCompany返回。nullCompany

在 Orika 中，您可以使用自定义转换器控制转换。对于您的示例，可能类似于：

public class CompanyConverter extends CustomConverter<Company, CNCompany> {
    public CNCompany convert(Company source, Type<? extends CNCompany> destinationType) {
        if (isNothingButNulls(source)) {
            return null;
        }
        final CNCompany newCompany = new CNCompany();
        // do your thing
        return newCompany;
    }
}

我从来没有写过一个CustomConverter可以返回的null东西，所以我不能 100% 确定这会起作用，但它应该会起作用。请注意，转换器仍需要注册。我链接的文档显示了如何根据您想要转换器的级别进行注册。

于 2016-05-31T20:48:17.330 回答