Move `internal.BatchActor` to `util.BatchActor`

1 files changed, 40 insertions, 0 deletions

diff --git a/cask/src/cask/util/BatchActor.scala b/cask/src/cask/util/BatchActor.scala
new file mode 100644
index 0000000..137b852
--- /dev/null
+++ b/cask/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
+      }
+    }
+
+  }
+}