| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 COMPONENTS_SCHEDULER_TEST_LAZY_SCHEDULER_MESSAGE_LOOP_DELEGATE_FOR_TESTS
_H_ | 5 #ifndef COMPONENTS_SCHEDULER_TEST_LAZY_SCHEDULER_MESSAGE_LOOP_DELEGATE_FOR_TESTS
_H_ |
| 6 #define COMPONENTS_SCHEDULER_TEST_LAZY_SCHEDULER_MESSAGE_LOOP_DELEGATE_FOR_TESTS
_H_ | 6 #define COMPONENTS_SCHEDULER_TEST_LAZY_SCHEDULER_MESSAGE_LOOP_DELEGATE_FOR_TESTS
_H_ |
| 7 | 7 |
| 8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
| 9 #include "components/scheduler/child/scheduler_task_runner_delegate.h" | 9 #include "components/scheduler/child/nestable_single_thread_task_runner.h" |
| 10 | 10 |
| 11 namespace scheduler { | 11 namespace scheduler { |
| 12 | 12 |
| 13 // This class connects the scheduler to a MessageLoop, but unlike | 13 // This class connects the scheduler to a MessageLoop, but unlike |
| 14 // SchedulerMessageLoopDelegate it allows the message loop to be created lazily | 14 // SchedulerMessageLoopDelegate it allows the message loop to be created lazily |
| 15 // after the scheduler has been brought up. This is needed in testing scenarios | 15 // after the scheduler has been brought up. This is needed in testing scenarios |
| 16 // where Blink is initialized before a MessageLoop has been created. | 16 // where Blink is initialized before a MessageLoop has been created. |
| 17 // | 17 // |
| 18 // TODO(skyostil): Fix the relevant test suites and remove this class | 18 // TODO(skyostil): Fix the relevant test suites and remove this class |
| 19 // (crbug.com/495659). | 19 // (crbug.com/495659). |
| 20 class LazySchedulerMessageLoopDelegateForTests | 20 class LazySchedulerMessageLoopDelegateForTests |
| 21 : public SchedulerTaskRunnerDelegate { | 21 : public NestableSingleThreadTaskRunner { |
| 22 public: | 22 public: |
| 23 static scoped_refptr<LazySchedulerMessageLoopDelegateForTests> Create(); | 23 static scoped_refptr<LazySchedulerMessageLoopDelegateForTests> Create(); |
| 24 | 24 |
| 25 // SchedulerTaskRunnerDelegate implementation | 25 // NestableSingleThreadTaskRunner implementation |
| 26 void SetDefaultTaskRunner( | |
| 27 scoped_refptr<base::SingleThreadTaskRunner> task_runner) override; | |
| 28 void RestoreDefaultTaskRunner() override; | |
| 29 bool PostDelayedTask(const tracked_objects::Location& from_here, | 26 bool PostDelayedTask(const tracked_objects::Location& from_here, |
| 30 const base::Closure& task, | 27 const base::Closure& task, |
| 31 base::TimeDelta delay) override; | 28 base::TimeDelta delay) override; |
| 32 bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here, | 29 bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here, |
| 33 const base::Closure& task, | 30 const base::Closure& task, |
| 34 base::TimeDelta delay) override; | 31 base::TimeDelta delay) override; |
| 35 bool RunsTasksOnCurrentThread() const override; | 32 bool RunsTasksOnCurrentThread() const override; |
| 36 bool IsNested() const override; | 33 bool IsNested() const override; |
| 37 void AddTaskObserver(base::MessageLoop::TaskObserver* task_observer) override; | 34 void AddTaskObserver(base::MessageLoop::TaskObserver* task_observer) override; |
| 38 void RemoveTaskObserver( | 35 void RemoveTaskObserver( |
| 39 base::MessageLoop::TaskObserver* task_observer) override; | 36 base::MessageLoop::TaskObserver* task_observer) override; |
| 40 | 37 |
| 41 private: | 38 private: |
| 42 LazySchedulerMessageLoopDelegateForTests(); | 39 LazySchedulerMessageLoopDelegateForTests(); |
| 43 ~LazySchedulerMessageLoopDelegateForTests() override; | 40 ~LazySchedulerMessageLoopDelegateForTests() override; |
| 44 | 41 |
| 45 bool HasMessageLoop() const; | 42 bool HasMessageLoop() const; |
| 46 base::MessageLoop* EnsureMessageLoop() const; | 43 base::MessageLoop* EnsureMessageLoop() const; |
| 47 | 44 |
| 48 mutable base::MessageLoop* message_loop_; | 45 mutable base::MessageLoop* message_loop_; |
| 49 base::PlatformThreadId thread_id_; | 46 base::PlatformThreadId thread_id_; |
| 50 | 47 |
| 51 // Task observers which have not yet been registered to a message loop. Not | 48 // Task observers which have not yet been registered to a message loop. Not |
| 52 // owned. | 49 // owned. |
| 53 mutable base::hash_set<base::MessageLoop::TaskObserver*> pending_observers_; | 50 mutable base::hash_set<base::MessageLoop::TaskObserver*> pending_observers_; |
| 54 | 51 |
| 55 // A task runner which hasn't yet been overridden in the message loop. | |
| 56 mutable scoped_refptr<base::SingleThreadTaskRunner> pending_task_runner_; | |
| 57 mutable scoped_refptr<base::SingleThreadTaskRunner> original_task_runner_; | |
| 58 | |
| 59 DISALLOW_COPY_AND_ASSIGN(LazySchedulerMessageLoopDelegateForTests); | 52 DISALLOW_COPY_AND_ASSIGN(LazySchedulerMessageLoopDelegateForTests); |
| 60 }; | 53 }; |
| 61 | 54 |
| 62 } // namespace scheduler | 55 } // namespace scheduler |
| 63 | 56 |
| 64 #endif // COMPONENTS_SCHEDULER_TEST_LAZY_SCHEDULER_MESSAGE_LOOP_DELEGATE_FOR_TE
STS_H_ | 57 #endif // COMPONENTS_SCHEDULER_TEST_LAZY_SCHEDULER_MESSAGE_LOOP_DELEGATE_FOR_TE
STS_H_ |
| OLD | NEW |