pekko/akka-docs-dev/rst/scala/code/docs/stream/io/StreamFileDocSpec.scala

/*
 * Copyright (C) 2015 Typesafe Inc. <http://www.typesafe.com>
 */
package docs.stream.io

import java.io.File

import akka.stream._
import akka.stream.io.SynchronousFileSource
import akka.stream.io.SynchronousFileSink
import akka.stream.testkit._
import akka.stream.testkit.Utils._
import akka.util.ByteString

class StreamFileDocSpec extends AkkaSpec(UnboundedMailboxConfig) {

  implicit val ec = system.dispatcher
  implicit val mat = ActorFlowMaterializer()

  // silence sysout
  def println(s: String) = ()

  val file = File.createTempFile(getClass.getName, ".tmp")

  override def afterTermination() = file.delete()

  {
    //#file-source
    import akka.stream.io._
    //#file-source
  }

  {
    //#file-source
    val file = new File("example.csv")
    //#file-source
  }

  "read data from a file" in {
    //#file-source
    def handle(b: ByteString): Unit //#file-source
    = ()

    //#file-source

    SynchronousFileSource(file)
      .runForeach((chunk: ByteString) ⇒ handle(chunk))
    //#file-source
  }

  "configure dispatcher in code" in {
    //#custom-dispatcher-code
    SynchronousFileSink(file)
      .withAttributes(ActorOperationAttributes.dispatcher("custom-file-io-dispatcher"))
    //#custom-dispatcher-code
  }
}