Track webcore worker message loops in chromium.

webkit/glue/worker_task_runner.h

+// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef WEBKIT_GLUE_WORKER_TASK_RUNNER_H_
+#define WEBKIT_GLUE_WORKER_TASK_RUNNER_H_
+#pragma once
+
+#include <map>
+
+#include "base/atomic_sequence_num.h"
+#include "base/callback_forward.h"
+#include "base/synchronization/lock.h"
+#include "base/threading/thread_local.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebWorkerRunLoop.h"
+#include "webkit/glue/webkit_glue_export.h"
+
+namespace webkit_glue {
+
+class WEBKIT_GLUE_EXPORT WorkerTaskRunner {
+ public:
+  WorkerTaskRunner();
+
+  void PostTask(int id, const base::Closure& task);
+  int CurrentWorkerId();
+  static WorkerTaskRunner* Instance();
+
+  class Observer {
+   public:
+    virtual ~Observer() {}
+    virtual void OnWorkerRunLoopStopped() = 0;
+  };
+  // Add/Remove an observer that will get notified when the current worker run
+  // loop is stopping. This observer will not get notified when other threads
+  // are stopping.  It's only valid to call these on a worker thread.
+  void AddStopObserver(Observer* observer);
+  void RemoveStopObserver(Observer* observer);
+
+ private:
+  friend class WebKitPlatformSupportImpl;
+  friend class WorkerTaskRunnerTest;
+
+  typedef std::map<int, WebKit::WebWorkerRunLoop> IDToLoopMap;
+
+  ~WorkerTaskRunner();
+  void OnWorkerRunLoopStarted(const WebKit::WebWorkerRunLoop& loop);
+  void OnWorkerRunLoopStopped(const WebKit::WebWorkerRunLoop& loop);
+
+  struct ThreadLocalState;
+  base::ThreadLocalPointer<ThreadLocalState> current_tls_;
+
+  base::AtomicSequenceNumber id_sequence_;
+  IDToLoopMap loop_map_;
+  base::Lock loop_map_lock_;
+};
+
+}  // namespace webkit_glue
+
+#endif // WEBKIT_GLUE_WORKER_TASK_RUNNER_H_