content/browser/scheduler/responsiveness/message_loop_observer.cc

// 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(TaskCallback will_run_task_callback,
                                         TaskCallback did_run_task_callback)
    : will_run_task_callback_(will_run_task_callback),
      did_run_task_callback_(did_run_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) {
  will_run_task_callback_.Run(&pending_task);
}

void MessageLoopObserver::DidProcessTask(
    const base::PendingTask& pending_task) {
  did_run_task_callback_.Run(&pending_task);
}

}  // namespace responsiveness
}  // namespace content