Use ordinary JS object instead of Scala Map

Author: armanbilge

core/src/main/scala/org/scalajs/macrotaskexecutor/MacrotaskExecutor.scala

@@ -16,9 +16,9 @@
 
 package org.scalajs.macrotaskexecutor
 
-import scala.collection.mutable
 import scala.concurrent.{ExecutionContext, ExecutionContextExecutor}
 import scala.scalajs.js
+import scala.scalajs.js.annotation._
 import scala.util.Random
 import scala.util.control.NonFatal
 
@@ -34,10 +34,18 @@ object MacrotaskExecutor extends ExecutionContextExecutor {
   def reportFailure(cause: Throwable): Unit =
     cause.printStackTrace()
 
+  @js.native
+  private[this] trait TaskMap extends js.Object {
+    @JSBracketAccess
+    def apply(handle: Int): Runnable
+    @JSBracketAccess
+    def update(handle: Int, task: Runnable): Unit
+  }
+
   private[this] val setImmediate: Runnable => Unit = {
     if (js.typeOf(js.Dynamic.global.setImmediate) == Undefined) {
       var nextHandle = 1
-      val tasksByHandle = mutable.Map[Int, Runnable]()
+      val tasksByHandle = (new js.Object).asInstanceOf[TaskMap]
       var currentlyRunningATask = false
 
       def canUsePostMessage(): Boolean = {
@@ -63,21 +71,17 @@ object MacrotaskExecutor extends ExecutionContextExecutor {
         }
       }
 
-      def runIfPresent(handle: Int): Unit = {
+      def runTaskForHandle(handle: Int): Unit = {
         if (currentlyRunningATask) {
-          js.Dynamic.global.setTimeout(() => runIfPresent(handle), 0)
+          js.Dynamic.global.setTimeout(() => runTaskForHandle(handle), 0)
         } else {
-          tasksByHandle.get(handle) match {
-            case Some(task) =>
-              currentlyRunningATask = true
-              try {
-                task.run()
-              } finally {
-                tasksByHandle -= handle
-                currentlyRunningATask = false
-              }
-
-            case None =>
+          val task = tasksByHandle(handle)
+          currentlyRunningATask = true
+          try {
+            task.run()
+          } finally {
+            js.special.delete(tasksByHandle, handle)
+            currentlyRunningATask = false
           }
         }
 
@@ -115,7 +119,7 @@ object MacrotaskExecutor extends ExecutionContextExecutor {
               .data
               .indexOf(messagePrefix)
               .asInstanceOf[Int] == 0) {
-            runIfPresent(event.data.toString.substring(messagePrefix.length).toInt)
+            runTaskForHandle(event.data.toString.substring(messagePrefix.length).toInt)
           }
         }
 
@@ -129,22 +133,22 @@ object MacrotaskExecutor extends ExecutionContextExecutor {
           val handle = nextHandle
           nextHandle += 1
 
-          tasksByHandle += (handle -> k)
+          tasksByHandle(handle) = k
           js.Dynamic.global.postMessage(messagePrefix + handle, "*")
           ()
         }
       } else if (js.typeOf(js.Dynamic.global.MessageChannel) != Undefined) {
         val channel = js.Dynamic.newInstance(js.Dynamic.global.MessageChannel)()
 
         channel.port1.onmessage = { (event: js.Dynamic) =>
-          runIfPresent(event.data.asInstanceOf[Int])
+          runTaskForHandle(event.data.asInstanceOf[Int])
         }
 
         { k =>
           val handle = nextHandle
           nextHandle += 1
 
-          tasksByHandle += (handle -> k)
+          tasksByHandle(handle) = k
           channel.port2.postMessage(handle)
           ()
         }