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 183 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
194 | 194 |
195 PriorityQueue* single_threaded_priority_queue() { | 195 PriorityQueue* single_threaded_priority_queue() { |
196 return &single_threaded_priority_queue_; | 196 return &single_threaded_priority_queue_; |
197 } | 197 } |
198 | 198 |
199 // SchedulerWorker::Delegate: | 199 // SchedulerWorker::Delegate: |
200 void OnMainEntry(SchedulerWorker* worker) override; | 200 void OnMainEntry(SchedulerWorker* worker) override; |
201 scoped_refptr<Sequence> GetWork(SchedulerWorker* worker) override; | 201 scoped_refptr<Sequence> GetWork(SchedulerWorker* worker) override; |
202 void ReEnqueueSequence(scoped_refptr<Sequence> sequence) override; | 202 void ReEnqueueSequence(scoped_refptr<Sequence> sequence) override; |
203 TimeDelta GetSleepTimeout() override; | 203 TimeDelta GetSleepTimeout() override; |
| 204 bool CanDetach(SchedulerWorker* worker) override; |
204 | 205 |
205 private: | 206 private: |
206 SchedulerWorkerPoolImpl* outer_; | 207 SchedulerWorkerPoolImpl* outer_; |
207 const ReEnqueueSequenceCallback re_enqueue_sequence_callback_; | 208 const ReEnqueueSequenceCallback re_enqueue_sequence_callback_; |
208 | 209 |
209 // Single-threaded PriorityQueue for the worker. | 210 // Single-threaded PriorityQueue for the worker. |
210 PriorityQueue single_threaded_priority_queue_; | 211 PriorityQueue single_threaded_priority_queue_; |
211 | 212 |
212 // True if the last Sequence returned by GetWork() was extracted from | 213 // True if the last Sequence returned by GetWork() was extracted from |
213 // |single_threaded_priority_queue_|. | 214 // |single_threaded_priority_queue_|. |
(...skipping 257 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
471 // |sequence| must be enqueued. | 472 // |sequence| must be enqueued. |
472 re_enqueue_sequence_callback_.Run(std::move(sequence)); | 473 re_enqueue_sequence_callback_.Run(std::move(sequence)); |
473 } | 474 } |
474 } | 475 } |
475 | 476 |
476 TimeDelta SchedulerWorkerPoolImpl::SchedulerWorkerDelegateImpl:: | 477 TimeDelta SchedulerWorkerPoolImpl::SchedulerWorkerDelegateImpl:: |
477 GetSleepTimeout() { | 478 GetSleepTimeout() { |
478 return TimeDelta::Max(); | 479 return TimeDelta::Max(); |
479 } | 480 } |
480 | 481 |
| 482 bool SchedulerWorkerPoolImpl::SchedulerWorkerDelegateImpl::CanDetach( |
| 483 SchedulerWorker* worker) { |
| 484 return false; |
| 485 } |
| 486 |
481 SchedulerWorkerPoolImpl::SchedulerWorkerPoolImpl( | 487 SchedulerWorkerPoolImpl::SchedulerWorkerPoolImpl( |
482 StringPiece name, | 488 StringPiece name, |
483 IORestriction io_restriction, | 489 IORestriction io_restriction, |
484 TaskTracker* task_tracker, | 490 TaskTracker* task_tracker, |
485 DelayedTaskManager* delayed_task_manager) | 491 DelayedTaskManager* delayed_task_manager) |
486 : name_(name.as_string()), | 492 : name_(name.as_string()), |
487 io_restriction_(io_restriction), | 493 io_restriction_(io_restriction), |
488 idle_workers_stack_lock_(shared_priority_queue_.container_lock()), | 494 idle_workers_stack_lock_(shared_priority_queue_.container_lock()), |
489 idle_workers_stack_cv_for_testing_( | 495 idle_workers_stack_cv_for_testing_( |
490 idle_workers_stack_lock_.CreateConditionVariable()), | 496 idle_workers_stack_lock_.CreateConditionVariable()), |
(...skipping 16 matching lines...) Expand all Loading... |
507 AutoSchedulerLock auto_lock(idle_workers_stack_lock_); | 513 AutoSchedulerLock auto_lock(idle_workers_stack_lock_); |
508 | 514 |
509 DCHECK(workers_.empty()); | 515 DCHECK(workers_.empty()); |
510 | 516 |
511 for (size_t i = 0; i < max_threads; ++i) { | 517 for (size_t i = 0; i < max_threads; ++i) { |
512 std::unique_ptr<SchedulerWorker> worker = | 518 std::unique_ptr<SchedulerWorker> worker = |
513 SchedulerWorker::Create( | 519 SchedulerWorker::Create( |
514 thread_priority, WrapUnique(new SchedulerWorkerDelegateImpl( | 520 thread_priority, WrapUnique(new SchedulerWorkerDelegateImpl( |
515 this, re_enqueue_sequence_callback, | 521 this, re_enqueue_sequence_callback, |
516 &shared_priority_queue_, static_cast<int>(i))), | 522 &shared_priority_queue_, static_cast<int>(i))), |
517 task_tracker_); | 523 task_tracker_, |
| 524 SchedulerWorker::InitialState::ALIVE); |
518 if (!worker) | 525 if (!worker) |
519 break; | 526 break; |
520 idle_workers_stack_.Push(worker.get()); | 527 idle_workers_stack_.Push(worker.get()); |
521 workers_.push_back(std::move(worker)); | 528 workers_.push_back(std::move(worker)); |
522 } | 529 } |
523 | 530 |
524 #if DCHECK_IS_ON() | 531 #if DCHECK_IS_ON() |
525 workers_created_.Signal(); | 532 workers_created_.Signal(); |
526 #endif | 533 #endif |
527 | 534 |
(...skipping 21 matching lines...) Expand all Loading... |
549 } | 556 } |
550 | 557 |
551 void SchedulerWorkerPoolImpl::RemoveFromIdleWorkersStack( | 558 void SchedulerWorkerPoolImpl::RemoveFromIdleWorkersStack( |
552 SchedulerWorker* worker) { | 559 SchedulerWorker* worker) { |
553 AutoSchedulerLock auto_lock(idle_workers_stack_lock_); | 560 AutoSchedulerLock auto_lock(idle_workers_stack_lock_); |
554 idle_workers_stack_.Remove(worker); | 561 idle_workers_stack_.Remove(worker); |
555 } | 562 } |
556 | 563 |
557 } // namespace internal | 564 } // namespace internal |
558 } // namespace base | 565 } // namespace base |
OLD | NEW |