| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef WEBKIT_QUOTA_QUOTA_TASK_H_ | 5 #ifndef WEBKIT_QUOTA_QUOTA_TASK_H_ |
| 6 #define WEBKIT_QUOTA_QUOTA_TASK_H_ | 6 #define WEBKIT_QUOTA_QUOTA_TASK_H_ |
| 7 #pragma once | 7 #pragma once |
| 8 | 8 |
| 9 #include <set> | 9 #include <set> |
| 10 | 10 |
| 11 #include "base/basictypes.h" | 11 #include "base/basictypes.h" |
| 12 #include "base/compiler_specific.h" | 12 #include "base/compiler_specific.h" |
| 13 #include "base/memory/ref_counted.h" | 13 #include "base/memory/ref_counted.h" |
| 14 #include "base/message_loop_proxy.h" | |
| 15 | 14 |
| 16 namespace base { | 15 namespace base { |
| 17 class MessageLoopProxy; | 16 class SingleThreadTaskRunner; |
| 17 class TaskRunner; |
| 18 } | 18 } |
| 19 | 19 |
| 20 namespace quota { | 20 namespace quota { |
| 21 | 21 |
| 22 class QuotaTaskObserver; | 22 class QuotaTaskObserver; |
| 23 class QuotaThreadTask; |
| 23 | 24 |
| 24 // A base class for quota tasks. | 25 // A base class for quota tasks. |
| 26 // TODO(kinuko): Revise this using base::Callback. |
| 25 class QuotaTask { | 27 class QuotaTask { |
| 26 public: | 28 public: |
| 27 virtual ~QuotaTask(); | 29 virtual ~QuotaTask(); |
| 28 void Start(); | 30 void Start(); |
| 29 | 31 |
| 30 protected: | 32 protected: |
| 31 explicit QuotaTask(QuotaTaskObserver* observer); | 33 explicit QuotaTask(QuotaTaskObserver* observer); |
| 32 | 34 |
| 33 // The task body. | 35 // The task body. |
| 34 virtual void Run() = 0; | 36 virtual void Run() = 0; |
| 35 | 37 |
| 36 // Called upon completion, on the original message loop. | 38 // Called upon completion, on the original message loop. |
| 37 virtual void Completed() = 0; | 39 virtual void Completed() = 0; |
| 38 | 40 |
| 39 // Called when the task is aborted. | 41 // Called when the task is aborted. |
| 40 virtual void Aborted() {} | 42 virtual void Aborted() {} |
| 41 | 43 |
| 42 void CallCompleted(); | 44 void CallCompleted(); |
| 43 | 45 |
| 44 // Call this to delete itself. | 46 // Call this to delete itself. |
| 45 void DeleteSoon(); | 47 void DeleteSoon(); |
| 46 | 48 |
| 47 QuotaTaskObserver* observer() const { return observer_; } | 49 QuotaTaskObserver* observer() const { return observer_; } |
| 48 scoped_refptr<base::MessageLoopProxy> original_message_loop() const { | 50 base::SingleThreadTaskRunner* original_task_runner() const { |
| 49 return original_message_loop_; | 51 return original_task_runner_; |
| 50 } | 52 } |
| 51 | 53 |
| 52 private: | 54 private: |
| 53 friend class QuotaTaskObserver; | 55 friend class QuotaTaskObserver; |
| 56 friend class QuotaThreadTask; |
| 54 void Abort(); | 57 void Abort(); |
| 55 QuotaTaskObserver* observer_; | 58 QuotaTaskObserver* observer_; |
| 56 scoped_refptr<base::MessageLoopProxy> original_message_loop_; | 59 scoped_refptr<base::SingleThreadTaskRunner> original_task_runner_; |
| 57 }; | 60 }; |
| 58 | 61 |
| 59 // For tasks that post tasks to the other thread. | 62 // For tasks that post tasks to the other thread. |
| 60 class QuotaThreadTask : public QuotaTask, | 63 class QuotaThreadTask : public QuotaTask, |
| 61 public base::RefCountedThreadSafe<QuotaThreadTask> { | 64 public base::RefCountedThreadSafe<QuotaThreadTask> { |
| 62 public: | 65 public: |
| 63 QuotaThreadTask(QuotaTaskObserver* observer, | 66 QuotaThreadTask(QuotaTaskObserver* observer, |
| 64 scoped_refptr<base::MessageLoopProxy> target_message_loop); | 67 base::TaskRunner* target_task_runner); |
| 65 | 68 |
| 66 protected: | 69 protected: |
| 67 virtual ~QuotaThreadTask(); | 70 virtual ~QuotaThreadTask(); |
| 68 | 71 |
| 69 // One of the following Run methods should be overridden for execution | 72 // One of the following Run methods should be overridden for execution |
| 70 // on the target thread. | 73 // on the target thread. |
| 71 | 74 |
| 72 // A task to invoke the CallCompleted() method on the original thread will | 75 // A task to invoke the CallCompleted() method on the original thread will |
| 73 // be scheduled immediately upon return from RunOnTargetThread(). | 76 // be scheduled immediately upon return from RunOnTargetThread(). |
| 74 virtual void RunOnTargetThread(); | 77 virtual void RunOnTargetThread(); |
| 75 | 78 |
| 76 // A task to invoke the CallCompleted() method on the original thread will | 79 // A task to invoke the CallCompleted() method on the original thread will |
| 77 // only be scheduled if RunOnTargetThreadAsync returns true. If false is | 80 // only be scheduled if RunOnTargetThreadAsync returns true. If false is |
| 78 // returned, the derived class should schedule a task to do so upon actual | 81 // returned, the derived class should schedule a task to do so upon actual |
| 79 // completion. | 82 // completion. |
| 80 virtual bool RunOnTargetThreadAsync(); | 83 virtual bool RunOnTargetThreadAsync(); |
| 81 | 84 |
| 82 virtual void Run() OVERRIDE; | 85 virtual void Run() OVERRIDE; |
| 83 scoped_refptr<base::MessageLoopProxy> target_message_loop() const { | 86 base::TaskRunner* target_task_runner() const { |
| 84 return target_message_loop_; | 87 return target_task_runner_; |
| 85 } | 88 } |
| 86 | 89 |
| 87 private: | 90 private: |
| 88 friend class base::RefCountedThreadSafe<QuotaThreadTask>; | 91 friend class base::RefCountedThreadSafe<QuotaThreadTask>; |
| 89 friend class QuotaTaskObserver; | 92 friend class QuotaTaskObserver; |
| 90 void CallRunOnTargetThread(); | 93 void CallRunOnTargetThread(); |
| 91 | 94 |
| 92 scoped_refptr<base::MessageLoopProxy> target_message_loop_; | 95 scoped_refptr<base::TaskRunner> target_task_runner_; |
| 93 }; | 96 }; |
| 94 | 97 |
| 95 class QuotaTaskObserver { | 98 class QuotaTaskObserver { |
| 96 public: | 99 public: |
| 97 virtual ~QuotaTaskObserver(); | 100 virtual ~QuotaTaskObserver(); |
| 98 | 101 |
| 99 protected: | 102 protected: |
| 100 friend class QuotaTask; | 103 friend class QuotaTask; |
| 101 friend class QuotaThreadTask; | 104 friend class QuotaThreadTask; |
| 102 | 105 |
| 103 QuotaTaskObserver(); | 106 QuotaTaskObserver(); |
| 104 void RegisterTask(QuotaTask* task); | 107 void RegisterTask(QuotaTask* task); |
| 105 void UnregisterTask(QuotaTask* task); | 108 void UnregisterTask(QuotaTask* task); |
| 106 | 109 |
| 107 typedef std::set<QuotaTask*> TaskSet; | 110 typedef std::set<QuotaTask*> TaskSet; |
| 108 TaskSet running_quota_tasks_; | 111 TaskSet running_quota_tasks_; |
| 109 }; | 112 }; |
| 110 } | 113 } |
| 111 | 114 |
| 112 #endif // WEBKIT_QUOTA_QUOTA_TASK_H_ | 115 #endif // WEBKIT_QUOTA_QUOTA_TASK_H_ |
| OLD | NEW |