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 "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"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.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_.emplace_back(new BrowserProcessTaskProvider());
   task_providers_.emplace_back(new ChildProcessTaskProvider());
   task_providers_.emplace_back(new WebContentsTaskProvider());
 #if defined(OS_CHROMEOS)
   if (arc::ArcBridgeService::GetEnabled(
           base::CommandLine::ForCurrentProcess())) {
     task_providers_.emplace_back(new ArcProcessTaskProvider());
@@ skipping 316 matching lines @@
 void TaskManagerImpl::TaskAdded(Task* task) {
   DCHECK(task);
 
   const base::ProcessId proc_id = task->process_id();
   const TaskId task_id = task->task_id();
 
   std::unique_ptr<TaskGroup>& task_group = task_groups_by_proc_id_[proc_id];
   if (!task_group)
     task_group.reset(new TaskGroup(task->process_handle(), proc_id,
                                    on_background_data_ready_callback_,
-                                   blocking_pool_runner_));
+                                   shared_sampler_, blocking_pool_runner_));
 
   task_group->AddTask(task);
 
   task_groups_by_task_id_[task_id] = task_group.get();
 
   // Invalidate the cached sorted IDs by clearing the list.
   sorted_task_ids_.clear();
 
   NotifyObserversOnTaskAdded(task_id);
 }
@@ skipping 140 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