Index: content/browser/process_resource_usage_impl.cc |
diff --git a/content/browser/process_resource_usage_impl.cc b/content/browser/process_resource_usage_impl.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..252aef3141cc3c4d11560367aab47a34bd10c00f |
--- /dev/null |
+++ b/content/browser/process_resource_usage_impl.cc |
@@ -0,0 +1,124 @@ |
+// 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 |