文件编写iteratee没有收到WS.get的EOF

时间:2014-03-26 08:46:38

标签: scala playframework playframework-2.2 iterate

我创建了一个简单的iteratee来使用WS下载文件,如this link中所述。

请考虑以下代码段:

import java.nio.ByteBuffer
import java.nio.channels.FileChannel
import org.specs2.mutable.Specification
import org.specs2.time.NoTimeConversions
import play.api.libs.Files.TemporaryFile
import play.api.libs.iteratee.{Done, Input, Cont, Iteratee}
import play.api.libs.ws.WS
import scala.concurrent.Await
import scala.concurrent.duration._
import scala.concurrent.ExecutionContext.Implicits.global

class DummySpec extends Specification with NoTimeConversions {
  def fileChannelIteratee(channel: FileChannel): Iteratee[Array[Byte], Unit] = Cont {
    case Input.EOF =>
      println("Input.EOF")
      channel.close()
      Done(Unit, Input.EOF)
    case Input.El(bytes) =>
      println("Input.El")
      val buf = ByteBuffer.wrap(bytes)
      channel.write(buf)
      fileChannelIteratee(channel)
    case Input.Empty =>
      println("Input.Empty")
      fileChannelIteratee(channel)
  }

  "fileChannelIteratee" should {
    "work" in {
      val file = TemporaryFile.apply("test").file
      val channel = FileChannel.open(file.toPath)

      val future = WS.url("http://www.example.com").get(_ => fileChannelIteratee(channel)).map(_.run)

      Await.result(future, 10.seconds)

      file.length !== 0
    }
  }
}

在WS.get之后调用.map(_。run)似乎没有效果,因为iteratee似乎没有接收到Input.EOF。它阻止我关闭频道。这是我得到的输出:

Input.El
[info] DummySpec
[info]
[info] fileChannelIteratee should
[info] x work (941 ms)
[error]    '0' is equal to '0' (DummySpec.scala:37)
[info]
[info]
[info] Total for specification DummySpec
[info] Finished in 948 ms
[info] 1 example, 1 failure, 0 error

我做错了什么?

我正在使用Play Framework 2.2.2。

提前致谢。

1 个答案:

答案 0 :(得分:0)

我正以错误的方式打开FileChannel。当没有给出参数时,似乎根据this link默认为读取模式。

channel.write操作吞并了map引发的异常,因为整个操作的返回类型为Future[Future[Unit]]。即使内部Future在这种情况下失败,外部flatMap也处于成功状态。应该使用{{1}}代替。