Track webcore worker message loops in chromium.

content/common/worker_task_runner.cc

+// 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.
+
+#include "base/callback.h"
+#include "content/common/worker_task_runner.h"
+
+using WebKit::WebWorkerRunLoop;
+
+namespace {
+
+class RunClosureTask : public WebKit::WebWorkerRunLoop::Task {
+ public:
+  RunClosureTask(const base::Closure& task) : task_(task) { }
+  virtual ~RunClosureTask() { }
+  virtual void Run() {
+    task_.Run();
+  }
+ private:
+  base::Closure task_;
+};
+
+}
+
+WorkerTaskRunner::WorkerTaskRunner() : id_sequence_(0) {
+}
+
+void WorkerTaskRunner::PostTask(
+    int id, const base::Closure& closure) {
+  DCHECK(id > 0);
+  base::AutoLock locker_(lock_);
+  IDToLoopMap::iterator found = loop_map_.find(id);
+  if (found != loop_map_.end())
+    found->second.postTask(new RunClosureTask(closure));
+}
+
+int WorkerTaskRunner::CurrentWorkerId() {
+  if (!current_tls_.Get())
+    return 0;
+  return current_tls_.Get()->first;
+}
+
+void WorkerTaskRunner::AddObserver(Observer* obs) {
+  base::AutoLock locker(observer_lock_);
+  observer_list_.AddObserver(obs);
+}
+
+void WorkerTaskRunner::RemoveObserver(Observer* obs) {
+  base::AutoLock locker(observer_lock_);
+  observer_list_.RemoveObserver(obs);
+}
+
+WorkerTaskRunner::~WorkerTaskRunner() {
+}
+
+void WorkerTaskRunner::OnWorkerRunLoopStarted(const WebWorkerRunLoop& loop) {
+  DCHECK(!current_tls_.Get());
+  {
+    base::AutoLock locker_(lock_);
+    int id = ++id_sequence_;
+    current_tls_.Set(new std::pair<int, WebWorkerRunLoop>(id, loop));
+    loop_map_[id] = loop;
+  }
+  base::AutoLock locker(observer_lock_);
+  FOR_EACH_OBSERVER(Observer, observer_list_, OnWorkerRunLoopStarted());
+}
+
+void WorkerTaskRunner::OnWorkerRunLoopStopped(
+    const WebWorkerRunLoop& loop) {
+  DCHECK(CurrentWorkerId() > 0);
+  DCHECK(current_tls_.Get());
+  {
+    base::AutoLock lock(observer_lock_);
+    FOR_EACH_OBSERVER(Observer, observer_list_, OnWorkerRunLoopStopped());
+  }
+  base::AutoLock locker_(lock_);
+  DCHECK(loop_map_[CurrentWorkerId()] == loop);
+  loop_map_.erase(CurrentWorkerId());
+  delete current_tls_.Get();
+  current_tls_.Set(NULL);
+}