使用Akka HTTP上传文件

Question

我正在尝试使用Akka HTTP在我的应用程序中实现文件上载功能.我正在使用akka-stream版本2.4.4.

这是代码(从akka-doc修改)

path("fileupload") {
    post {
      extractRequestContext {
        ctx => {
          implicit val materializer = ctx.materializer
          implicit val ec = ctx.executionContext
          fileUpload("fileUpload") {
            case (metadata, byteSource) =>
              val location = FileUtil.getUploadPath(metadata)
              val updatedFileName = metadata.fileName.replaceAll(" ", "").replaceAll("\"", "")
              val uniqFileName = uniqueFileId.concat(updatedFileName)
              val fullPath = location + File.separator + uniqFileName
              val writer = new FileOutputStream(fullPath)
              val bufferedWriter = new BufferedOutputStream(writer)

              val result = byteSource.map(s => {
                bufferedWriter.write(s.toArray)
              }).runWith(Sink.ignore)

              val result1 = byteSource.runWith(Sink.foreach(s=>bufferedWriter.write(s.toArray)))
              Await.result(result1, 5.seconds)
              bufferedWriter.flush()
              bufferedWriter.close()
              complete(uniqFileName)
            /*onSuccess(result) { x =>
              bufferedWriter.flush()
              bufferedWriter.close()
              complete("hello world")
            }*/
          }
        }
      }
    }
  }

此代码工作正常,并将文件上载到给定路径.我通过附加UUID来生成新的文件名,以确保文件名是唯一的.所以我需要将新文件名返回给调用者.但是,此方法始终不返回文件名.有时,它正在完成Response has no content.

任何人都可以让我知道我在做错了什么吗？

Answer 1

当您为此目的而拥有反应流时,无需使用标准阻塞流:

  path("fileUpload") {
    post {
      fileUpload("fileUpload") {
        case (fileInfo, fileStream) =>
          val sink = FileIO.toPath(Paths.get("/tmp") resolve fileInfo.fileName)
          val writeResult = fileStream.runWith(sink)
          onSuccess(writeResult) { result =>
            result.status match {
              case Success(_) => complete(s"Successfully written ${result.count} bytes")
              case Failure(e) => throw e
            }
          }
      }
    }
  }

此代码将fileUploadmultipart字段上载到/tmp目录中的文件.它只是将输入源的内容转储到相应的文件接收器,在写入操作完成时返回一条消息.

您可能还想调整用于FileIO源和接收器的调度程序,如scaladocs中所述.