如何通过MapStruct根据MediaDto的type字段实例化Image/Video类？

我来给你梳理下具体的配置步骤，MapStruct完全支持这种基于字段的多态实例化，咱们一步步来：

1. 对齐类结构（先确认基础定义） #

先把你提到的类结构明确下来，确保咱们的基础一致：

• 抽象父类 Media，子类 Image、Video（继承自Media）
• DTO类 MediaDto，包含type字段（值为"image"或"video"），以及和Media子类对应的属性
• Library 实体包含Media类型的media字段，LibraryDto包含MediaDto类型的media字段

2. 配置MapStruct映射接口 #

核心思路是自定义转换逻辑，根据MediaDto.type的值决定实例化哪个Media子类，同时让MapStruct帮我们处理属性映射的脏活。

下面是完整的Mapper示例：

@Mapper(componentModel = "spring") // 如果用Spring框架选这个；非Spring环境用default
public interface LibraryMapper {

    // 非Spring环境可通过这个静态实例直接调用
    LibraryMapper INSTANCE = Mappers.getMapper(LibraryMapper.class);

    // LibraryDto -> Library 的映射，自动关联media字段的转换
    @Mapping(target = "media", source = "media")
    Library toLibraryEntity(LibraryDto libraryDto);

    // 核心：根据MediaDto的type判断返回Image还是Video
    default Media toMedia(MediaDto mediaDto) {
        if (mediaDto == null) {
            return null;
        }

        return switch (mediaDto.getType()) {
            case "image" -> toImage(mediaDto);
            case "video" -> toVideo(mediaDto);
            default -> throw new IllegalArgumentException("不支持的媒体类型: " + mediaDto.getType());
        };
    }

    // MapStruct会自动生成这个方法，把MediaDto的属性映射到Image实例
    Image toImage(MediaDto mediaDto);

    // 同理，映射到Video实例
    Video toVideo(MediaDto mediaDto);

    // ------------------------ 反向映射（如果需要的话） ------------------------
    @Mapping(target = "media", source = "media")
    LibraryDto toLibraryDto(Library library);

    default MediaDto toMediaDto(Media media) {
        if (media == null) {
            return null;
        }

        MediaDto mediaDto = new MediaDto();
        if (media instanceof Image image) {
            mediaDto.setType("image");
            mapImageToDto(image, mediaDto);
        } else if (media instanceof Video video) {
            mediaDto.setType("video");
            mapVideoToDto(video, mediaDto);
        } else {
            throw new IllegalArgumentException("未知的媒体实体类型: " + media.getClass().getName());
        }
        return mediaDto;
    }

    // MapStruct实现属性映射，把Image的属性赋值给MediaDto
    void mapImageToDto(Image image, @MappingTarget MediaDto mediaDto);

    // 同理处理Video到MediaDto的属性映射
    void mapVideoToDto(Video video, @MappingTarget MediaDto mediaDto);
}

3. 关键细节说明 #

• default方法的作用：我们用default方法封装了类型判断的逻辑，MapStruct会优先使用这个方法来处理MediaDto到Media的转换，然后再调用它生成的toImage/toVideo方法完成属性赋值。
• 属性自动映射：只要MediaDto的属性名和Image/Video的属性名一致，MapStruct会自动完成映射；如果名称不一致，你可以在toImage/toVideo方法上添加@Mapping注解指定对应关系，比如：

  @Mapping(target = "imageUrl", source = "url")
  Image toImage(MediaDto mediaDto);
  

• 组件模型选择：如果是Spring项目，用componentModel = "spring"，这样Mapper会被注册为Spring Bean，你可以直接注入使用；非Spring环境用default，通过INSTANCE静态变量调用。
• 异常处理：加入了类型不匹配的异常抛出，方便调试和处理非法输入。

4. 使用示例 #

在你的业务代码里，直接调用Mapper的方法即可：

// Spring环境注入使用
@Autowired
private LibraryMapper libraryMapper;

// 转换Dto到实体
Library library = libraryMapper.toLibraryEntity(libraryDto);

// 转换实体到Dto
LibraryDto dto = libraryMapper.toLibraryDto(library);

这样就能完美实现根据MediaDto.type字段来实例化对应的Image或Video实体了。

内容的提问来源于stack exchange，提问作者Vetouz