Task manager should support Idle Wakeups on Windows

chrome/browser/task_management/sampling/task_manager_impl.cc

 // 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 "chrome/browser/task_management/sampling/task_manager_impl.h"
 
 #include <algorithm>
 #include <string>
 #include <unordered_map>
 #include <unordered_set>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/containers/adapters.h"
 #include "base/stl_util.h"
 #include "build/build_config.h"
 #include "chrome/browser/task_management/providers/browser_process_task_provider.h"
 #include "chrome/browser/task_management/providers/child_process_task_provider.h"
 #include "chrome/browser/task_management/providers/web_contents/web_contents_task_provider.h"
+#include "chrome/browser/task_management/sampling/shared_sampler.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/gpu_data_manager.h"
 
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/task_management/providers/arc/arc_process_task_provider.h"
 #include "components/arc/arc_bridge_service.h"
 #endif  // defined(OS_CHROMEOS)
 
 namespace task_management {
 
 namespace {
 
 scoped_refptr<base::SequencedTaskRunner> GetBlockingPoolRunner() {
   base::SequencedWorkerPool* blocking_pool =
       content::BrowserThread::GetBlockingPool();
   return blocking_pool->GetSequencedTaskRunner(
       blocking_pool->GetSequenceToken());
 }
 
 base::LazyInstance<TaskManagerImpl> lazy_task_manager_instance =
     LAZY_INSTANCE_INITIALIZER;
 
 }  // namespace
 
 TaskManagerImpl::TaskManagerImpl()
-    : on_background_data_ready_callback_(base::Bind(
-          &TaskManagerImpl::OnTaskGroupBackgroundCalculationsDone,
-          base::Unretained(this))),
+    : on_background_data_ready_callback_(
+          base::Bind(&TaskManagerImpl::OnTaskGroupBackgroundCalculationsDone,
+                     base::Unretained(this))),
       blocking_pool_runner_(GetBlockingPoolRunner()),
+      shared_sampler_(new SharedSampler(blocking_pool_runner_)),
       is_running_(false) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 
   task_providers_.push_back(new BrowserProcessTaskProvider());
   task_providers_.push_back(new ChildProcessTaskProvider());
   task_providers_.push_back(new WebContentsTaskProvider());
 #if defined(OS_CHROMEOS)
   if (arc::ArcBridgeService::GetEnabled(
           base::CommandLine::ForCurrentProcess())) {
     task_providers_.push_back(new ArcProcessTaskProvider());
@@ skipping 305 matching lines @@
 
 void TaskManagerImpl::TaskAdded(Task* task) {
   DCHECK(task);
 
   TaskGroup* task_group = nullptr;
   const base::ProcessId proc_id = task->process_id();
   const TaskId task_id = task->task_id();
 
   auto itr = task_groups_by_proc_id_.find(proc_id);
   if (itr == task_groups_by_proc_id_.end()) {
-    task_group = new TaskGroup(task->process_handle(),
-                               proc_id,
+    task_group = new TaskGroup(task->process_handle(), proc_id,
                                on_background_data_ready_callback_,
-                               blocking_pool_runner_);
+                               shared_sampler_, blocking_pool_runner_);
     task_groups_by_proc_id_[proc_id] = task_group;
   } else {
     task_group = itr->second;
   }
 
   task_group->AddTask(task);
 
   task_groups_by_task_id_[task_id] = task_group;
 
   // Invalidate the cached sorted IDs by clearing the list.
@@ skipping 139 matching lines @@
         groups_itr.second->AreBackgroundCalculationsDone();
   }
   if (are_all_processes_data_ready) {
     NotifyObserversOnRefreshWithBackgroundCalculations(GetTaskIdsList());
     for (const auto& groups_itr : task_groups_by_proc_id_)
       groups_itr.second->ClearCurrentBackgroundCalculationsFlags();
   }
 }
 
 }  // namespace task_management