Akka-HTTP:文件上传

Akka-HTTP: File Upload

我正在尝试使用 akka http 实现一个简单的文件上传。 我的尝试如下所示:

    import akka.actor.ActorSystem
    import akka.event.{LoggingAdapter, Logging}
    import akka.http.scaladsl.Http
    import akka.http.scaladsl.model.{HttpResponse, HttpRequest}
    import akka.http.scaladsl.model.StatusCodes._
    import akka.http.scaladsl.server.Directives._
    import akka.stream.{ActorMaterializer, Materializer}
    import com.typesafe.config.Config
    import com.typesafe.config.ConfigFactory
    import scala.concurrent.{ExecutionContextExecutor, Future}
    import akka.http.scaladsl.model.StatusCodes
    import akka.http.scaladsl.model.HttpEntity
    import java.io._
    import akka.stream.io._

    object UploadTest extends App {
      implicit val system = ActorSystem()
      implicit val executor = system.dispatcher
      implicit val materializer = ActorMaterializer()

      val config = ConfigFactory.load()
      val logger = Logging(system, getClass)

      val routes = {
        pathSingleSlash {
          (post & extractRequest) { 
            request => {
              val source = request.entity.dataBytes
              val outFile = new File("/tmp/outfile.dat")
              val sink = SynchronousFileSink.create(outFile)
              source.to(sink).run()
              complete(HttpResponse(status = StatusCodes.OK))
            }
          }
        }
      }

      Http().bindAndHandle(routes, config.getString("http.interface"), config.getInt("http.port"))

    }

这段代码有几个问题:

  1. 无法上传大于配置的实体大小的文件: Request Content-Length 24090745 exceeds the configured limit of 8388608
  2. 连续执行两次上传导致 dead letters encountered. 异常。

克服大小限制的最佳方法是什么?如何正确关闭文件以便后续上传将覆盖现有文件(暂时忽略并发上传)?

对于第2点,我认为source.to(sink).run()是异步执行的。它具体化了一个Future。因此,您的 HTTP 请求可能 return 在文件写入完成之前,因此如果您在第一个请求 return 后立即在客户端开始第二次上传,则第一个可能尚未完成写入文件。

您可以使用 onCompleteonSuccess 指令仅在未来完成时完成 http 请求:

http://doc.akka.io/docs/akka-stream-and-http-experimental/1.0-M2/scala/http/directives/alphabetically.html

http://doc.akka.io/docs/akka-stream-and-http-experimental/1.0/scala/http/routing-dsl/directives/future-directives/onSuccess.html

编辑:

对于内容长度问题,您可以做的一件事是增加 application.conf 中 属性 的大小。默认值为:

akka.server.parsing.max-content-length = 8m

http://doc.akka.io/docs/akka-stream-and-http-experimental/1.0/java/http/configuration.html

总结mattinbits的评论,下面的解决方案有效:

  1. 增加akka.server.parsing.max-content-length
  2. 使用onSuccess

这是代码片段:

val routes = {
  pathSingleSlash {
    (post & extractRequest) { 
      request => {
        val source = request.entity.dataBytes
        val outFile = new File("/tmp/outfile.dat")
        val sink = SynchronousFileSink.create(outFile)
        val repl = source.runWith(sink).map(x => s"Finished uploading ${x} bytes!")
        onSuccess(repl) { repl =>
          complete(HttpResponse(status = StatusCodes.OK, entity = repl))
        }
      }
    }
  }