Report utility process JS memory in task manager.

content/browser/process_resource_usage_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 "content/browser/process_resource_usage_impl.h"
+
+#include "base/bind.h"
+#include "base/memory/weak_ptr.h"
+#include "base/threading/thread_checker.h"
+#include "content/public/browser/browser_thread.h"
+#include "third_party/mojo/src/mojo/public/cpp/bindings/error_handler.h"
+
+namespace content {
+
+class ProcessResourceUsageImpl::Client : public base::SupportsWeakPtr<Client>,
+                                         public mojo::ErrorHandler {
+ public:
+  explicit Client(ResourceUsageReporterPtr service);
+
+  // Deletes self and closes the Mojo connection.
+  void Destroy();
+
+  // Refresh the resource data.
+  void Refresh(const base::Callback<void(ResourceUsageDataPtr)>& callback);
+
+ private:
+  // Overridden from mojo::ErrorHandler:
+  void OnConnectionError() override;
+
+  // Mojo IPC callback.
+  void OnGetUsageDataDone(
+      const base::Callback<void(ResourceUsageDataPtr)>& callback,
+      ResourceUsageDataPtr data);
+
+  ResourceUsageReporterPtr service_;
+  bool update_in_progress_;
+
+  base::ThreadChecker thread_checker_;
+
+  DISALLOW_COPY_AND_ASSIGN(Client);
+};
+
+ProcessResourceUsageImpl::Client::Client(ResourceUsageReporterPtr service)
+    : service_(service.Pass()), update_in_progress_(false) {
+  service_.set_error_handler(this);
+}
+
+void ProcessResourceUsageImpl::Client::Destroy() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  delete this;
+}
+
+void ProcessResourceUsageImpl::Client::OnConnectionError() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  Destroy();
+}
+
+void ProcessResourceUsageImpl::Client::Refresh(
+    const base::Callback<void(ResourceUsageDataPtr)>& callback) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  if (!update_in_progress_ && service_) {
+    update_in_progress_ = true;
+    service_->GetUsageData(
+        base::Bind(&ProcessResourceUsageImpl::Client::OnGetUsageDataDone,
+                   base::Unretained(this), callback));
+  }
+}
+
+void ProcessResourceUsageImpl::Client::OnGetUsageDataDone(
+    const base::Callback<void(ResourceUsageDataPtr)>& callback,
+    ResourceUsageDataPtr data) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  update_in_progress_ = false;
+  callback.Run(data.Pass());
+}
+
+ProcessResourceUsageImpl::ProcessResourceUsageImpl(
+    ResourceUsageReporterPtr service)
+    : mojo_message_loop_(base::MessageLoopProxy::current()),
+      client_((new Client(service.Pass()))->AsWeakPtr()) {
+}
+
+ProcessResourceUsageImpl::~ProcessResourceUsageImpl() {
+  mojo_message_loop_->PostTask(
+      FROM_HERE,
+      base::Bind(&ProcessResourceUsageImpl::Client::Destroy, client_));
+}
+
+void ProcessResourceUsageImpl::Refresh() {
+  mojo_message_loop_->PostTask(
+      FROM_HERE,
+      base::Bind(&ProcessResourceUsageImpl::Client::Refresh, client_,
+                 base::Bind(&ProcessResourceUsageImpl::OnRefreshDone, this)));
+}
+
+void ProcessResourceUsageImpl::OnRefreshDone(ResourceUsageDataPtr data) {
+  base::AutoLock l(lock_);
+  stats_ = data.Pass();
+}
+
+bool ProcessResourceUsageImpl::ReportsV8MemoryStats() const {
+  base::AutoLock l(lock_);
+  if (stats_)
+    return stats_->reports_v8_stats;
+  return false;
+}
+
+size_t ProcessResourceUsageImpl::GetV8MemoryAllocated() const {
+  base::AutoLock l(lock_);
+  if (stats_ && stats_->reports_v8_stats)
+    return stats_->v8_memory_allocated;
+
+  return 0;
+}
+
+size_t ProcessResourceUsageImpl::GetV8MemoryUsed() const {
+  base::AutoLock l(lock_);
+  if (stats_ && stats_->reports_v8_stats)
+    return stats_->v8_memory_used;
+
+  return 0;
+}
+
+}  // namespace content