使用 mapStruct 映射抽象类

Question

我有一个项目，其中有一个 Library 实体，其中包含 media 类型的 Media 字段。 Media 是一个抽象类，由类Image 和类Video 实现。

我还有一个 dto LibraryDto，其字段 media 类型为 MediaDto。 MediaDto 包含一个字段，其值可以是 "image" 或 "video"。我的问题是：如何告诉 mapstruct 根据MediaDto 的type 字段的值来实例化Image 类或Video 类？

感谢您的回复。

Answer 1

对不起，我第一次误解了你的问题。

您可以创建对象工厂方法，该方法基于 type 字段实例化 Media，如下所示：

@Mapper
public interface LibraryMapper {
    Library libraryDtoToLibrary(LibraryDto libraryDto);

    Media mediaDtoToMedia(MediaDto mediaDto);

    @ObjectFactory
    default Media createMedia(MediaDto mediaDto) {
        if ("image".equals(mediaDto.getType())) {
            return new Image();
        } else if ("video".equals(mediaDto.getType())) {
            return new Video();
        }
        throw new IllegalArgumentException("Unknown media type.");
    }
}