使用 ModelMapper 防止循环引用 - 列表

Question

我遇到了 Json 错误，因为我的 DTO 彼此保持循环引用。

公司DTO：

@Data
public class CompanyDTO {

    private Long id;
    private String name;
    private List<CompanyCountryDTO> companyCountries;
    //getters and setters
}

CompanyCountryDTO：

@Data
public class CompanyCountryDTO {

    private Long id;
    private LocalDateTime createdAt;
    private CompanyDTO company;
    private CountryDTO country;
    //getters and setters

CompanyService 实现将 Company 列表转换为 CompanyDTO 列表：

@Override
public List<CompanyDTO> getAllCompaniesWithCountryDTO() {
   List<Company> listCompanies = companyRep.findAll();
   return listCompanies.stream().map(this::convertToDto).collect(Collectors.toList());
}

private CompanyDTO convertToDto(Company company) {
    CompanyDTO companyWithServiceDTO = modelMapper.map(company, CompanyDTO.class);
    return companyWithServiceDTO;
}

我想做这样的事情，但使用 ModelMapper，因为我还有其他循环引用：

listCompanies.stream().forEach(company -> {
    if (!ObjectUtils.isEmpty(company.getCompanyCountries())) {
        company.getCompanyCountries().forEach(companyCountry -> {
           companyCountry.setCompany(null);             
        });
    }
});

我如何从 CompanyCountryDTO 中删除公司引用，因为公司还有另一个 CompanyCountryDTO 列表？

Answer 1

http://modelmapper.org/user-manual/configuration/ 说

首选嵌套属性 |确定隐式映射是否应映射嵌套属性，我们强烈建议您在映射包含循环引用的模型时禁用此选项 |真的

所以另一个解决方法可能是

getConfiguration().setPreferNestedProperties(false);

Answer 2

我认为您只需要跳过“反向引用”即可避免循环和无限递归。在这种情况下，CompanyCountryDTO 中的字段 company。

要让它工作，请配置您的ModelMapper，例如：

modelMapper.addMappings(new PropertyMap<CompanyCountry, CompanyCountryDTO>() {
    @Override
    protected void configure() {
        // Tells ModelMapper to NOT populate company
        skip(destination.getCompany());
    }
});