如何在Mapstruct集合映射器中使用修饰方法？

Question

我正在使用MapStruct从一个JPA实体映射到一个POJO DTO，在一个带有依赖项注入的Spring应用程序中。

@Override
public Set<DimensionItemTreeDTO> missionSetToTreeDtoSet(Set<Mission> set)  {
    return delegate.missionSetToTreeDtoSet( set );
}

@Override
public Set<DimensionItemTreeDTO> missionSetToTreeDtoSet(Set<Mission> set) {
    if ( set == null ) {
        return null;
    }

    Set<DimensionItemTreeDTO> set__ = new HashSet<DimensionItemTreeDTO>();
    for ( Mission mission : set ) {
        set__.add( missionToTreeDto( mission ) ); //here the decorator is not called !
    }

    return set__;
}

...并且从不为集合中的项调用修饰方法。

有没有一种方法可以让Mapstruct在集合映射中使用decorator方法，而不是在我的decorator中手动编写集合方法（这种方法很有效，但很冗长，并且违背了Mapstruct最初的目的，即不必编写这种代码）？

Answer 1

我找到了问题的解决方案：实际上，我的用例更适合MapStruct@AfterMapping方法，我使用了它，现在它对所有情况都很好地工作：

@Mapper
public abstract class ConstraintsPostProcessor {

    @Inject
    private UserService userService; // can use normal Spring DI here

    @AfterMapping
    public void setConstraintsOnMissionTreeDTO(Mission mission, @MappingTarget MissionDTO dto){ // do not forget the @MappingTarget annotation or it will not work
        dto.setUser(userService.getCurrentUser()); // can do any additional logic here, using services etc.
    }
}

在主映射器中：

@Mapper(uses = {ConstraintsPostProcessor.class}) // just add the previous class here in the uses attribute
public interface DimensionMapper {
    ...
}