如何将文件上传集成到Spring Data REST存储库中？

Question

我想将文件上传和下载集成到Spring Data REST存储库中。

我有一个＜code＞DataFile＜/code＞类（见下面的实现），它总是伴随着一些元数据。有一个DataFileRepository来存储有关元数据和文件之间关系的信息。

用户应该能够以web应用程序的形式上载文件和元数据。元数据应存储在html" target="_blank">数据库表中，同时文件被上传到某个文件存储，然后文件路径再次存储在实体中。

我想使用Spring Data REST执行此操作。我的方法是在DataFile实体中包含一个MultipartFile并将其标记为@瞬态和@RestResource（导出=false），这样它就不在实体的任何表示中，但可以上传。上传时，AbstractRepositoryEventListener会覆盖onBeforeCreate并在此步骤中上传文件。

但是使用这种方法，我很难将文件包含在HTTP请求体中(一个< code>DataFile的JSON表示)。

其他方法包括实现一个< code > @ rest repository controller 来扩充< code>DataFileRepository，但前面提到的方法是我最喜欢的。

使用SpringDataREST（存储库）进行文件上传的最佳实践是什么？

@Data
@Entity
@Table(name = "data_files")
public class DataFile {
    @Id
    @GeneratedValue
    @Setter(AccessLevel.NONE)
    private long identifier;

    @NotBlank
    @Column(name = "path")
    @RestResource(exported = false)
    private String path;

    @Embedded
    private Metadata metadata;

    public DataFile() {
        metadata = new Metadata();
    }

    public DataFile(Metadata metadata) {
        this.metadata = metadata;
    }
}

Answer 1

我没有找到只使用SpringDataREST的方法，但这个解决方案很好地集成到存储库中，不需要太多手写代码。

下面是针对与问题不同的项目的此类解决方案的示例，但您应该明白这一点。

基本上创建一个@BasePathAwareController，使其适应您可能为Spring Data REST设置的基本路径。然后在控制器内创建具有适当路径的映射。在本例中，有一个@PostMapping，其中包含指向导师存储库的路径和在此上载的相关附件的子路径。

您可以在这个方法中进行验证，或者在存储库的save方法中引入一个方面。

在这种情况下，我使用了一个方面将处理应用于存储库的 save（） 方法的每次调用。

@BasePathAwareController
@ExposesResourceFor(Tutor.class)
public class TutorRepositoryController {
    private AttachmentAssembler attachmentAssembler;

    private AttachmentRepository attachmentRepository;

    private TutorRepository tutorRepository;

    private RepositoryEntityLinks entityLinks;

    @Autowired
    public TutorRepositoryController(AttachmentAssembler attachmentAssembler,
                                     AttachmentRepository attachmentRepository,
                                     TutorRepository tutorRepository,
                                     RepositoryEntityLinks entityLinks) {
        this.attachmentAssembler  = attachmentAssembler;
        this.attachmentRepository = attachmentRepository;
        this.tutorRepository      = tutorRepository;
        this.entityLinks          = entityLinks;
    }

    @PostMapping(value = "/tutors/{id}/attachments/{name}")
    public ResponseEntity<EntityModel<Attachment>> saveAttachment(@RequestParam("data") MultipartFile file,
                                                                  @PathVariable long id,
                                                                  @PathVariable String name) {
        Tutor thisTutor = tutorRepository.findById(id);
        File tempFile;
        try {
            tempFile = File.createTempFile(FilenameUtils.getBaseName(file.getOriginalFilename()),
                                           FilenameUtils.getExtension(file.getOriginalFilename()));

            StreamUtils.copy(file.getResource().getInputStream(), new FileOutputStream(tempFile));
        } catch (IOException e) {
            throw new RuntimeException("Could not create temporary file of uploaded file.");
        }

        Attachment saved =
                    attachmentRepository.save(new Attachment(name, thisTutor, tempFile, file.getOriginalFilename()));

        return ResponseEntity.created(entityLinks.linkForItemResource(Attachment.class, saved.getIdentifier())
                                                  .toUri()).body(attachmentAssembler.toModel(saved));
    }
}

这是处理文件上载的方面的代码。

@Aspect
@Component
public class TutorRepositoryAspect {
    private Logger log = LoggerFactory.getLogger(TutorRepositoryAspect.class.getName());

    private AttachmentRepository attachmentRepository;

    private ApplicationConfiguration configuration;

    private S3Service s3Service;

    @Autowired
    public TutorRepositoryAspect(AttachmentRepository attachmentRepository,
                                 ApplicationConfiguration configuration,
                                 S3Service s3Service) {
        this.attachmentRepository = attachmentRepository;
        this.configuration        = configuration;
        this.s3Service            = s3Service;
    }

    @Pointcut(value = "execution(* com.example.tutor.backend.repository.attachment.AttachmentRepository.save(..)) && args(attachment)")
    private void repositorySave(Attachment attachment) {
    }

    @Before(value = "repositorySave(attachment)", argNames = "attachment")
    private Attachment beforeSave(Attachment attachment) {
        log.info("Received request to add attachment to tutor \"{}\".", attachment.getIdentifier());

        File file = attachment.getFile();

        // Validation

        String filePath = attachment.getAttachedTo().getIdentifier() + "-" + attachment.getName() + "-"
                          + attachment.getOriginalFileName();

        if (s3Service.isFilePresent(filePath)) {
            log.error("Could not upload file as there is a file already with the same name.");
            throw new IllegalArgumentException("File is already present.");
        }

        String s3Path = s3Service.uploadFile(file, filePath);

        attachment.setFilePath(s3Path);

        // This attachment is now being saved to database.
        return attachment;
    }
}