[Memory-UMA] Implement basic working prototype.

services/resource_coordinator/memory/coordinator/process_map.h

+// 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>

[Primiano Tucci (use gerrit), 2017/05/15 03:48:10]

s/unordered_map/map/ (see other comment below)

[erikchen, 2017/05/15 17:29:15]

Done.

+#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 {

[Primiano Tucci (use gerrit), 2017/05/15 03:48:10]

no need of this. Use a std::map (see brett's post about unordered-map on
chromium-dev)
Identity has already operator< defined, so you can just use it in a map

[erikchen, 2017/05/15 17:29:15]

Done.

+  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);

[Primiano Tucci (use gerrit), 2017/05/15 03:48:10]

if you just pass the Connector in the ctor there is no need of this function

[erikchen, 2017/05/15 17:29:15]

Done.

+
+  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_