Bug 1677851 - simplify DataStorage background task handling r=kjacobs,bbeurdouche

This patch removes the hand-rolled shared background thread in favor of
individual background synchronous event targets. Also, the timer configuration
was moved to the main thread. It now dispatches events to the background task
queue, which makes it easier to reason about.

Differential Revision: https://phabricator.services.mozilla.com/D98977

security/manager/ssl/DataStorage.cpp

@@ -16,6 +16,7 @@
 #include "mozilla/Services.h"
 #include "mozilla/StaticMutex.h"
 #include "mozilla/StaticPtr.h"
+#include "mozilla/TaskQueue.h"
 #include "mozilla/Telemetry.h"
 #include "mozilla/Unused.h"
 #include "nsAppDirectoryServiceDefs.h"
@@ -25,6 +26,7 @@
 #endif
 #include "nsIMemoryReporter.h"
 #include "nsIObserverService.h"
+#include "nsISerialEventTarget.h"
 #include "nsITimer.h"
 #include "nsIThread.h"
 #include "nsNetUtil.h"
@@ -48,108 +50,6 @@ static const uint32_t sMaxDataEntries = 1024;
 static const int64_t sOneDayInMicroseconds =
     int64_t(24 * 60 * 60) * PR_USEC_PER_SEC;
 
