| 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 <memory> | 9 #include <memory> |
| 10 #include <unordered_set> | 10 #include <unordered_set> |
| 11 #include <vector> | 11 #include <vector> |
| 12 | 12 |
| 13 #include "base/bind.h" | 13 #include "base/bind.h" |
| 14 #include "base/bind_helpers.h" | 14 #include "base/bind_helpers.h" |
| 15 #include "base/callback.h" | 15 #include "base/callback.h" |
| 16 #include "base/macros.h" | 16 #include "base/macros.h" |
| 17 #include "base/memory/ptr_util.h" | 17 #include "base/memory/ptr_util.h" |
| 18 #include "base/memory/ref_counted.h" | 18 #include "base/memory/ref_counted.h" |
| 19 #include "base/synchronization/condition_variable.h" | 19 #include "base/synchronization/condition_variable.h" |
| 20 #include "base/synchronization/lock.h" | 20 #include "base/synchronization/lock.h" |
| 21 #include "base/synchronization/waitable_event.h" | 21 #include "base/synchronization/waitable_event.h" |
| 22 #include "base/task_runner.h" | 22 #include "base/task_runner.h" |
| 23 #include "base/task_scheduler/delayed_task_manager.h" | 23 #include "base/task_scheduler/delayed_task_manager.h" |
| 24 #include "base/task_scheduler/scheduler_worker_pool_params.h" |
| 24 #include "base/task_scheduler/sequence.h" | 25 #include "base/task_scheduler/sequence.h" |
| 25 #include "base/task_scheduler/sequence_sort_key.h" | 26 #include "base/task_scheduler/sequence_sort_key.h" |
| 26 #include "base/task_scheduler/task_tracker.h" | 27 #include "base/task_scheduler/task_tracker.h" |
| 27 #include "base/task_scheduler/test_task_factory.h" | 28 #include "base/task_scheduler/test_task_factory.h" |
| 28 #include "base/task_scheduler/test_utils.h" | 29 #include "base/task_scheduler/test_utils.h" |
| 29 #include "base/threading/platform_thread.h" | 30 #include "base/threading/platform_thread.h" |
| 30 #include "base/threading/simple_thread.h" | 31 #include "base/threading/simple_thread.h" |
| 31 #include "base/threading/thread_restrictions.h" | 32 #include "base/threading/thread_restrictions.h" |
| 32 #include "testing/gtest/include/gtest/gtest.h" | 33 #include "testing/gtest/include/gtest/gtest.h" |
| 33 | 34 |
| 34 namespace base { | 35 namespace base { |
| 35 namespace internal { | 36 namespace internal { |
| 36 namespace { | 37 namespace { |
| 37 | 38 |
| 38 const size_t kNumWorkersInWorkerPool = 4; | 39 const size_t kNumWorkersInWorkerPool = 4; |
| 39 const size_t kNumThreadsPostingTasks = 4; | 40 const size_t kNumThreadsPostingTasks = 4; |
| 40 const size_t kNumTasksPostedPerThread = 150; | 41 const size_t kNumTasksPostedPerThread = 150; |
| 41 | 42 |
| 42 using IORestriction = SchedulerWorkerPoolImpl::IORestriction; | 43 using IORestriction = SchedulerWorkerPoolParams::IORestriction; |
| 43 | 44 |
| 44 class TestDelayedTaskManager : public DelayedTaskManager { | 45 class TestDelayedTaskManager : public DelayedTaskManager { |
| 45 public: | 46 public: |
| 46 TestDelayedTaskManager() : DelayedTaskManager(Bind(&DoNothing)) {} | 47 TestDelayedTaskManager() : DelayedTaskManager(Bind(&DoNothing)) {} |
| 47 | 48 |
| 48 void SetCurrentTime(TimeTicks now) { now_ = now; } | 49 void SetCurrentTime(TimeTicks now) { now_ = now; } |
| 49 | 50 |
| 50 // DelayedTaskManager: | 51 // DelayedTaskManager: |
| 51 TimeTicks Now() const override { return now_; } | 52 TimeTicks Now() const override { return now_; } |
| 52 | 53 |
| 53 private: | 54 private: |
| 54 TimeTicks now_ = TimeTicks::Now(); | 55 TimeTicks now_ = TimeTicks::Now(); |
| 55 | 56 |
| 56 DISALLOW_COPY_AND_ASSIGN(TestDelayedTaskManager); | 57 DISALLOW_COPY_AND_ASSIGN(TestDelayedTaskManager); |
| 57 }; | 58 }; |
| 58 | 59 |
| 59 class TaskSchedulerWorkerPoolImplTest | 60 class TaskSchedulerWorkerPoolImplTest |
| 60 : public testing::TestWithParam<ExecutionMode> { | 61 : public testing::TestWithParam<ExecutionMode> { |
| 61 protected: | 62 protected: |
| 62 TaskSchedulerWorkerPoolImplTest() = default; | 63 TaskSchedulerWorkerPoolImplTest() = default; |
| 63 | 64 |
| 64 void SetUp() override { | 65 void SetUp() override { |
| 65 worker_pool_ = SchedulerWorkerPoolImpl::Create( | 66 worker_pool_ = SchedulerWorkerPoolImpl::Create( |
| 66 "TestWorkerPoolWithFileIO", ThreadPriority::NORMAL, | 67 SchedulerWorkerPoolParams("TestWorkerPoolWithFileIO", |
| 67 kNumWorkersInWorkerPool, IORestriction::ALLOWED, | 68 ThreadPriority::NORMAL, |
| 69 IORestriction::ALLOWED, |
| 70 kNumWorkersInWorkerPool), |
| 68 Bind(&TaskSchedulerWorkerPoolImplTest::ReEnqueueSequenceCallback, | 71 Bind(&TaskSchedulerWorkerPoolImplTest::ReEnqueueSequenceCallback, |
| 69 Unretained(this)), | 72 Unretained(this)), |
| 70 &task_tracker_, &delayed_task_manager_); | 73 &task_tracker_, &delayed_task_manager_); |
| 71 ASSERT_TRUE(worker_pool_); | 74 ASSERT_TRUE(worker_pool_); |
| 72 } | 75 } |
| 73 | 76 |
| 74 void TearDown() override { | 77 void TearDown() override { |
| 75 worker_pool_->WaitForAllWorkersIdleForTesting(); | 78 worker_pool_->WaitForAllWorkersIdleForTesting(); |
| 76 worker_pool_->JoinForTesting(); | 79 worker_pool_->JoinForTesting(); |
| 77 } | 80 } |
| (...skipping 278 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 356 DISALLOW_COPY_AND_ASSIGN(TaskSchedulerWorkerPoolImplIORestrictionTest); | 359 DISALLOW_COPY_AND_ASSIGN(TaskSchedulerWorkerPoolImplIORestrictionTest); |
| 357 }; | 360 }; |
| 358 | 361 |
| 359 } // namespace | 362 } // namespace |
| 360 | 363 |
| 361 TEST_P(TaskSchedulerWorkerPoolImplIORestrictionTest, IORestriction) { | 364 TEST_P(TaskSchedulerWorkerPoolImplIORestrictionTest, IORestriction) { |
| 362 TaskTracker task_tracker; | 365 TaskTracker task_tracker; |
| 363 DelayedTaskManager delayed_task_manager(Bind(&DoNothing)); | 366 DelayedTaskManager delayed_task_manager(Bind(&DoNothing)); |
| 364 | 367 |
| 365 auto worker_pool = SchedulerWorkerPoolImpl::Create( | 368 auto worker_pool = SchedulerWorkerPoolImpl::Create( |
| 366 "TestWorkerPoolWithParam", ThreadPriority::NORMAL, 1U, GetParam(), | 369 SchedulerWorkerPoolParams("TestWorkerPoolWithParam", |
| 370 ThreadPriority::NORMAL, GetParam(), 1U), |
| 367 Bind(&NotReachedReEnqueueSequenceCallback), &task_tracker, | 371 Bind(&NotReachedReEnqueueSequenceCallback), &task_tracker, |
| 368 &delayed_task_manager); | 372 &delayed_task_manager); |
| 369 ASSERT_TRUE(worker_pool); | 373 ASSERT_TRUE(worker_pool); |
| 370 | 374 |
| 371 WaitableEvent task_ran(WaitableEvent::ResetPolicy::MANUAL, | 375 WaitableEvent task_ran(WaitableEvent::ResetPolicy::MANUAL, |
| 372 WaitableEvent::InitialState::NOT_SIGNALED); | 376 WaitableEvent::InitialState::NOT_SIGNALED); |
| 373 worker_pool->CreateTaskRunnerWithTraits(TaskTraits(), ExecutionMode::PARALLEL) | 377 worker_pool->CreateTaskRunnerWithTraits(TaskTraits(), ExecutionMode::PARALLEL) |
| 374 ->PostTask(FROM_HERE, Bind(&ExpectIORestriction, GetParam(), &task_ran)); | 378 ->PostTask(FROM_HERE, Bind(&ExpectIORestriction, GetParam(), &task_ran)); |
| 375 task_ran.Wait(); | 379 task_ran.Wait(); |
| 376 | 380 |
| 377 worker_pool->JoinForTesting(); | 381 worker_pool->JoinForTesting(); |
| 378 } | 382 } |
| 379 | 383 |
| 380 INSTANTIATE_TEST_CASE_P(IOAllowed, | 384 INSTANTIATE_TEST_CASE_P(IOAllowed, |
| 381 TaskSchedulerWorkerPoolImplIORestrictionTest, | 385 TaskSchedulerWorkerPoolImplIORestrictionTest, |
| 382 ::testing::Values(IORestriction::ALLOWED)); | 386 ::testing::Values(IORestriction::ALLOWED)); |
| 383 INSTANTIATE_TEST_CASE_P(IODisallowed, | 387 INSTANTIATE_TEST_CASE_P(IODisallowed, |
| 384 TaskSchedulerWorkerPoolImplIORestrictionTest, | 388 TaskSchedulerWorkerPoolImplIORestrictionTest, |
| 385 ::testing::Values(IORestriction::DISALLOWED)); | 389 ::testing::Values(IORestriction::DISALLOWED)); |
| 386 | 390 |
| 387 } // namespace internal | 391 } // namespace internal |
| 388 } // namespace base | 392 } // namespace base |
| OLD | NEW |