• Example Search
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• zio-nio-master
  • .circleci
    • config.yml
  • nio-core
    • src
      • main
        • scala
          • zio
            • nio
              • core
                • InterfaceAddress.scala
                • InetSocketAddress.scala
                • Buffer.scala
                • package.scala
                • FloatBuffer.scala
                • CharBuffer.scala
                • channels
                  • GatheringByteChannel.scala
                  • Pipe.scala
                  • Channel.scala
                  • Selector.scala
                  • SelectionKey.scala
                  • SelectableChannel.scala
                  • AsynchronousFileChannel.scala
                  • FileLock.scala
                  • AsynchronousChannel.scala
                  • ScatteringByteChannel.scala
                  • FileChannel.scala
                  • DatagramChannel.scala
                  • AsynchronousChannelGroup.scala
                  • spi
                    • SelectorProvider.scala
                • ByteBuffer.scala
                • file
                  • FileSystem.scala
                  • WatchService.scala
                  • Path.scala
                • SocketOption.scala
                • InetAddress.scala
                • ShortBuffer.scala
                • IntBuffer.scala
                • DoubleBuffer.scala
                • MappedByteBuffer.scala
                • NetworkInterface.scala
                • LongBuffer.scala
                • charset
                  • CoderResult.scala
                  • AutoDetect.scala
                  • CharsetDecoder.scala
                  • Charset.scala
                  • CharsetEncoder.scala
                  • StreamCodeState.scala
      • test
        • resources
          • scattering_read_test.txt
          • async_file_read_test.txt
        • scala
          • zio
            • nio
              • core
                • BufferSpec.scala
                • channels
                  • DatagramChannelSpec.scala
                  • SelectorSpec.scala
                  • ChannelSpec.scala
                  • ScatterGatherChannelSpec.scala
                  • AsynchronousChannelGroupSpec.scala
                  • FileChannelSpec.scala
                • BaseSpec.scala
                • file
                  • PathSpec.scala
                  • WathServiceSpec.scala
                • charset
                  • CharsetSpec.scala
  • examples
    • src
      • main
        • scala
          • StreamsBasedServer.scala
  • .nvmrc
  • .scala-steward.conf
  • LICENSE
  • project
    • build.properties
    • BuildHelper.scala
    • plugins.sbt
  • .gitattributes
  • sbt
  • website
    • pages
      • en
        • index.js
    • siteConfig.js
    • sidebars.json
    • package.json
    • core
      • Footer.js
    • static
      • img
        • undraw_code_review.svg
        • undraw_open_source.svg
        • undraw_online.svg
        • undraw_tweetstorm.svg
        • undraw_operating_system.svg
      • css
        • custom.css
  • CLA.md
  • README.md
  • .scalafmt.conf
  • build.sbt
  • .gitignore
  • docs
    • essentials
      • index.md
      • files.md
      • sockets.md
    • usecases
      • index.md
  • nio
    • src
      • main
        • scala
          • zio
            • nio
              • package.scala
              • channels
                • GatheringByteChannel.scala
                • Channel.scala
                • Selector.scala
                • SelectableChannel.scala
                • AsynchronousFileChannel.scala
                • AsynchronousChannel.scala
                • ScatteringByteChannel.scala
                • FileChannel.scala
                • DatagramChannel.scala
                • spi
                  • SelectorProvider.scala
              • file
                • Files.scala
                • FileSystem.scala
      • test
        • resources
          • scattering_read_test.txt
          • async_file_read_test.txt
        • scala
          • zio
            • nio
              • channels
                • DatagramChannelSpec.scala
                • SelectorSpec.scala
                • ChannelSpec.scala
                • ScatterGatherChannelSpec.scala
                • FileChannelSpec.scala
              • BaseSpec.scala

package zio.nio.channels

import java.io.{ File, RandomAccessFile }

import zio.nio.core.Buffer
import zio.nio.BaseSpec
import zio.test.Assertion._
import zio.test._
import zio.{ Chunk, IO, ZIO }

import scala.io.Source

object ScatterGatherChannelSpec extends BaseSpec {

  override def spec = suite("ScatterGatherChannelSpec")(
    testM("scattering read") {
      for {
        raf         <- ZIO.effectTotal(new RandomAccessFile("nio/src/test/resources/scattering_read_test.txt", "r"))
        fileChannel = raf.getChannel
        readLine = (buffer: Buffer[Byte]) =>
          for {
            _     <- buffer.flip
            array <- buffer.array
            text  = array.takeWhile(_ != 10).map(_.toChar).mkString.trim
          } yield text
        buffs <- IO.collectAll(Seq(Buffer.byte(5), Buffer.byte(5)))
        list <- FileChannel(fileChannel).use { channel =>
                 for {
                   _    <- channel.readBuffer(buffs)
                   list <- IO.collectAll(buffs.map(readLine))
                 } yield list
               }
      } yield assert(list)(equalTo("Hello" :: "World" :: Nil))
    },
    testM("gathering write") {
      for {
        file        <- ZIO.effect(new File("nio/src/test/resources/gathering_write_test.txt"))
        raf         = new RandomAccessFile(file, "rw")
        fileChannel = raf.getChannel

        buffs <- IO.collectAll(
                  Seq(
                    Buffer.byte(Chunk.fromArray("Hello".getBytes)),
                    Buffer.byte(Chunk.fromArray("World".getBytes))
                  )
                )
        _      <- FileChannel(fileChannel).use(_.writeBuffer(buffs).unit)
        result = Source.fromFile(file).getLines().toSeq
        _      = file.delete()
      } yield assert(result)(equalTo(Seq("HelloWorld")))
    }
  )
}