diff options
Diffstat (limited to 'cask/util/src/cask/util/BatchActor.scala')
-rw-r--r-- | cask/util/src/cask/util/BatchActor.scala | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/cask/util/src/cask/util/BatchActor.scala b/cask/util/src/cask/util/BatchActor.scala new file mode 100644 index 0000000..137b852 --- /dev/null +++ b/cask/util/src/cask/util/BatchActor.scala @@ -0,0 +1,40 @@ +package cask.util + +import cask.util.Logger + +import scala.collection.mutable +import scala.concurrent.ExecutionContext + +/** + * A simple asynchrous actor, allowing safe concurrent asynchronous processing + * of queued items. `run` handles items in batches, to allow for batch + * processing optimizations to be used where relevant. + */ +abstract class BatchActor[T]()(implicit ec: ExecutionContext, + log: Logger) { + def run(items: Seq[T]): Unit + + private val queue = new mutable.Queue[T]() + private var scheduled = false + def send(t: => T): Unit = synchronized{ + queue.enqueue(t) + if (!scheduled){ + scheduled = true + ec.execute(() => runWithItems()) + } + } + + def runWithItems(): Unit = { + val items = synchronized(queue.dequeueAll(_ => true)) + try run(items) + catch{case e: Throwable => log.exception(e)} + synchronized{ + if (queue.nonEmpty) ec.execute(() => runWithItems()) + else{ + assert(scheduled) + scheduled = false + } + } + + } +} |