OLD | NEW |
1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "services/resource_coordinator/public/cpp/memory/memory_dump_manager_de
legate_impl.h" | 5 #include "services/resource_coordinator/public/cpp/memory/memory_dump_manager_de
legate_impl.h" |
6 | 6 |
7 #include "base/bind.h" | |
8 #include "base/bind_helpers.h" | |
9 #include "base/lazy_instance.h" | |
10 #include "base/single_thread_task_runner.h" | |
11 #include "base/synchronization/lock.h" | |
12 #include "base/trace_event/memory_dump_request_args.h" | 7 #include "base/trace_event/memory_dump_request_args.h" |
13 #include "mojo/public/cpp/bindings/interface_request.h" | 8 #include "mojo/public/cpp/bindings/interface_request.h" |
14 #include "services/resource_coordinator/public/cpp/memory/coordinator.h" | 9 #include "services/resource_coordinator/public/cpp/memory/coordinator.h" |
15 #include "services/resource_coordinator/public/interfaces/memory/memory_instrume
ntation.mojom.h" | 10 #include "services/resource_coordinator/public/interfaces/memory/memory_instrume
ntation.mojom.h" |
16 #include "services/service_manager/public/cpp/interface_provider.h" | 11 #include "services/service_manager/public/cpp/interface_provider.h" |
17 | 12 |
18 namespace memory_instrumentation { | 13 namespace memory_instrumentation { |
19 | 14 |
20 namespace { | 15 MemoryDumpManagerDelegateImpl::MemoryDumpManagerDelegateImpl( |
21 | 16 service_manager::InterfaceProvider* interface_provider) |
22 base::LazyInstance<MemoryDumpManagerDelegateImpl>::Leaky | 17 : is_coordinator_(false), binding_(this) { |
23 g_memory_dump_manager_delegate = LAZY_INSTANCE_INITIALIZER; | 18 interface_provider->GetInterface(mojo::MakeRequest(&coordinator_)); |
24 | 19 coordinator_->RegisterProcessLocalDumpManager( |
25 } // namespace | 20 binding_.CreateInterfacePtrAndBind()); |
26 | |
27 // static | |
28 MemoryDumpManagerDelegateImpl* MemoryDumpManagerDelegateImpl::GetInstance() { | |
29 return g_memory_dump_manager_delegate.Pointer(); | |
30 } | 21 } |
31 | 22 |
32 MemoryDumpManagerDelegateImpl::MemoryDumpManagerDelegateImpl() | 23 MemoryDumpManagerDelegateImpl::MemoryDumpManagerDelegateImpl( |
33 : initialized_(false), | 24 Coordinator* coordinator) |
34 binding_(this), | 25 : is_coordinator_(true), binding_(this) { |
35 task_runner_(nullptr), | 26 coordinator->BindCoordinatorRequest(mojo::MakeRequest(&coordinator_)); |
36 pending_memory_dump_guid_(0) {} | 27 coordinator_->RegisterProcessLocalDumpManager( |
| 28 binding_.CreateInterfacePtrAndBind()); |
| 29 } |
37 | 30 |
38 MemoryDumpManagerDelegateImpl::~MemoryDumpManagerDelegateImpl() {} | 31 MemoryDumpManagerDelegateImpl::~MemoryDumpManagerDelegateImpl() {} |
39 | 32 |
40 void MemoryDumpManagerDelegateImpl::InitializeWithInterfaceProvider( | |
41 service_manager::InterfaceProvider* interface_provider) { | |
42 { | |
43 base::AutoLock lock(initialized_lock_); | |
44 DCHECK(!initialized_); | |
45 initialized_ = true; | |
46 } | |
47 | |
48 interface_provider->GetInterface(mojo::MakeRequest(&coordinator_)); | |
49 coordinator_->RegisterProcessLocalDumpManager( | |
50 binding_.CreateInterfacePtrAndBind()); | |
51 base::trace_event::MemoryDumpManager::GetInstance()->Initialize(this); | |
52 } | |
53 | |
54 void MemoryDumpManagerDelegateImpl::InitializeWithCoordinator( | |
55 Coordinator* coordinator, | |
56 scoped_refptr<base::SingleThreadTaskRunner> task_runner) { | |
57 DCHECK(task_runner); | |
58 if (!task_runner->RunsTasksOnCurrentThread()) { | |
59 task_runner->PostTask( | |
60 FROM_HERE, | |
61 base::Bind(&MemoryDumpManagerDelegateImpl::InitializeWithCoordinator, | |
62 base::Unretained(this), base::Unretained(coordinator), | |
63 task_runner)); | |
64 return; | |
65 } | |
66 | |
67 { | |
68 base::AutoLock lock(initialized_lock_); | |
69 DCHECK(!initialized_); | |
70 initialized_ = true; | |
71 } | |
72 | |
73 task_runner_ = task_runner; | |
74 coordinator->BindCoordinatorRequest(mojo::MakeRequest(&coordinator_)); | |
75 coordinator_->RegisterProcessLocalDumpManager( | |
76 binding_.CreateInterfacePtrAndBind()); | |
77 base::trace_event::MemoryDumpManager::GetInstance()->Initialize(this); | |
78 } | |
79 | |
80 bool MemoryDumpManagerDelegateImpl::IsCoordinator() const { | 33 bool MemoryDumpManagerDelegateImpl::IsCoordinator() const { |
81 return task_runner_ != nullptr; | 34 return is_coordinator_; |
82 } | 35 } |
83 | 36 |
84 void MemoryDumpManagerDelegateImpl::RequestProcessMemoryDump( | 37 void MemoryDumpManagerDelegateImpl::RequestProcessMemoryDump( |
85 const base::trace_event::MemoryDumpRequestArgs& args, | 38 const base::trace_event::MemoryDumpRequestArgs& args, |
86 const RequestProcessMemoryDumpCallback& callback) { | 39 const RequestProcessMemoryDumpCallback& callback) { |
87 MemoryDumpManagerDelegate::CreateProcessDump(args, callback); | 40 MemoryDumpManagerDelegate::CreateProcessDump(args, callback); |
88 } | 41 } |
89 | 42 |
90 void MemoryDumpManagerDelegateImpl::RequestGlobalMemoryDump( | 43 void MemoryDumpManagerDelegateImpl::RequestGlobalMemoryDump( |
91 const base::trace_event::MemoryDumpRequestArgs& args, | 44 const base::trace_event::MemoryDumpRequestArgs& args, |
92 const base::trace_event::MemoryDumpCallback& callback) { | 45 const base::trace_event::MemoryDumpCallback& callback) { |
93 // Note: This condition is here to match the old behavior. If the delegate is | 46 coordinator_->RequestGlobalMemoryDump(args, callback); |
94 // in the browser process, we do not drop parallel requests in the delegate | |
95 // and so they will be queued by the Coordinator service (see | |
96 // CoordinatorImpl::RequestGlobalMemoryDump). If the delegate is in a child | |
97 // process, parallel requests will be cancelled. | |
98 // | |
99 // TODO(chiniforooshan): After transitioning to the mojo-based service is | |
100 // completed, we should enable queueing parallel global memory dump requests | |
101 // by delegates on all processes. | |
102 if (task_runner_) { | |
103 DCHECK(task_runner_); | |
104 task_runner_->PostTask( | |
105 FROM_HERE, | |
106 base::Bind(&mojom::Coordinator::RequestGlobalMemoryDump, | |
107 base::Unretained(coordinator_.get()), args, callback)); | |
108 return; | |
109 } | |
110 | |
111 { | |
112 base::AutoLock lock(pending_memory_dump_guid_lock_); | |
113 if (pending_memory_dump_guid_) { | |
114 callback.Run(args.dump_guid, false); | |
115 return; | |
116 } | |
117 pending_memory_dump_guid_ = args.dump_guid; | |
118 } | |
119 DCHECK(!task_runner_); | |
120 auto callback_proxy = | |
121 base::Bind(&MemoryDumpManagerDelegateImpl::MemoryDumpCallbackProxy, | |
122 base::Unretained(this), callback); | |
123 coordinator_->RequestGlobalMemoryDump(args, callback_proxy); | |
124 } | |
125 | |
126 void MemoryDumpManagerDelegateImpl::MemoryDumpCallbackProxy( | |
127 const base::trace_event::MemoryDumpCallback& callback, | |
128 uint64_t dump_guid, | |
129 bool success) { | |
130 DCHECK_NE(0U, pending_memory_dump_guid_); | |
131 pending_memory_dump_guid_ = 0; | |
132 callback.Run(dump_guid, success); | |
133 } | |
134 | |
135 void MemoryDumpManagerDelegateImpl::SetAsNonCoordinatorForTesting() { | |
136 task_runner_ = nullptr; | |
137 } | 47 } |
138 | 48 |
139 } // namespace memory_instrumentation | 49 } // namespace memory_instrumentation |
OLD | NEW |