[Memory-UMA] Implement basic working prototype.

services/resource_coordinator/memory/coordinator/process_map.h

Index: services/resource_coordinator/memory/coordinator/process_map.h
diff --git a/services/resource_coordinator/memory/coordinator/process_map.h b/services/resource_coordinator/memory/coordinator/process_map.h
new file mode 100644
index 0000000000000000000000000000000000000000..f349e7a1d67f3f965d3aecd4d4b2b0b4d88e0213
--- /dev/null
+++ b/services/resource_coordinator/memory/coordinator/process_map.h
@@ -0,0 +1,68 @@
+// Copyright 2017 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.
+
+#ifndef SERVICES_RESOURCE_COORDINATOR_MEMORY_COORDINATOR_PROCESS_MAP_H_
+#define SERVICES_RESOURCE_COORDINATOR_MEMORY_COORDINATOR_PROCESS_MAP_H_
+
+#include <unordered_map>
+#include <vector>
+
+#include "base/gtest_prod_util.h"
+#include "base/hash.h"
+#include "base/process/process_handle.h"
+#include "mojo/public/cpp/bindings/binding.h"
+#include "services/resource_coordinator/public/interfaces/memory/memory_instrumentation.mojom.h"
+#include "services/service_manager/public/cpp/identity.h"
+#include "services/service_manager/public/interfaces/service_manager.mojom.h"
+
+namespace memory_instrumentation {
+
+struct IdentityHasher {
+  std::size_t operator()(const service_manager::Identity& identity) const {
+    return base::Hash(identity.name() + identity.user_id() +
+                      identity.instance());
+  }
+};
+
+// Maintains an unordered_map from service_manager::Identity to
+// base::ProcessId pid. This allows |pid| lookup by service_manager::Identity.
+class ProcessMap : public service_manager::mojom::ServiceManagerListener {
+ public:
+  ProcessMap();
+  ~ProcessMap() override;
+
+  // Must be called exactly once to start receiving callbacks.
+  void BindRequest(
+      service_manager::mojom::ServiceManagerListenerRequest request);
+
+  base::ProcessId GetProcessId(service_manager::Identity identity) const;
+
+ protected:
+ private:
+  FRIEND_TEST_ALL_PREFIXES(ProcessMapTest, TypicalCase);
+  FRIEND_TEST_ALL_PREFIXES(ProcessMapTest, PresentInInit);
+
+  using RunningServiceInfoPtr = service_manager::mojom::RunningServiceInfoPtr;
+
+  // Overridden from service_manager::mojom::ServiceManagerListener:
+  void OnInit(std::vector<RunningServiceInfoPtr> instances) override;
+
+  void OnServiceCreated(RunningServiceInfoPtr instance) override;
+
+  void OnServiceStarted(const service_manager::Identity& identity,
+                        uint32_t pid) override;
+  void OnServiceFailedToStart(
+      const service_manager::Identity& identity) override {}
+
+  void OnServiceStopped(const service_manager::Identity& identity) override;
+
+  mojo::Binding<service_manager::mojom::ServiceManagerListener> binding_;
+  std::unordered_map<service_manager::Identity, base::ProcessId, IdentityHasher>
+      instances_;
+
+  DISALLOW_COPY_AND_ASSIGN(ProcessMap);
+};
+
+}  // namespace memory_instrumentation
+#endif  // SERVICES_RESOURCE_COORDINATOR_MEMORY_COORDINATOR_PROCESS_MAP_H_