OLD | NEW |
1 // Copyright (c) 2012 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 | 7 |
8 #include <set> | 8 #include <set> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
11 #include "base/compiler_specific.h" | 11 #include "base/compiler_specific.h" |
12 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
13 #include "base/sequenced_task_runner_helpers.h" | 13 #include "base/sequenced_task_runner_helpers.h" |
14 | 14 |
15 namespace base { | 15 namespace base { |
16 class SingleThreadTaskRunner; | 16 class SingleThreadTaskRunner; |
17 class TaskRunner; | 17 class TaskRunner; |
18 } | 18 } |
19 | 19 |
20 namespace quota { | 20 namespace quota { |
21 | 21 |
22 class QuotaTaskObserver; | 22 class QuotaTaskObserver; |
23 class QuotaThreadTask; | |
24 | 23 |
25 // A base class for quota tasks. | 24 // A base class for quota tasks. |
26 // TODO(kinuko): Revise this using base::Callback. | 25 // TODO(kinuko): Revise this using base::Callback. |
27 class QuotaTask { | 26 class QuotaTask { |
28 public: | 27 public: |
29 void Start(); | 28 void Start(); |
30 | 29 |
31 protected: | 30 protected: |
32 explicit QuotaTask(QuotaTaskObserver* observer); | 31 explicit QuotaTask(QuotaTaskObserver* observer); |
33 virtual ~QuotaTask(); | 32 virtual ~QuotaTask(); |
(...skipping 13 matching lines...) Expand all Loading... |
47 void DeleteSoon(); | 46 void DeleteSoon(); |
48 | 47 |
49 QuotaTaskObserver* observer() const { return observer_; } | 48 QuotaTaskObserver* observer() const { return observer_; } |
50 base::SingleThreadTaskRunner* original_task_runner() const { | 49 base::SingleThreadTaskRunner* original_task_runner() const { |
51 return original_task_runner_; | 50 return original_task_runner_; |
52 } | 51 } |
53 | 52 |
54 private: | 53 private: |
55 friend class base::DeleteHelper<QuotaTask>; | 54 friend class base::DeleteHelper<QuotaTask>; |
56 friend class QuotaTaskObserver; | 55 friend class QuotaTaskObserver; |
57 friend class QuotaThreadTask; | |
58 | 56 |
59 void Abort(); | 57 void Abort(); |
60 QuotaTaskObserver* observer_; | 58 QuotaTaskObserver* observer_; |
61 scoped_refptr<base::SingleThreadTaskRunner> original_task_runner_; | 59 scoped_refptr<base::SingleThreadTaskRunner> original_task_runner_; |
62 bool delete_scheduled_; | 60 bool delete_scheduled_; |
63 }; | 61 }; |
64 | 62 |
65 // For tasks that post tasks to the other thread. | |
66 class QuotaThreadTask : public QuotaTask, | |
67 public base::RefCountedThreadSafe<QuotaThreadTask> { | |
68 public: | |
69 QuotaThreadTask(QuotaTaskObserver* observer, | |
70 base::TaskRunner* target_task_runner); | |
71 | |
72 protected: | |
73 virtual ~QuotaThreadTask(); | |
74 | |
75 // One of the following Run methods should be overridden for execution | |
76 // on the target thread. | |
77 | |
78 // A task to invoke the CallCompleted() method on the original thread will | |
79 // be scheduled immediately upon return from RunOnTargetThread(). | |
80 virtual void RunOnTargetThread(); | |
81 | |
82 // A task to invoke the CallCompleted() method on the original thread will | |
83 // only be scheduled if RunOnTargetThreadAsync returns true. If false is | |
84 // returned, the derived class should schedule a task to do so upon actual | |
85 // completion. | |
86 virtual bool RunOnTargetThreadAsync(); | |
87 | |
88 virtual void Run() OVERRIDE; | |
89 base::TaskRunner* target_task_runner() const { | |
90 return target_task_runner_; | |
91 } | |
92 | |
93 private: | |
94 friend class base::RefCountedThreadSafe<QuotaThreadTask>; | |
95 friend class QuotaTaskObserver; | |
96 void CallRunOnTargetThread(); | |
97 | |
98 scoped_refptr<base::TaskRunner> target_task_runner_; | |
99 }; | |
100 | |
101 class QuotaTaskObserver { | 63 class QuotaTaskObserver { |
102 protected: | 64 protected: |
103 friend class QuotaTask; | 65 friend class QuotaTask; |
104 friend class QuotaThreadTask; | |
105 | 66 |
106 QuotaTaskObserver(); | 67 QuotaTaskObserver(); |
107 virtual ~QuotaTaskObserver(); | 68 virtual ~QuotaTaskObserver(); |
108 | 69 |
109 void RegisterTask(QuotaTask* task); | 70 void RegisterTask(QuotaTask* task); |
110 void UnregisterTask(QuotaTask* task); | 71 void UnregisterTask(QuotaTask* task); |
111 | 72 |
112 typedef std::set<QuotaTask*> TaskSet; | 73 typedef std::set<QuotaTask*> TaskSet; |
113 TaskSet running_quota_tasks_; | 74 TaskSet running_quota_tasks_; |
114 }; | 75 }; |
115 } | 76 } |
116 | 77 |
117 #endif // WEBKIT_QUOTA_QUOTA_TASK_H_ | 78 #endif // WEBKIT_QUOTA_QUOTA_TASK_H_ |
OLD | NEW |