OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 #include "base/task_scheduler/scheduler_worker_pool_impl.h" | 5 #include "base/task_scheduler/scheduler_worker_pool_impl.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 | 8 |
9 #include <algorithm> | 9 #include <algorithm> |
10 #include <utility> | 10 #include <utility> |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
51 // long as |worker_pool| is alive. | 51 // long as |worker_pool| is alive. |
52 // TODO(robliao): Find a concrete way to manage |worker_pool|'s memory. | 52 // TODO(robliao): Find a concrete way to manage |worker_pool|'s memory. |
53 SchedulerParallelTaskRunner(const TaskTraits& traits, | 53 SchedulerParallelTaskRunner(const TaskTraits& traits, |
54 SchedulerWorkerPool* worker_pool) | 54 SchedulerWorkerPool* worker_pool) |
55 : traits_(traits), worker_pool_(worker_pool) { | 55 : traits_(traits), worker_pool_(worker_pool) { |
56 DCHECK(worker_pool_); | 56 DCHECK(worker_pool_); |
57 } | 57 } |
58 | 58 |
59 // TaskRunner: | 59 // TaskRunner: |
60 bool PostDelayedTask(const tracked_objects::Location& from_here, | 60 bool PostDelayedTask(const tracked_objects::Location& from_here, |
61 const Closure& closure, | 61 Closure closure, |
62 TimeDelta delay) override { | 62 TimeDelta delay) override { |
63 // Post the task as part of a one-off single-task Sequence. | 63 // Post the task as part of a one-off single-task Sequence. |
64 return worker_pool_->PostTaskWithSequence( | 64 return worker_pool_->PostTaskWithSequence( |
65 MakeUnique<Task>(from_here, closure, traits_, delay), | 65 MakeUnique<Task>(from_here, std::move(closure), traits_, delay), |
66 make_scoped_refptr(new Sequence)); | 66 make_scoped_refptr(new Sequence)); |
67 } | 67 } |
68 | 68 |
69 bool RunsTasksOnCurrentThread() const override { | 69 bool RunsTasksOnCurrentThread() const override { |
70 return tls_current_worker_pool.Get().Get() == worker_pool_; | 70 return tls_current_worker_pool.Get().Get() == worker_pool_; |
71 } | 71 } |
72 | 72 |
73 private: | 73 private: |
74 ~SchedulerParallelTaskRunner() override = default; | 74 ~SchedulerParallelTaskRunner() override = default; |
75 | 75 |
(...skipping 10 matching lines...) Expand all Loading... |
86 // so long as |worker_pool| is alive. | 86 // so long as |worker_pool| is alive. |
87 // TODO(robliao): Find a concrete way to manage |worker_pool|'s memory. | 87 // TODO(robliao): Find a concrete way to manage |worker_pool|'s memory. |
88 SchedulerSequencedTaskRunner(const TaskTraits& traits, | 88 SchedulerSequencedTaskRunner(const TaskTraits& traits, |
89 SchedulerWorkerPool* worker_pool) | 89 SchedulerWorkerPool* worker_pool) |
90 : traits_(traits), worker_pool_(worker_pool) { | 90 : traits_(traits), worker_pool_(worker_pool) { |
91 DCHECK(worker_pool_); | 91 DCHECK(worker_pool_); |
92 } | 92 } |
93 | 93 |
94 // SequencedTaskRunner: | 94 // SequencedTaskRunner: |
95 bool PostDelayedTask(const tracked_objects::Location& from_here, | 95 bool PostDelayedTask(const tracked_objects::Location& from_here, |
96 const Closure& closure, | 96 Closure closure, |
97 TimeDelta delay) override { | 97 TimeDelta delay) override { |
98 std::unique_ptr<Task> task(new Task(from_here, closure, traits_, delay)); | 98 std::unique_ptr<Task> task( |
| 99 new Task(from_here, std::move(closure), traits_, delay)); |
99 task->sequenced_task_runner_ref = this; | 100 task->sequenced_task_runner_ref = this; |
100 | 101 |
101 // Post the task as part of |sequence_|. | 102 // Post the task as part of |sequence_|. |
102 return worker_pool_->PostTaskWithSequence(std::move(task), sequence_); | 103 return worker_pool_->PostTaskWithSequence(std::move(task), sequence_); |
103 } | 104 } |
104 | 105 |
105 bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here, | 106 bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here, |
106 const Closure& closure, | 107 Closure closure, |
107 base::TimeDelta delay) override { | 108 base::TimeDelta delay) override { |
108 // Tasks are never nested within the task scheduler. | 109 // Tasks are never nested within the task scheduler. |
109 return PostDelayedTask(from_here, closure, delay); | 110 return PostDelayedTask(from_here, std::move(closure), delay); |
110 } | 111 } |
111 | 112 |
112 bool RunsTasksOnCurrentThread() const override { | 113 bool RunsTasksOnCurrentThread() const override { |
113 // TODO(fdoray): Rename TaskRunner::RunsTaskOnCurrentThread() to something | 114 // TODO(fdoray): Rename TaskRunner::RunsTaskOnCurrentThread() to something |
114 // that reflects this behavior more accurately. crbug.com/646905 | 115 // that reflects this behavior more accurately. crbug.com/646905 |
115 return sequence_->token() == SequenceToken::GetForCurrentThread(); | 116 return sequence_->token() == SequenceToken::GetForCurrentThread(); |
116 } | 117 } |
117 | 118 |
118 private: | 119 private: |
119 ~SchedulerSequencedTaskRunner() override = default; | 120 ~SchedulerSequencedTaskRunner() override = default; |
(...skipping 465 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
585 AutoSchedulerLock auto_lock(idle_workers_stack_lock_); | 586 AutoSchedulerLock auto_lock(idle_workers_stack_lock_); |
586 idle_workers_stack_.Remove(worker); | 587 idle_workers_stack_.Remove(worker); |
587 } | 588 } |
588 | 589 |
589 bool SchedulerWorkerPoolImpl::CanWorkerDetachForTesting() { | 590 bool SchedulerWorkerPoolImpl::CanWorkerDetachForTesting() { |
590 return !worker_detachment_disallowed_.IsSet(); | 591 return !worker_detachment_disallowed_.IsSet(); |
591 } | 592 } |
592 | 593 |
593 } // namespace internal | 594 } // namespace internal |
594 } // namespace base | 595 } // namespace base |
OLD | NEW |