| Index: components/scheduler/child/task_queue_manager_perftest.cc
|
| diff --git a/components/scheduler/child/task_queue_manager_perftest.cc b/components/scheduler/child/task_queue_manager_perftest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..f9d23f327ea2c7abc19335e82b5ebe0ad00e64de
|
| --- /dev/null
|
| +++ b/components/scheduler/child/task_queue_manager_perftest.cc
|
| @@ -0,0 +1,141 @@
|
| +// Copyright 2015 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#include "components/scheduler/child/task_queue_manager.h"
|
| +
|
| +#include "base/bind.h"
|
| +#include "base/threading/thread.h"
|
| +#include "components/scheduler/child/scheduler_task_runner_delegate_impl.h"
|
| +#include "components/scheduler/child/task_queue_impl.h"
|
| +#include "components/scheduler/child/task_queue_selector.h"
|
| +#include "components/scheduler/child/task_queue_sets.h"
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
| +#include "testing/perf/perf_test.h"
|
| +
|
| +namespace scheduler {
|
| +
|
| +class TaskQueueManagerPerfTest : public testing::Test {
|
| + public:
|
| + TaskQueueManagerPerfTest()
|
| + : num_queues_(0),
|
| + max_tasks_in_flight_(0),
|
| + num_tasks_in_flight_(0),
|
| + num_tasks_to_post_(0),
|
| + num_tasks_to_run_(0) {}
|
| +
|
| + void Initialize(size_t num_queues) {
|
| + num_queues_ = num_queues;
|
| + message_loop_.reset(new base::MessageLoop());
|
| + manager_ = make_scoped_ptr(new TaskQueueManager(
|
| + SchedulerTaskRunnerDelegateImpl::Create(message_loop_.get()),
|
| + "fake.category", "fake.category.debug"));
|
| + for (size_t i = 0; i < num_queues; i++)
|
| + queues_.push_back(manager_->NewTaskQueue(TaskQueue::Spec("test")));
|
| + }
|
| +
|
| + void TestDelayedTask() {
|
| + if (--num_tasks_to_run_ == 0) {
|
| + message_loop_->Quit();
|
| + }
|
| +
|
| + num_tasks_in_flight_--;
|
| + // NOTE there are only up to max_tasks_in_flight_ pending delayed tasks at
|
| + // any one time. Thanks to the lower_num_tasks_to_post going to zero if
|
| + // there are a lot of tasks in flight, the total number of task in flight at
|
| + // any one time is very variable.
|
| + unsigned int lower_num_tasks_to_post =
|
| + num_tasks_in_flight_ < (max_tasks_in_flight_ / 2) ? 1 : 0;
|
| + unsigned int max_tasks_to_post =
|
| + num_tasks_to_post_ % 2 ? lower_num_tasks_to_post : 10;
|
| + for (unsigned int i = 0;
|
| + i < max_tasks_to_post && num_tasks_in_flight_ < max_tasks_in_flight_ &&
|
| + num_tasks_to_post_ > 0;
|
| + i++) {
|
| + // Choose a queue weighted towards queue 0.
|
| + unsigned int queue = num_tasks_to_post_ % (num_queues_ + 1);
|
| + if (queue == num_queues_) {
|
| + queue = 0;
|
| + }
|
| + // Simulate a mix of short and longer delays.
|
| + unsigned int delay =
|
| + num_tasks_to_post_ % 2 ? 1 : (10 + num_tasks_to_post_ % 10);
|
| + queues_[queue]->PostDelayedTask(
|
| + FROM_HERE, base::Bind(&TaskQueueManagerPerfTest::TestDelayedTask,
|
| + base::Unretained(this)),
|
| + base::TimeDelta::FromMicroseconds(delay));
|
| + num_tasks_in_flight_++;
|
| + num_tasks_to_post_--;
|
| + }
|
| + }
|
| +
|
| + void ResetAndCallTestDelayedTask(unsigned int num_tasks_to_run) {
|
| + num_tasks_in_flight_ = 1;
|
| + num_tasks_to_post_ = num_tasks_to_run;
|
| + num_tasks_to_run_ = num_tasks_to_run;
|
| + TestDelayedTask();
|
| + }
|
| +
|
| + void Benchmark(const std::string& trace, const base::Closure& test_task) {
|
| + base::ThreadTicks start = base::ThreadTicks::Now();
|
| + base::ThreadTicks now;
|
| + unsigned long long num_iterations = 0;
|
| + do {
|
| + test_task.Run();
|
| + message_loop_->Run();
|
| + now = base::ThreadTicks::Now();
|
| + num_iterations++;
|
| + } while (now - start < base::TimeDelta::FromSeconds(5));
|
| + perf_test::PrintResult(
|
| + "task", "", trace,
|
| + (now - start).InMicroseconds() / static_cast<double>(num_iterations),
|
| + "us/run", true);
|
| + }
|
| +
|
| + size_t num_queues_;
|
| + unsigned int max_tasks_in_flight_;
|
| + unsigned int num_tasks_in_flight_;
|
| + unsigned int num_tasks_to_post_;
|
| + unsigned int num_tasks_to_run_;
|
| + scoped_ptr<TaskQueueManager> manager_;
|
| + scoped_ptr<base::MessageLoop> message_loop_;
|
| + std::vector<scoped_refptr<base::SingleThreadTaskRunner>> queues_;
|
| +};
|
| +
|
| +TEST_F(TaskQueueManagerPerfTest, RunTenThousandDelayedTasks_OneQueue) {
|
| + if (!base::ThreadTicks::IsSupported())
|
| + return;
|
| + Initialize(1u);
|
| +
|
| + max_tasks_in_flight_ = 200;
|
| + Benchmark("run 10000 delayed tasks with one queue",
|
| + base::Bind(&TaskQueueManagerPerfTest::ResetAndCallTestDelayedTask,
|
| + base::Unretained(this), 10000));
|
| +}
|
| +
|
| +TEST_F(TaskQueueManagerPerfTest, RunTenThousandDelayedTasks_FourQueues) {
|
| + if (!base::ThreadTicks::IsSupported())
|
| + return;
|
| + Initialize(4u);
|
| +
|
| + max_tasks_in_flight_ = 200;
|
| + Benchmark("run 10000 delayed tasks with four queues",
|
| + base::Bind(&TaskQueueManagerPerfTest::ResetAndCallTestDelayedTask,
|
| + base::Unretained(this), 10000));
|
| +}
|
| +
|
| +TEST_F(TaskQueueManagerPerfTest, RunTenThousandDelayedTasks_EightQueues) {
|
| + if (!base::ThreadTicks::IsSupported())
|
| + return;
|
| + Initialize(8u);
|
| +
|
| + max_tasks_in_flight_ = 200;
|
| + Benchmark("run 10000 delayed tasks with eight queues",
|
| + base::Bind(&TaskQueueManagerPerfTest::ResetAndCallTestDelayedTask,
|
| + base::Unretained(this), 10000));
|
| +}
|
| +
|
| +// TODO(alexclarke): Add additional tests with different mixes of non-delayed vs
|
| +// delayed tasks.
|
| +
|
| +} // namespace scheduler
|
|
|