gopher.impl.MappedReadChannel.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scala-gopher_3.0.0-RC2 Show documentation

scala-gopher: asynchronous implementation of CSP ( go-like channels/selectors ) in scala

The newest version!

package gopher.impl

import gopher._
import scala.util._
import scala.util.control.NonFatal

class MappedReadChannel[F[_],A, B](internal: ReadChannel[F,A], f: A=> B) extends ReadChannel[F,B] {


  class MReader(nested: Reader[B]) extends Reader[A] {

    def wrappedFun(fun: (Try[B] => Unit) ): (Try[A] => Unit) = {
      case Success(a) =>
        val b = f(a)
        fun(Success(b))
      case Failure(ex) =>
        fun(Failure(ex))
    }

    //TODO: think, are we want to pass error to the next level ?
    override def capture(): Option[Try[A]=>Unit] = 
      nested.capture().map{ fun =>
         wrappedFun(fun)   
      }

    override def canExpire: Boolean = nested.canExpire

    override def isExpired: Boolean = nested.isExpired

    override def markUsed(): Unit = nested.markUsed()

    override def markFree(): Unit = nested.markFree()

  }

  def addReader(reader: Reader[B]): Unit = 
    internal.addReader(MReader(reader))
  
  def addDoneReader(reader: Reader[Unit]): Unit = internal.addDoneReader(reader)

  def gopherApi:Gopher[F] = internal.gopherApi

}

class MappedAsyncReadChannel[F[_],A, B](internal: ReadChannel[F,A], f: A=> F[B]) extends ReadChannel[F,B] {

  def addDoneReader(reader: Reader[Unit]): Unit = internal.addDoneReader(reader)

  class MReader(nested: Reader[B]) extends Reader[A] {

    def wrappedFun(fun: (Try[B] => Unit) ): (Try[A] => Unit) = {
      case Success(a) =>
        try{
          asyncMonad.spawn(
            asyncMonad.mapTry(f(a))(fun)
          )
        }catch{
          case NonFatal(ex) =>
            fun(Failure(ex))
        }
      case Failure(ex) =>
        fun(Failure(ex))
    }

    //TODO: think, are we want to pass error to the next level ?
    override def capture(): Option[Try[A]=>Unit] = 
      nested.capture().map{ fun =>
         wrappedFun(fun)   
      }

    override def canExpire: Boolean = nested.canExpire

    override def isExpired: Boolean = nested.isExpired

    override def markUsed(): Unit = nested.markUsed()

    override def markFree(): Unit = nested.markFree()

  }

  def addReader(reader: Reader[B]): Unit = 
    internal.addReader(MReader(reader))
  
  def gopherApi:Gopher[F] = internal.gopherApi

}