Implement WebThread::{add,remove}TaskObserver.

webkit/glue/webthread_impl.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_WEBTHREAD_IMPL_H_
 #define WEBKIT_GLUE_WEBTHREAD_IMPL_H_
 
+#include <map>
+
 #include "base/threading/thread.h"
 #include "base/memory/scoped_ptr.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebThread.h"
 
 namespace webkit_glue {
 
-class WebThreadImpl : public WebKit::WebThread {
+class WebThreadBase : public WebKit::WebThread {
  public:
-  WebThreadImpl(const char* name);
+  virtual void addTaskObserver(TaskObserver* observer);
+  virtual void removeTaskObserver(TaskObserver* observer);
+
+ protected:
+  virtual bool IsCurrentThread() const = 0;
+
+ private:
+  class TaskObserverAdapter;
+  typedef std::map<TaskObserver*, TaskObserverAdapter*> TaskObserverMap;
+  TaskObserverMap task_observer_map_;
+};
+
+class WebThreadImpl : public WebThreadBase {
+ public:
+  explicit WebThreadImpl(const char* name);
   virtual ~WebThreadImpl();
 
   virtual void postTask(Task* task) OVERRIDE;
   virtual void postDelayedTask(Task* task, long long delay_ms) OVERRIDE;
 
   MessageLoop* message_loop() const { return thread_->message_loop(); }
 
  protected:
+  virtual bool IsCurrentThread() const;
+
+ private:
   scoped_ptr<base::Thread> thread_;
 };
 
-class WebThreadImplForMessageLoop : public WebKit::WebThread {
+class WebThreadImplForMessageLoop : public WebThreadBase {
  public:
-  WebThreadImplForMessageLoop(base::MessageLoopProxy* message_loop);
+  explicit WebThreadImplForMessageLoop(base::MessageLoopProxy* message_loop);
   virtual ~WebThreadImplForMessageLoop();
 
   virtual void postTask(Task* task) OVERRIDE;
   virtual void postDelayedTask(Task* task, long long delay_ms) OVERRIDE;
 
  protected:
+  virtual bool IsCurrentThread() const;
+
+ private:
   scoped_refptr<base::MessageLoopProxy> message_loop_;
 };
 
 } // namespace webkit_glue
 
 #endif