Play Framework 2.6 Alpakka S3文件上传

时间:2017-09-10 08:53:25

标签: scala amazon-s3 playframework akka-stream alpakka

我使用Play Framework 2.6(Scala)和Alpakka AWS S3 Connector将文件异步上传到S3存储桶。我的代码如下所示:

def richUpload(extension: String, checkFunction: (String, Option[String]) => Boolean, cannedAcl: CannedAcl, bucket: String) = userAction(parse.multipartFormData(handleFilePartAsFile)).async { implicit request =>
  val s3Filename = request.user.get.id + "/" + java.util.UUID.randomUUID.toString + "." + extension
  val fileOption = request.body.file("file").map {
    case FilePart(key, filename, contentType, file) =>
      Logger.info(s"key = ${key}, filename = ${filename}, contentType = ${contentType}, file = $file")
      if(checkFunction(filename, contentType)) {
        s3Service.uploadSink(s3Filename, cannedAcl, bucket).runWith(FileIO.fromPath(file.toPath))
      } else {
        throw new Exception("Upload failed")
      }
  }

  fileOption match {
    case Some(opt) => opt.map(o => Ok(s3Filename))
    case _ => Future.successful(BadRequest("ERROR"))
  }
}

它可以工作,但它在上传到S3之前返回文件名。但我希望在上传到S3后返回值。有没有解决方案?

此外,是否可以直接将文件上传到S3,以正确显示进度并且不使用临时磁盘文件?

1 个答案:

答案 0 :(得分:2)

您需要翻转源和接收器以获取您感兴趣的materialized value。 你有:

  1. 从您的本地文件读取的来源,并在完成阅读文件后具体化为Future[IOResult]
  2. 写入S3并在完成写入S3后实现Future[MultipartUploadResult]的接收器。
  3. 您对后者感兴趣,但在您的代码中,您使用的是前者。这是因为runWith函数始终将stage的实现值保持为参数。

    下面的示例代码段中的类型应该澄清:

      val fileSource: Source[ByteString, Future[IOResult]]              = ???
      val s3Sink    : Sink  [ByteString, Future[MultipartUploadResult]] = ???
    
      val m1: Future[IOResult]              = s3Sink.runWith(fileSource)
      val m2: Future[MultipartUploadResult] = fileSource.runWith(s3Sink)
    

    获得Future[MultipartUploadResult]后,您可以map以同样的方式访问location字段以获取文件的URI,例如:

      val location: URI = fileSource.runWith(s3Sink).map(_.location)