使用MapStruct映射器根据应用程序的环境以编程方式为DTO设置自定义值

Question

我有一个MarkdownService，它执行以下操作：

>

打开事务处理

 Markdown markdown = markdownRepository
      .findById(markdownId)
      .orElseThrow(EntityNotFoundException::new);

String originalFilename = storageService.getOriginalFilename(file);

String filename = Markdown.normalizeMdPath(markdownId + "-" + originalFilename);

 storageService.store(
      file,
      this.uploadMarkdownsPath,
      filename
    );

    markdown.setMdPath(filename);

    MarkdownDto dto = MarkdownMapper.INSTANCE.toMarkdownDto(markdown);

 String fullPath = ServletUriComponentsBuilder
      .fromCurrentContextPath()
      .path("/markdown/download/")
      .path(dto.getMdPath())
      .toUriString();

    dto.setMdPath(fullPath);

    return dto;

@Mapper
public interface MarkdownMapper {
  MarkdownMapper INSTANCE = Mappers.getMapper(MarkdownMapper.class);

  MarkdownDto toMarkdownDto(Markdown markdown);
}

  @Transactional
  public MarkdownDto uploadMarkdownFile(Long markdownId, MultipartFile file) throws FileNotFoundException {
    Markdown markdown = markdownRepository
      .findById(markdownId)
      .orElseThrow(EntityNotFoundException::new);

    String originalFilename = storageService.getOriginalFilename(file);

    String filename = Markdown.normalizeMdPath(markdownId + "-" + originalFilename);

    storageService.store(
      file,
      this.uploadMarkdownsPath,
      filename
    );

    markdown.setMdPath(filename);

    MarkdownDto dto = MarkdownMapper.INSTANCE.toMarkdownDto(markdown);

    String fullPath = ServletUriComponentsBuilder
      .fromCurrentContextPath()
      .path("/markdown/download/")
      .path(dto.getMdPath())
      .toUriString();

    dto.setMdPath(fullPath);

    return dto;
  }

谢谢！

Answer 1

您可以使用将上下文注入映射器调用并定义@AfterMapping。

@Mapper(componentModel = "spring", unmappedTargetPolicy = ReportingPolicy.ERROR)
public interface UserPOMapper {
    UserDTO toUserDTO(UserPO userPo);
    
    @Mapping(target = "id", ignore = true)
    UserPO fromUserDTO(UserDTO userDto);
    
    @AfterMapping
    default void after(@MappingTarget UserDTO.UserDTOBuilder u, @Context String url) {
        u.firstName(url);
    }
    
    UserDTO toUserDTOWithDetails(UserPO userPO, @Context String url);
}