58 lines
1.3 KiB
Scala
58 lines
1.3 KiB
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.AkkaSpec
|
||
|
|
import akka.stream.testkit.StreamTestKit
|
||
|
|
import akka.util.ByteString
|
||
|
|
|
||
|
|
class StreamFileDocSpec extends AkkaSpec(StreamTestKit.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
|
||
|
|
}
|
||
|
|
}
|