First pass at providing a convenient API for handling websockets

1 files changed, 37 insertions, 0 deletions

diff --git a/cask/src/cask/internal/BatchActor.scala b/cask/src/cask/internal/BatchActor.scala
new file mode 100644
index 0000000..1566a18
--- /dev/null
+++ b/cask/src/cask/internal/BatchActor.scala
@@ -0,0 +1,37 @@
+package cask.internal
+
+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) {
+  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 => e.printStackTrace()}
+    synchronized{
+      if (queue.nonEmpty) ec.execute(() => runWithItems())
+      else{
+        assert(scheduled)
+        scheduled = false
+      }
+    }
+
+  }
+}