使用akka-http websockets上传和处理文件

Question

我正在使用一些示例Scala代码制作一个服务器，该服务器通过websocket接收文件，临时存储文件，在其上运行bash脚本，然后通过TextMessage返回stdout。

示例代码来自于这个github项目 。

我在echoService中稍微编辑了代码，以便它运行另一个处理临时文件的功能。

object WebServer {
  def main(args: Array[String]) {

    implicit val actorSystem = ActorSystem("akka-system")
    implicit val flowMaterializer = ActorMaterializer()

    val interface = "localhost"
    val port = 3000

    import Directives._

    val route = get {
      pathEndOrSingleSlash {
        complete("Welcome to websocket server")
      }
    } ~
      path("upload") {
        handleWebSocketMessages(echoService)
      }

      val binding = Http().bindAndHandle(route, interface, port)
      println(s"Server is now online at http://$interface:$port\nPress RETURN to stop...")
      StdIn.readLine()

      binding.flatMap(_.unbind()).onComplete(_ => actorSystem.shutdown())
      println("Server is down...")

    }

    implicit val actorSystem = ActorSystem("akka-system")
    implicit val flowMaterializer = ActorMaterializer()


    val echoService: Flow[Message, Message, _] = Flow[Message].mapConcat {

      case BinaryMessage.Strict(msg) => {
        val decoded: Array[Byte] = msg.toArray
        val imgOutFile = new File("/tmp/" + "filename")
        val fileOuputStream = new FileOutputStream(imgOutFile)
        fileOuputStream.write(decoded)
        fileOuputStream.close()
        TextMessage(analyze(imgOutFile))
      }

      case BinaryMessage.Streamed(stream) => {

        stream
          .limit(Int.MaxValue) // Max frames we are willing to wait for
          .completionTimeout(50 seconds) // Max time until last frame
          .runFold(ByteString(""))(_ ++ _) // Merges the frames
          .flatMap { (msg: ByteString) =>

          val decoded: Array[Byte] = msg.toArray
          val imgOutFile = new File("/tmp/" + "filename")
          val fileOuputStream = new FileOutputStream(imgOutFile)
          fileOuputStream.write(decoded)
          fileOuputStream.close()
          Future(Source.single(""))
        }
        TextMessage(analyze(imgOutFile))
      }


      private def analyze(imgfile: File): String = {
        val p = Runtime.getRuntime.exec(Array("./run-vision.sh", imgfile.toString))
        val br = new BufferedReader(new InputStreamReader(p.getInputStream, StandardCharsets.UTF_8))
        try {
          val result = Stream
            .continually(br.readLine())
            .takeWhile(_ ne null)
            .mkString
          result

        } finally {
          br.close()
        }
      }
    }




}

在使用Dark WebSocket Terminal进行测试期间， case BinaryMessage.Strict可以正常工作。

问题：但是， case BinaryMessage.Streaming在运行analyze功能之前没有完成写入文件的操作， case BinaryMessage.Streaming导致服务器的响应为空。

我试图围绕Akka-HTTP中的Flows在这里如何使用Futures进行思考，但是在尝试阅读所有正式文档之外，我没有太多运气。

目前，.mapAsync似乎很有希望，或者基本上是找到一种链接期货的方法。

我真的很感谢一些见识。

Answer 1

是的，在这种情况下， mapAsync将为您提供帮助。 它是一个组合器，用于在流中执行Future （可能并行），并将其结果显示在输出端。

为了使事情变得同质并使类型检查器满意，您需要将Strict case的结果包装到Future.successful 。

您的代码的快速解决方案可以是：

  val echoService: Flow[Message, Message, _] = Flow[Message].mapAsync(parallelism = 5) {

    case BinaryMessage.Strict(msg) => {
      val decoded: Array[Byte] = msg.toArray
      val imgOutFile = new File("/tmp/" + "filename")
      val fileOuputStream = new FileOutputStream(imgOutFile)
      fileOuputStream.write(decoded)
      fileOuputStream.close()
      Future.successful(TextMessage(analyze(imgOutFile)))
    }

    case BinaryMessage.Streamed(stream) =>

      stream
        .limit(Int.MaxValue) // Max frames we are willing to wait for
        .completionTimeout(50 seconds) // Max time until last frame
        .runFold(ByteString(""))(_ ++ _) // Merges the frames
        .flatMap { (msg: ByteString) =>

        val decoded: Array[Byte] = msg.toArray
        val imgOutFile = new File("/tmp/" + "filename")
        val fileOuputStream = new FileOutputStream(imgOutFile)
        fileOuputStream.write(decoded)
        fileOuputStream.close()
        Future.successful(TextMessage(analyze(imgOutFile)))
      }
  }