Bug 1216175, ensure we GC/CC often enough on workers, r=baku

--HG--
extra : rebase_source : 04d97418c729087cbad157a35c6f57fa2f2edc60

dom/workers/WorkerPrivate.cpp

@@ -1448,6 +1448,11 @@ private:
   WorkerRun(JSContext* aCx, WorkerPrivate* aWorkerPrivate) override
   {
     aWorkerPrivate->GarbageCollectInternal(aCx, mShrinking, mCollectChildren);
+    if (mShrinking) {
+      // Either we've run the idle GC or explicit GC call from the parent,
+      // we can cancel the current timers.
+      aWorkerPrivate->CancelGCTimers();
+    }
     return true;
   }
 };
@@ -4484,7 +4489,9 @@ WorkerPrivate::DoRunLoop(JSContext* aCx)
       }
     }
 
+    bool hadDebuggerOrNormalRunnables = false;
     if (debuggerRunnablesPending || normalRunnablesPending) {
+      hadDebuggerOrNormalRunnables = true;
       // Start the periodic GC timer if it is not already running.
       SetGCTimerMode(PeriodicTimer);
     }
@@ -4525,7 +4532,8 @@ WorkerPrivate::DoRunLoop(JSContext* aCx)
       }
     }
 
-    if (!debuggerRunnablesPending && !normalRunnablesPending) {
+    if (hadDebuggerOrNormalRunnables &&
+        (!debuggerRunnablesPending && !normalRunnablesPending)) {
       // Both the debugger event queue and the normal event queue has been
       // exhausted, cancel the periodic GC timer and schedule the idle GC timer.
       SetGCTimerMode(IdleTimer);
@@ -4578,20 +4586,20 @@ WorkerPrivate::InitializeGCTimers()
 {
   AssertIsOnWorkerThread();
 
-  // We need a timer for GC. The basic plan is to run a non-shrinking GC
+  // We need timers for GC. The basic plan is to run a non-shrinking GC
   // periodically (PERIODIC_GC_TIMER_DELAY_SEC) while the worker is running.
   // Once the worker goes idle we set a short (IDLE_GC_TIMER_DELAY_SEC) timer to
-  // run a shrinking GC. If the worker receives more messages then the short
-  // timer is canceled and the periodic timer resumes.
-  mGCTimer = do_CreateInstance(NS_TIMER_CONTRACTID);
-  MOZ_ASSERT(mGCTimer);
-
+  // run a shrinking GC.
+  mPeriodicGCTimer = do_CreateInstance(NS_TIMER_CONTRACTID);
   RefPtr<GarbageCollectRunnable> runnable =
     new GarbageCollectRunnable(this, false, false);
-  mPeriodicGCTimerTarget = new TimerThreadEventTarget(this, runnable);
+  nsCOMPtr<nsIEventTarget> target = new TimerThreadEventTarget(this, runnable);
+  MOZ_ALWAYS_TRUE(NS_SUCCEEDED(mPeriodicGCTimer->SetTarget(target)));
 
+  mIdleGCTimer = do_CreateInstance(NS_TIMER_CONTRACTID);
   runnable = new GarbageCollectRunnable(this, true, false);
-  mIdleGCTimerTarget = new TimerThreadEventTarget(this, runnable);
+  target = new TimerThreadEventTarget(this, runnable);
+  MOZ_ALWAYS_TRUE(NS_SUCCEEDED(mIdleGCTimer->SetTarget(target)));
 
   mPeriodicGCTimerRunning = false;
   mIdleGCTimerRunning = false;
@@ -4601,56 +4609,49 @@ void
 WorkerPrivate::SetGCTimerMode(GCTimerMode aMode)
 {
   AssertIsOnWorkerThread();
-  MOZ_ASSERT(mGCTimer);
-  MOZ_ASSERT(mPeriodicGCTimerTarget);
-  MOZ_ASSERT(mIdleGCTimerTarget);
+  MOZ_ASSERT(mPeriodicGCTimer);
+  MOZ_ASSERT(mIdleGCTimer);
 
-  if ((aMode == PeriodicTimer && mPeriodicGCTimerRunning) ||
-      (aMode == IdleTimer && mIdleGCTimerRunning)) {
+  // If we schedule the idle timer, cancel the periodic timer.
+  // But if the idle timer is running, don't cancel it when the periodic timer
+  // is scheduled since we do want shrinking GC to be called occasionally.
+  if (aMode == PeriodicTimer && mPeriodicGCTimerRunning) {
     return;
   }
 
-  MOZ_ALWAYS_TRUE(NS_SUCCEEDED(mGCTimer->Cancel()));
-
+  if (aMode == IdleTimer || aMode == NoTimer) {
+    MOZ_ALWAYS_TRUE(NS_SUCCEEDED(mPeriodicGCTimer->Cancel()));
     mPeriodicGCTimerRunning = false;
-  mIdleGCTimerRunning = false;
+  }
 
-  LOG(("Worker %p canceled GC timer because %s\n", this,
-       aMode == PeriodicTimer ?
-       "periodic" :
-       aMode == IdleTimer ? "idle" : "none"));
+  if (aMode == IdleTimer && mIdleGCTimerRunning) {
+    return;
+  }
 
   if (aMode == NoTimer) {
+    MOZ_ALWAYS_TRUE(NS_SUCCEEDED(mIdleGCTimer->Cancel()));
+    mIdleGCTimerRunning = false;
     return;
   }
 
   MOZ_ASSERT(aMode == PeriodicTimer || aMode == IdleTimer);
 
-  nsIEventTarget* target;
-  uint32_t delay;
-  int16_t type;
-
   if (aMode == PeriodicTimer) {
-    target = mPeriodicGCTimerTarget;
-    delay = PERIODIC_GC_TIMER_DELAY_SEC * 1000;
-    type = nsITimer::TYPE_REPEATING_SLACK;
-  }
-  else {
-    target = mIdleGCTimerTarget;
-    delay = IDLE_GC_TIMER_DELAY_SEC * 1000;
-    type = nsITimer::TYPE_ONE_SHOT;
-  }
-
-  MOZ_ALWAYS_TRUE(NS_SUCCEEDED(mGCTimer->SetTarget(target)));
+    uint32_t delay = PERIODIC_GC_TIMER_DELAY_SEC * 1000;
+    int16_t type = nsITimer::TYPE_REPEATING_SLACK;
     MOZ_ALWAYS_TRUE(NS_SUCCEEDED(
-    mGCTimer->InitWithNamedFuncCallback(DummyCallback, nullptr, delay, type,
+      mPeriodicGCTimer->
+        InitWithNamedFuncCallback(DummyCallback, nullptr, delay, type,
                                   "dom::workers::DummyCallback(2)")));
-
-  if (aMode == PeriodicTimer) {
     LOG(("Worker %p scheduled periodic GC timer\n", this));
     mPeriodicGCTimerRunning = true;
-  }
-  else {
+  } else {
+    uint32_t delay = IDLE_GC_TIMER_DELAY_SEC * 1000;
+    int16_t type = nsITimer::TYPE_ONE_SHOT;
+    MOZ_ALWAYS_TRUE(NS_SUCCEEDED(
+      mIdleGCTimer->
+        InitWithNamedFuncCallback(DummyCallback, nullptr, delay, type,
+                                  "dom::workers::DummyCallback(3)")));
     LOG(("Worker %p scheduled idle GC timer\n", this));
     mIdleGCTimerRunning = true;
   }
@@ -4661,16 +4662,17 @@ WorkerPrivate::ShutdownGCTimers()
 {
   AssertIsOnWorkerThread();
 
-  MOZ_ASSERT(mGCTimer);
+  MOZ_ASSERT(mPeriodicGCTimer);
+  MOZ_ASSERT(mIdleGCTimer);
 
-  // Always make sure the timer is canceled.
-  MOZ_ALWAYS_TRUE(NS_SUCCEEDED(mGCTimer->Cancel()));
+  // Always make sure the timers are canceled.
+  MOZ_ALWAYS_TRUE(NS_SUCCEEDED(mPeriodicGCTimer->Cancel()));
+  MOZ_ALWAYS_TRUE(NS_SUCCEEDED(mIdleGCTimer->Cancel()));
 
-  LOG(("Worker %p killed the GC timer\n", this));
+  LOG(("Worker %p killed the GC timers\n", this));
 
-  mGCTimer = nullptr;
-  mPeriodicGCTimerTarget = nullptr;
-  mIdleGCTimerTarget = nullptr;
+  mPeriodicGCTimer = nullptr;
+  mIdleGCTimer = nullptr;
   mPeriodicGCTimerRunning = false;
   mIdleGCTimerRunning = false;
 }
@@ -6088,7 +6090,7 @@ WorkerPrivate::RescheduleTimeoutTimer(JSContext* aCx)
 
   nsresult rv = mTimer->InitWithNamedFuncCallback(
     DummyCallback, nullptr, delay, nsITimer::TYPE_ONE_SHOT,
-    "dom::workers::DummyCallback(3)");
+    "dom::workers::DummyCallback(4)");
   if (NS_FAILED(rv)) {
     JS_ReportError(aCx, "Failed to start timer!");
     return false;

dom/workers/WorkerPrivate.h

@@ -927,9 +927,8 @@ class WorkerPrivate : public WorkerPrivateParent<WorkerPrivate>
 
   nsCOMPtr<nsITimer> mTimer;
 
-  nsCOMPtr<nsITimer> mGCTimer;
-  nsCOMPtr<nsIEventTarget> mPeriodicGCTimerTarget;
-  nsCOMPtr<nsIEventTarget> mIdleGCTimerTarget;
+  nsCOMPtr<nsITimer> mPeriodicGCTimer;
+  nsCOMPtr<nsITimer> mIdleGCTimer;
 
   RefPtr<MemoryReporter> mMemoryReporter;
 
@@ -1294,6 +1293,12 @@ public:
   void
   MaybeDispatchLoadFailedRunnable();
 
+  void
+  CancelGCTimers()
+  {
+    SetGCTimerMode(NoTimer);
+  }
+
 private:
   WorkerPrivate(JSContext* aCx, WorkerPrivate* aParent,
                 const nsAString& aScriptURL, bool aIsChromeWorker,