-namespace {
-
-// DataStorageSharedThread provides one shared thread that every DataStorage
-// instance can use to do background work (reading/writing files and scheduling
-// timers). This means we don't have to have one thread per DataStorage
-// instance. The shared thread is initialized when the first DataStorage
-// instance is initialized (Initialize is idempotent, so it's safe to call
-// multiple times in any case).
-// When Gecko shuts down, it will send a "profile-before-change" notification.
-// The first DataStorage instance to observe the notification will dispatch an
-// event for each known DataStorage (as tracked by sDataStorages) to write out
-// their backing data. That instance will then shut down the shared thread,
-// which ensures those events actually run. At that point sDataStorages is
-// cleared and any subsequent attempt to create a DataStorage will fail because
-// the shared thread will refuse to be instantiated again.
-// In some cases (e.g. xpcshell), no profile notification will be sent, so
-// instead we rely on the notification "xpcom-shutdown-threads".
-class DataStorageSharedThread final {
- public:
-  static nsresult Initialize();
-  static nsresult Shutdown();
-  static nsresult Dispatch(nsIRunnable* event);
-
-  virtual ~DataStorageSharedThread() = default;
-
- private:
-  DataStorageSharedThread() : mThread(nullptr) {}
-
-  nsCOMPtr<nsIThread> mThread;
-};
-
-mozilla::StaticMutex sDataStorageSharedThreadMutex;
-static mozilla::StaticAutoPtr<DataStorageSharedThread> gDataStorageSharedThread;
-static bool gDataStorageSharedThreadShutDown = false;
-
-nsresult DataStorageSharedThread::Initialize() {
-  MOZ_ASSERT(XRE_IsParentProcess() || XRE_IsSocketProcess());
-  mozilla::StaticMutexAutoLock lock(sDataStorageSharedThreadMutex);
-
-  // If this happens, we initialized a DataStorage after shutdown notifications
-  // were sent, so don't re-initialize the shared thread.
-  if (gDataStorageSharedThreadShutDown) {
-    return NS_ERROR_FAILURE;
-  }
-
-  if (!gDataStorageSharedThread) {
-    gDataStorageSharedThread = new DataStorageSharedThread();
-    nsresult rv = NS_NewNamedThread(
-        "DataStorage", getter_AddRefs(gDataStorageSharedThread->mThread));
-    if (NS_FAILED(rv)) {
-      gDataStorageSharedThread = nullptr;
-      return rv;
-    }
-  }
-
-  return NS_OK;
-}
-
-nsresult DataStorageSharedThread::Shutdown() {
-  MOZ_ASSERT(XRE_IsParentProcess() || XRE_IsSocketProcess());
-  mozilla::StaticMutexAutoLock lock(sDataStorageSharedThreadMutex);
-
-  if (!gDataStorageSharedThread || gDataStorageSharedThreadShutDown) {
-    return NS_OK;
-  }
-
-  MOZ_ASSERT(gDataStorageSharedThread->mThread);
-  if (!gDataStorageSharedThread->mThread) {
-    return NS_ERROR_FAILURE;
-  }
-
-  // We can't hold sDataStorageSharedThreadMutex while shutting down the thread,
-  // because we might process events that try to acquire it (e.g. via
-  // DataStorage::GetAllChildProcessData). So, we set our shutdown sentinel
-  // boolean to true, get a handle on the thread, release the lock, shut down
-  // the thread (thus processing all pending events on it), and re-acquire the
-  // lock.
-  gDataStorageSharedThreadShutDown = true;
-  nsCOMPtr<nsIThread> threadHandle = gDataStorageSharedThread->mThread;
-  nsresult rv;
-  {
-    mozilla::StaticMutexAutoUnlock unlock(sDataStorageSharedThreadMutex);
-    rv = threadHandle->Shutdown();
-  }
-  gDataStorageSharedThread->mThread = nullptr;
-  gDataStorageSharedThread = nullptr;
-
-  return rv;
-}
-
-nsresult DataStorageSharedThread::Dispatch(nsIRunnable* event) {
-  MOZ_ASSERT(XRE_IsParentProcess() || XRE_IsSocketProcess());
-  mozilla::StaticMutexAutoLock lock(sDataStorageSharedThreadMutex);
-  if (gDataStorageSharedThreadShutDown || !gDataStorageSharedThread ||
-      !gDataStorageSharedThread->mThread) {
-    return NS_ERROR_FAILURE;
-  }
-  return gDataStorageSharedThread->mThread->Dispatch(event, NS_DISPATCH_NORMAL);
-}
-
-}  // unnamed namespace
-
 namespace mozilla {
 
 class DataStorageMemoryReporter final : public nsIMemoryReporter {
@@ -184,7 +84,6 @@ mozilla::StaticAutoPtr<DataStorage::DataStorages> DataStorage::sDataStorages;
 
 DataStorage::DataStorage(const nsString& aFilename)
     : mMutex("DataStorage::mMutex"),
-      mTimerDelay(sDataStorageDefaultTimerDelay),
       mPendingWrite(false),
       mShuttingDown(false),
       mInitCalled(false),
@@ -192,11 +91,6 @@ DataStorage::DataStorage(const nsString& aFilename)
       mReady(false),
       mFilename(aFilename) {}
 
-DataStorage::~DataStorage() {
-  Preferences::UnregisterCallback(DataStorage::PrefChanged,
-                                  "test.datastorage.write_timer_ms", this);
-}
-
 // static
 already_AddRefed<DataStorage> DataStorage::Get(DataStorageClass aFilename) {
   switch (aFilename) {
@@ -341,16 +235,31 @@ nsresult DataStorage::Init(const nsTArray<DataStorageItem>* aItems,
     memoryReporterRegistered = true;
   }
 
-  nsresult rv;
+  if (XRE_IsParentProcess() || XRE_IsSocketProcess()) {
-  if (XRE_IsParentProcess()) {
+    nsCOMPtr<nsISerialEventTarget> target;
-    MOZ_ASSERT(!aItems);
+    nsresult rv = NS_CreateBackgroundTaskQueue(
-
+        "DataStorage::mBackgroundTaskQueue", getter_AddRefs(target));
-    rv = DataStorageSharedThread::Initialize();
     if (NS_WARN_IF(NS_FAILED(rv))) {
       return rv;
     }
+    mBackgroundTaskQueue = new TaskQueue(target.forget());
 
-    rv = AsyncReadData(lock);
+    // For test purposes, we can set the write timer to be very fast.
+    uint32_t timerDelayMS = Preferences::GetInt(
+        "test.datastorage.write_timer_ms", sDataStorageDefaultTimerDelay);
+    rv = NS_NewTimerWithFuncCallback(
+        getter_AddRefs(mTimer), DataStorage::TimerCallback, this, timerDelayMS,
+        nsITimer::TYPE_REPEATING_SLACK_LOW_PRIORITY, "DataStorageTimer",
+        mBackgroundTaskQueue);
+    if (NS_WARN_IF(NS_FAILED(rv))) {
+      return rv;
+    }
+  }
+
+  if (XRE_IsParentProcess()) {
+    MOZ_ASSERT(!aItems);
+
+    nsresult rv = AsyncReadData(lock);
     if (NS_FAILED(rv)) {
       return rv;
     }
@@ -361,18 +270,13 @@ nsresult DataStorage::Init(const nsTArray<DataStorageItem>* aItems,
     MOZ_ASSERT(aItems);
 
     if (XRE_IsSocketProcess() && aWriteFd.IsValid()) {
-      rv = DataStorageSharedThread::Initialize();
-      if (NS_WARN_IF(NS_FAILED(rv))) {
-        return rv;
-      }
-
       mWriteFd = aWriteFd;
     }
 
     for (auto& item : *aItems) {
       Entry entry;
       entry.mValue = item.value();
-      rv = PutInternal(item.key(), entry, item.type(), lock);
+      nsresult rv = PutInternal(item.key(), entry, item.type(), lock);
       if (NS_FAILED(rv)) {
         return rv;
       }
@@ -387,24 +291,16 @@ nsresult DataStorage::Init(const nsTArray<DataStorageItem>* aItems,
   }
   // Clear private data as appropriate.
   os->AddObserver(this, "last-pb-context-exited", false);
-  // Observe shutdown; save data and prevent any further writes.
+  if (XRE_IsParentProcess() || XRE_IsSocketProcess()) {
-  // In the parent process, we need to write to the profile directory, so
+    // Observe shutdown; save data and prevent any further writes.
-  // we should listen for profile-before-change so that we can safely write to
+    // In the parent process, we need to write to the profile directory, so
-  // the profile. In the content process however we
+    // we should listen for profile-before-change so that we can safely write to
-  // don't have access to the profile directory and profile notifications are
+    // the profile.
-  // not dispatched, so we need to clean up on xpcom-shutdown-threads.
-  if (XRE_IsParentProcess()) {
     os->AddObserver(this, "profile-before-change", false);
+    // In the Parent process, this is a backstop for xpcshell and other cases
+    // where profile-before-change might not get sent.
+    os->AddObserver(this, "xpcom-shutdown-threads", false);
   }
-  // In the Parent process, this is a backstop for xpcshell and other cases
-  // where profile-before-change might not get sent.
-  os->AddObserver(this, "xpcom-shutdown-threads", false);
-
-  // For test purposes, we can set the write timer to be very fast.
-  mTimerDelay = Preferences::GetInt("test.datastorage.write_timer_ms",
-                                    sDataStorageDefaultTimerDelay);
-  Preferences::RegisterCallback(DataStorage::PrefChanged,
-                                "test.datastorage.write_timer_ms", this);
 
   return NS_OK;
 }
@@ -480,8 +376,8 @@ nsresult DataStorage::AsyncTakeFileDesc(
     return NS_ERROR_NOT_AVAILABLE;
   }
 
-  RefPtr<Opener> job(new Opener(mBackingFile, std::move(aResolver)));
+  nsCOMPtr<nsIRunnable> job(new Opener(mBackingFile, std::move(aResolver)));
-  nsresult rv = DataStorageSharedThread::Dispatch(job);
+  nsresult rv = mBackgroundTaskQueue->Dispatch(job.forget());
   if (NS_WARN_IF(NS_FAILED(rv))) {
     return rv;
   }
@@ -707,7 +603,7 @@ nsresult DataStorage::AsyncReadData(const MutexAutoLock& /*aProofOfLock*/) {
   // Allocate a Reader so that even if it isn't dispatched,
   // the data-storage-ready notification will be fired and Get
   // will be able to proceed (this happens in its destructor).
-  RefPtr<Reader> job(new Reader(this));
+  nsCOMPtr<nsIRunnable> job(new Reader(this));
   nsresult rv;
   // If we don't have a profile directory, this will fail.
   // That's okay - it just means there is no persistent state.
@@ -723,7 +619,7 @@ nsresult DataStorage::AsyncReadData(const MutexAutoLock& /*aProofOfLock*/) {
     return rv;
   }
 
-  rv = DataStorageSharedThread::Dispatch(job);
+  rv = mBackgroundTaskQueue->Dispatch(job.forget());
   if (NS_WARN_IF(NS_FAILED(rv))) {
     return rv;
   }
@@ -907,8 +803,8 @@ nsresult DataStorage::PutInternal(const nsCString& aKey, Entry& aEntry,
   DataStorageTable& table = GetTableForType(aType, aProofOfLock);
   table.Put(aKey, aEntry);
 
-  if (aType == DataStorage_Persistent && !mPendingWrite) {
+  if (aType == DataStorage_Persistent) {
-    return AsyncSetTimer(aProofOfLock);
+    mPendingWrite = true;
   }
 
   return NS_OK;
@@ -921,8 +817,8 @@ void DataStorage::Remove(const nsCString& aKey, DataStorageType aType) {
   DataStorageTable& table = GetTableForType(aType, lock);
   table.Remove(aKey);
 
-  if (aType == DataStorage_Persistent && !mPendingWrite) {
+  if (aType == DataStorage_Persistent) {
-    Unused << AsyncSetTimer(lock);
+    mPendingWrite = true;
   }
 
   nsString filename(mFilename);
@@ -1025,7 +921,8 @@ DataStorage::Writer::Run() {
 nsresult DataStorage::AsyncWriteData(const MutexAutoLock& /*aProofOfLock*/) {
   MOZ_ASSERT(XRE_IsParentProcess() || XRE_IsSocketProcess());
 
-  if (mShuttingDown || (!mBackingFile && !mWriteFd.IsValid())) {
+  if (!mPendingWrite || mShuttingDown ||
+      (!mBackingFile && !mWriteFd.IsValid())) {
     return NS_OK;
   }
 
@@ -1042,8 +939,8 @@ nsresult DataStorage::AsyncWriteData(const MutexAutoLock& /*aProofOfLock*/) {
     output.Append('\n');
   }
 
-  RefPtr<Writer> job(new Writer(output, this));
+  nsCOMPtr<nsIRunnable> job(new Writer(output, this));
-  nsresult rv = DataStorageSharedThread::Dispatch(job);
+  nsresult rv = mBackgroundTaskQueue->Dispatch(job.forget());
   mPendingWrite = false;
   if (NS_WARN_IF(NS_FAILED(rv))) {
     return rv;
@@ -1058,6 +955,7 @@ nsresult DataStorage::Clear() {
   mPersistentDataTable.Clear();
   mTemporaryDataTable.Clear();
   mPrivateDataTable.Clear();
+  mPendingWrite = true;
 
   if (XRE_IsParentProcess() || XRE_IsSocketProcess()) {
     // Asynchronously clear the file. This is similar to the permission manager
@@ -1086,43 +984,6 @@ void DataStorage::TimerCallback(nsITimer* aTimer, void* aClosure) {
   Unused << aDataStorage->AsyncWriteData(lock);
 }
 
-// We only initialize the timer on the worker thread because it's not safe
-// to mix what threads are operating on the timer.
-nsresult DataStorage::AsyncSetTimer(const MutexAutoLock& /*aProofOfLock*/) {
-  if (mShuttingDown || (!XRE_IsParentProcess() && !XRE_IsSocketProcess())) {
-    return NS_OK;
-  }
-
-  mPendingWrite = true;
-  nsCOMPtr<nsIRunnable> job =
-      NewRunnableMethod("DataStorage::SetTimer", this, &DataStorage::SetTimer);
-  nsresult rv = DataStorageSharedThread::Dispatch(job);
-  if (NS_WARN_IF(NS_FAILED(rv))) {
-    return rv;
-  }
-  return NS_OK;
-}
-
-void DataStorage::SetTimer() {
-  MOZ_ASSERT(!NS_IsMainThread());
-  MOZ_ASSERT(XRE_IsParentProcess() || XRE_IsSocketProcess());
-
-  MutexAutoLock lock(mMutex);
-
-  nsresult rv;
-  if (!mTimer) {
-    mTimer = NS_NewTimer();
-    if (NS_WARN_IF(!mTimer)) {
-      return;
-    }
-  }
-
-  rv = mTimer->InitWithNamedFuncCallback(TimerCallback, this, mTimerDelay,
-                                         nsITimer::TYPE_ONE_SHOT,
-                                         "DataStorage::SetTimer");
-  Unused << NS_WARN_IF(NS_FAILED(rv));
-}
-
 void DataStorage::NotifyObservers(const char* aTopic) {
   // Don't access the observer service off the main thread.
   if (!NS_IsMainThread()) {
@@ -1137,26 +998,14 @@ void DataStorage::NotifyObservers(const char* aTopic) {
   }
 }
 
-nsresult DataStorage::DispatchShutdownTimer(
-    const MutexAutoLock& /*aProofOfLock*/) {
-  MOZ_ASSERT(XRE_IsParentProcess() || XRE_IsSocketProcess());
-
-  nsCOMPtr<nsIRunnable> job = NewRunnableMethod(
-      "DataStorage::ShutdownTimer", this, &DataStorage::ShutdownTimer);
-  nsresult rv = DataStorageSharedThread::Dispatch(job);
-  if (NS_WARN_IF(NS_FAILED(rv))) {
-    return rv;
-  }
-  return NS_OK;
-}
-
 void DataStorage::ShutdownTimer() {
   MOZ_ASSERT(XRE_IsParentProcess() || XRE_IsSocketProcess());
-  MOZ_ASSERT(!NS_IsMainThread());
+  MOZ_ASSERT(NS_IsMainThread());
-  MutexAutoLock lock(mMutex);
+  if (mTimer) {
-  nsresult rv = mTimer->Cancel();
+    nsresult rv = mTimer->Cancel();
-  Unused << NS_WARN_IF(NS_FAILED(rv));
+    Unused << NS_WARN_IF(NS_FAILED(rv));
-  mTimer = nullptr;
+    mTimer = nullptr;
+  }
 }
 
 //------------------------------------------------------------
@@ -1166,7 +1015,6 @@ void DataStorage::ShutdownTimer() {
 NS_IMETHODIMP
 DataStorage::Observe(nsISupports* /*aSubject*/, const char* aTopic,
                      const char16_t* /*aData*/) {
-  // Don't access preferences off the main thread.
   if (!NS_IsMainThread()) {
     MOZ_ASSERT_UNREACHABLE("DataStorage::Observe called off main thread");
     return NS_ERROR_NOT_SAME_THREAD;
@@ -1175,52 +1023,38 @@ DataStorage::Observe(nsISupports* /*aSubject*/, const char* aTopic,
   if (strcmp(aTopic, "last-pb-context-exited") == 0) {
     MutexAutoLock lock(mMutex);
     mPrivateDataTable.Clear();
-  }
-
-  if (!XRE_IsParentProcess() && !XRE_IsSocketProcess()) {
-    if (strcmp(aTopic, "xpcom-shutdown-threads") == 0) {
-      sDataStorages->Clear();
-    }
     return NS_OK;
   }
 
-  // The first DataStorage to observe a shutdown notification will dispatch
+  if (!XRE_IsParentProcess() && !XRE_IsSocketProcess()) {
-  // events for all DataStorages to write their data out. It will then shut down
+    MOZ_ASSERT_UNREACHABLE("unexpected observation topic for content proces");
-  // the shared background thread, which actually runs those events. The table
+    return NS_ERROR_UNEXPECTED;
-  // of DataStorages is then cleared, turning further observations by any other
+  }
-  // DataStorages into no-ops.
+
   if (strcmp(aTopic, "profile-before-change") == 0 ||
       strcmp(aTopic, "xpcom-shutdown-threads") == 0) {
-    for (auto iter = sDataStorages->Iter(); !iter.Done(); iter.Next()) {
+    RefPtr<TaskQueue> taskQueueToAwait;
-      RefPtr<DataStorage> storage = iter.UserData();
+    {
-      MutexAutoLock lock(storage->mMutex);
+      MutexAutoLock lock(mMutex);
-      if (!storage->mShuttingDown) {
+      if (!mShuttingDown) {
-        nsresult rv = storage->AsyncWriteData(lock);
+        nsresult rv = AsyncWriteData(lock);
-        storage->mShuttingDown = true;
         Unused << NS_WARN_IF(NS_FAILED(rv));
-        if (storage->mTimer) {
+        mShuttingDown = true;
-          Unused << storage->DispatchShutdownTimer(lock);
+        mBackgroundTaskQueue->BeginShutdown();
-        }
+        taskQueueToAwait = mBackgroundTaskQueue;
       }
     }
-    sDataStorages->Clear();
+    // Tasks on the background queue may take the lock, so it can't be held
-    DataStorageSharedThread::Shutdown();
+    // while waiting for them to finish.
+    if (taskQueueToAwait) {
+      taskQueueToAwait->AwaitShutdownAndIdle();
+    }
+    ShutdownTimer();
   }
 
   return NS_OK;
 }
 
-// static
-void DataStorage::PrefChanged(const char* aPref, void* aSelf) {
-  static_cast<DataStorage*>(aSelf)->PrefChanged(aPref);
-}
-
-void DataStorage::PrefChanged(const char* aPref) {
-  MutexAutoLock lock(mMutex);
-  mTimerDelay = Preferences::GetInt("test.datastorage.write_timer_ms",
-                                    sDataStorageDefaultTimerDelay);
-}
-
 DataStorage::Entry::Entry()
     : mScore(0), mLastAccessed((int32_t)(PR_Now() / sOneDayInMicroseconds)) {}
 

security/manager/ssl/DataStorage.h

@@ -25,6 +25,7 @@ class psm_DataStorageTest;
 
 namespace mozilla {
 class DataStorageMemoryReporter;
+class TaskQueue;
 
 namespace dom {
 class ContentChild;
@@ -52,24 +53,24 @@ class DataStorageItem;
  * - Should the profile directory not be available, (e.g. in xpcshell),
  *   DataStorage will not initially read any persistent data. The
  *   "data-storage-ready" event will still be emitted. This follows semantics
- *   similar to the permission manager and allows tests that test
+ *   similar to the permission manager and allows tests that test unrelated
- *   unrelated components to proceed without a profile.
+ *   components to proceed without a profile.
- * - When any persistent data changes, a timer is initialized that will
+ * - A timer periodically fires on a background thread that checks if any
- *   eventually asynchronously write all persistent data to the backing file.
+ *   persistent data has changed, and if so writes all persistent data to the
- *   When this happens, observers will be notified with the topic
+ *   backing file. When this happens, observers will be notified with the
- *   "data-storage-written" and the backing filename as the data.
+ *   topic "data-storage-written" and the backing filename as the data.
  *   It is possible to receive a "data-storage-written" event while there exist
- *   pending persistent data changes. However, those changes will cause the
+ *   pending persistent data changes. However, those changes will eventually be
- *   timer to be reinitialized and another "data-storage-written" event will
+ *   written when the timer fires again, and eventually another
- *   be sent.
+ *   "data-storage-written" event will be sent.
- * - When any DataStorage observes the topic "profile-before-change" in
+ * - When a DataStorage instance observes the topic "profile-before-change" in
- *   anticipation of shutdown, all persistent data for all DataStorage instances
+ *   anticipation of shutdown, all persistent data for that DataStorage is
- *   is synchronously written to the appropriate backing file. The worker thread
+ *   written to the backing file (this blocks the main thread). In the process
- *   responsible for these writes is then disabled to prevent further writes to
+ *   of doing this, the background serial event target responsible for these
- *   that file (the delayed-write timer is cancelled when this happens). Note
+ *   writes is then shut down to prevent further writes to that file (the
- *   that the "worker thread" is actually a single thread shared between all
+ *   background timer is also cancelled when this happens).
- *   DataStorage instances. If "profile-before-change" is not observed, this
+ *   If "profile-before-change" is not observed, this happens upon observing
- *   happens upon observing "xpcom-shutdown-threads".
+ *   "xpcom-shutdown-threads".
  * - For testing purposes, the preference "test.datastorage.write_timer_ms" can
  *   be set to cause the asynchronous writing of data to happen more quickly.
  * - To prevent unbounded memory and disk use, the number of entries in each
@@ -167,9 +168,11 @@ class DataStorage : public nsIObserver {
   // Return true if this data storage is ready to be used.
   bool IsReady();
 
+  void ShutdownTimer();
+
  private:
   explicit DataStorage(const nsString& aFilename);
-  virtual ~DataStorage();
+  virtual ~DataStorage() = default;
 
   static already_AddRefed<DataStorage> GetFromRawFileName(
       const nsString& aFilename);
@@ -205,19 +208,12 @@ class DataStorage : public nsIObserver {
   void WaitForReady();
   nsresult AsyncWriteData(const MutexAutoLock& aProofOfLock);
   nsresult AsyncReadData(const MutexAutoLock& aProofOfLock);
-  nsresult AsyncSetTimer(const MutexAutoLock& aProofOfLock);
-  nsresult DispatchShutdownTimer(const MutexAutoLock& aProofOfLock);
 
   static nsresult ValidateKeyAndValue(const nsCString& aKey,
                                       const nsCString& aValue);
   static void TimerCallback(nsITimer* aTimer, void* aClosure);
-  void SetTimer();
-  void ShutdownTimer();
   void NotifyObservers(const char* aTopic);
 
-  static void PrefChanged(const char* aPref, void* aSelf);
-  void PrefChanged(const char* aPref);
-
   bool GetInternal(const nsCString& aKey, Entry* aEntry, DataStorageType aType,
                    const MutexAutoLock& aProofOfLock);
   nsresult PutInternal(const nsCString& aKey, Entry& aEntry,
@@ -237,13 +233,13 @@ class DataStorage : public nsIObserver {
   DataStorageTable mTemporaryDataTable;
   DataStorageTable mPrivateDataTable;
   nsCOMPtr<nsIFile> mBackingFile;
-  nsCOMPtr<nsITimer>
+  bool mPendingWrite;  // true if a write is needed but hasn't been dispatched
-      mTimer;            // All uses after init must be on the worker thread
-  uint32_t mTimerDelay;  // in milliseconds
-  bool mPendingWrite;    // true if a write is needed but hasn't been dispatched
   bool mShuttingDown;
+  RefPtr<TaskQueue> mBackgroundTaskQueue;
   // (End list of members protected by mMutex)
 
+  nsCOMPtr<nsITimer> mTimer;  // Must only be accessed on the main thread
+
   mozilla::Atomic<bool> mInitCalled;  // Indicates that Init() has been called.
 
   Monitor mReadyMonitor;  // Do not acquire this at the same time as mMutex.