Отображение объектов с двусторонними связями с Mapstruct

Question

Интересно, а может ли Mapstruct помочь с сопоставлением объектов с двунаправленными отношениями (в моем случае один ко многим):

public class A{
     private Set<B> listB;
}

public class B{
     private A a;
}

Отображение из / в сущность дает StackOverflowError. (я бы ожидал, что это произойдет). С другой стороны, закрытые проблемы Mapstruct 469 и 1163 , похоже, подразумевают, что mapstruct не будет напрямую поддерживать его. Я попробовал этот пример:

https://github.com/mapstruct/mapstruct-examples/tree/master/mapstruct-mapping-with-cycles

Но это просто не работает. С применением или без применения CycleAvoidingMappingContext стек переполнения остается неизменным.

Так как же сопоставить объекты с циклами и использовать mapstruct? (Я хочу, чтобы избежать полного ручного отображения)

Answer 1

Тем временем я нашел решение: сначала я игнорирую рассматриваемое поле и отображаю его в «AfterMapping». В моем случае я хочу нанести на карту и ребенка самостоятельно. В моем случае @Context содержит только Boolean (boolean не работает), который сообщает, где я вошел в цикл (startedFromPru pru - родитель):

Mapper для родителя (содержит набор лицензий):

@Mapper(componentModel = "spring", uses = {DateTimeMapper.class, LicenseMapper.class, CompanyCodeMapper.class, ProductHierarchyMapper.class})
public abstract class ProductResponsibleUnitMapper {

    @Autowired
    private LicenseMapper licenseMapper;

    @Mappings({@Mapping(target = "licenses", ignore = true)})
    public abstract ProductResponsibleUnit fromEntity(ProductResponsibleUnitEntity entity, @Context Boolean startedFromPru);

    @Mappings({@Mapping(target = "licenses", ignore = true)})
    public abstract ProductResponsibleUnitEntity toEntity(ProductResponsibleUnit domain, @Context Boolean startedFromPru);

    @AfterMapping
    protected void mapLicenseEntities(ProductResponsibleUnit pru, @MappingTarget ProductResponsibleUnitEntity pruEntity, @Context Boolean startedFromPru){
        if(startedFromPru) {
            pruEntity.getLicenses().clear(); //probably not necessary
            pru.getLicenses().stream().map(license -> licenseMapper.toEntity(license, startedFromPru)).forEach(pruEntity::addLicense);
        }
    }

    @AfterMapping
    protected void mapLicenseEntities(ProductResponsibleUnitEntity pruEntity, @MappingTarget ProductResponsibleUnit pru, @Context Boolean startedFromPru){
        if(startedFromPru) {
            pru.getLicenses().clear(); //probably not necessary
            pruEntity.getLicenses().stream().map(licenseEntity -> licenseMapper.fromEntity(licenseEntity, startedFromPru)).forEach(pru::addLicense);
        }
    }

}

Mapper для детей:

@Mapper(componentModel = "spring", uses = { DateTimeMapper.class, CompanyCodeMapper.class, ProductHierarchyMapper.class,
        CountryCodeMapper.class })
public abstract class LicenseMapper {

    @Autowired
    private ProductResponsibleUnitMapper pruMapper;

    @Mappings({ @Mapping(source = "licenseeId", target = "licensee"), @Mapping(target = "licensor", ignore = true) })
    public abstract License fromEntity(LicenseEntity entity, @Context Boolean startedFromPru);


    @Mappings({ @Mapping(source = "licensee", target = "licenseeId"), @Mapping(target = "licensor", ignore = true) })
    public abstract LicenseEntity toEntity(License domain, @Context Boolean startedFromPru);

    @AfterMapping
    protected void mapPru(LicenseEntity licenseEntity, @MappingTarget License license,
            @Context Boolean startedFromPru) {
        //only call ProductResponsibleUnitMapper if mapping did not start from PRU -> startedFromPru is false
        if (!startedFromPru) {
            license.setLicensor(pruMapper.fromEntity(licenseEntity.getLicensor(), startedFromPru));
        }
    }

    @AfterMapping
    protected void mapPru(License license, @MappingTarget LicenseEntity licenseEntity,
            @Context Boolean startedFromPru) {
        //only call ProductResponsibleUnitMapper if mapping did not start from PRU -> startedFromPru is false
        if (!startedFromPru) {
            licenseEntity.setLicensor(pruMapper.toEntity(license.getLicensor(), startedFromPru));
        }
    }

Answer 2

Чтобы сопоставление заработало, вы можете попробовать использовать следующее сопоставление:

@Mapper
public interface MyMapper {

    A map(A a, @Context CycleAvoidingMappingContext context);

    Set<B> map(Set<B> b, @Context CycleAvoidingMappingContext context);

    B map(B b, @Context CycleAvoidingMappingContext context);
}

Если методов для отображения из Set<B> в Set<B> нет, то набор в A не будет отображен.