| // Copyright 2018 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 "content/browser/scheduler/responsiveness/message_loop_observer.h" |
| |
| #include "base/message_loop/message_loop_current.h" |
| |
| namespace content { |
| namespace responsiveness { |
| |
| MessageLoopObserver::MessageLoopObserver( |
| WillProcessTaskCallback will_process_task_callback, |
| DidProcessTaskCallback did_process_task_callback) |
| : will_process_task_callback_(std::move(will_process_task_callback)), |
| did_process_task_callback_(std::move(did_process_task_callback)) { |
| base::MessageLoopCurrent::Get()->SetAddQueueTimeToTasks(true); |
| base::MessageLoopCurrent::Get()->AddTaskObserver(this); |
| } |
| |
| MessageLoopObserver::~MessageLoopObserver() { |
| base::MessageLoopCurrent::Get()->RemoveTaskObserver(this); |
| base::MessageLoopCurrent::Get()->SetAddQueueTimeToTasks(false); |
| } |
| |
| void MessageLoopObserver::WillProcessTask(const base::PendingTask& pending_task, |
| bool was_blocked_or_low_priority) { |
| will_process_task_callback_.Run(&pending_task, was_blocked_or_low_priority); |
| } |
| |
| void MessageLoopObserver::DidProcessTask( |
| const base::PendingTask& pending_task) { |
| did_process_task_callback_.Run(&pending_task); |
| } |
| |
| } // namespace responsiveness |
| } // namespace content |