Expose MemoryCoordinator's global budget information.

content/child/memory/child_memory_coordinator_impl.cc

 // Copyright 2016 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/child/memory/child_memory_coordinator_impl.h"
 
 #include "base/lazy_instance.h"
 #include "base/memory/memory_coordinator_client_registry.h"
+#include "base/memory/memory_coordinator_proxy.h"
 #include "base/synchronization/lock.h"
 #include "base/trace_event/trace_event.h"
 
 namespace content {
 
 namespace {
 
 base::LazyInstance<base::Lock>::Leaky g_lock = LAZY_INSTANCE_INITIALIZER;
 ChildMemoryCoordinatorImpl* g_child_memory_coordinator = nullptr;
 
 base::MemoryState ToBaseMemoryState(mojom::MemoryState state) {
   switch (state) {
     case mojom::MemoryState::UNKNOWN:
       return base::MemoryState::UNKNOWN;
     case mojom::MemoryState::NORMAL:
       return base::MemoryState::NORMAL;
     case mojom::MemoryState::THROTTLED:
       return base::MemoryState::THROTTLED;
     case mojom::MemoryState::SUSPENDED:
       return base::MemoryState::SUSPENDED;
     default:
       NOTREACHED();
       return base::MemoryState::UNKNOWN;
   }
 }
 
+// Wrapper for GetCurrentMemoryState that converts converts to a
+// base::MemoryState.
+base::MemoryState GetCurrentMemoryState() {
+  return ToBaseMemoryState(g_child_memory_coordinator->GetCurrentMemoryState());

[haraken, 2016/11/05 12:52:22]

g_child_memory_coordinator => GetInstance() ?

[chrisha, 2016/11/07 19:33:41]

Done.

+}
+
 }  // namespace
 
 // static
 ChildMemoryCoordinatorImpl* ChildMemoryCoordinatorImpl::GetInstance() {
   base::AutoLock lock(*g_lock.Pointer());
   return g_child_memory_coordinator;
 }
 
 ChildMemoryCoordinatorImpl::ChildMemoryCoordinatorImpl(
     mojom::MemoryCoordinatorHandlePtr parent,
     ChildMemoryCoordinatorDelegate* delegate)
-    : binding_(this), parent_(std::move(parent)), delegate_(delegate) {
+    : binding_(this), parent_(std::move(parent)), delegate_(delegate),
+      state_(mojom::MemoryState::UNKNOWN), remaining_global_budget_mb_(0) {
   base::AutoLock lock(*g_lock.Pointer());
   DCHECK(delegate_);
   DCHECK(!g_child_memory_coordinator);
   g_child_memory_coordinator = this;
   parent_->AddChild(binding_.CreateInterfacePtrAndBind());
 }
 
 ChildMemoryCoordinatorImpl::~ChildMemoryCoordinatorImpl() {
   base::AutoLock lock(*g_lock.Pointer());
   DCHECK(g_child_memory_coordinator == this);
   g_child_memory_coordinator = nullptr;
 }
 
 void ChildMemoryCoordinatorImpl::OnStateChange(mojom::MemoryState state) {
   base::MemoryState base_state = ToBaseMemoryState(state);
   TRACE_EVENT1("memory-infra", "ChildMemoryCoordinatorImpl::OnStateChange",
                "state", MemoryStateToString(base_state));
   base::MemoryCoordinatorClientRegistry::GetInstance()->Notify(
       base_state);
 }
 
+void ChildMemoryCoordinatorImpl::SetRemainingGlobalBudget(
+    int64_t remaining_global_budget_mb) {
+  remaining_global_budget_mb_ = remaining_global_budget_mb;
+}
+
 #if !defined(OS_ANDROID)
 std::unique_ptr<ChildMemoryCoordinatorImpl> CreateChildMemoryCoordinator(
     mojom::MemoryCoordinatorHandlePtr parent,
     ChildMemoryCoordinatorDelegate* delegate) {
-  return base::WrapUnique(
-      new ChildMemoryCoordinatorImpl(std::move(parent), delegate));
+  auto* cmc = new ChildMemoryCoordinatorImpl(std::move(parent), delegate);
+
+  // Hookup the coordinator to the MemoryCoordinatorProxy. base::Unretained is
+  // safe because the lifetime of the ChildMemoryCoordinatorImpl is tied to the
+  // lifetime of the renderer process.
+  base::MemoryCoordinatorProxy::GetInstance()->
+      SetGetCurrentMemoryStateCallback(base::Bind(&GetCurrentMemoryState));
+  base::MemoryCoordinatorProxy::GetInstance()->
+      SetGetRemainingGlobalBudgetCallback(base::Bind(
+          &ChildMemoryCoordinatorImpl::GetRemainingGlobalBudget,
+          base::Unretained(cmc)));
+
+  return base::WrapUnique(cmc);
 }
 #endif
 
 }  // namespace content