Actor で Generator

generatoractor.scala

package net.shomah4a.utils.generator.actor

import scala.actors


class GeneratorMessage
object StopIteration

class Yielder[T](postto: actors.Actor, msgobj: GeneratorMessage)
{
  def ! (v: T)
  {
    postto ! (msgobj, v)
  }
}

class Generator[T](genfunc: (Yielder[T])=>Unit) extends Iterator[T]
{
  private val self = actors.Actor.self

  private object message extends GeneratorMessage

  private val actor = actors.Actor.actor
  {
    val msger = new Yielder[T](this.self, message)

    genfunc(msger)

    this.self ! (this.message, StopIteration)
  }

  private var nextValue: Option[T] = None

  def iterate()
  {
    this.nextValue = this.self.receive
    {
      case (this.message, StopIteration) =>
        {
          None
        }
      case (this.message, v: T) => Some(v)
    }
  }

  this.iterate

  def hasNext() = this.nextValue != None

  def next() =
    {
      val ret = this.nextValue.get
      this.iterate
      ret
    }
}

object Generator
{
  def gen[T](func: Yielder[T] => Unit) = new Generator[T](func)
  def apply[T] = gen[T] _
}