[GRC] Coordination Unit Graph Observer

services/resource_coordinator/coordination_unit/coordination_unit_manager.cc

 // 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.
 
 #include "services/resource_coordinator/coordination_unit/coordination_unit_manager.h"
 
+#include <memory>
+
 #include "base/bind.h"
 #include "base/bind_helpers.h"
+#include "base/macros.h"
+#include "services/resource_coordinator/coordination_unit/coordination_unit_graph_observer.h"
+#include "services/resource_coordinator/coordination_unit/coordination_unit_impl.h"
 #include "services/resource_coordinator/coordination_unit/coordination_unit_provider_impl.h"
+#include "services/resource_coordinator/public/cpp/coordination_unit_types.h"
 #include "services/service_manager/public/cpp/binder_registry.h"
 #include "services/service_manager/public/cpp/service_context.h"
 
 namespace resource_coordinator {
 
-class CoordinationUnitImpl;
-
 CoordinationUnitManager::CoordinationUnitManager() = default;
 
-CoordinationUnitManager::~CoordinationUnitManager() = default;
+CoordinationUnitManager::~CoordinationUnitManager() {
+  for (CoordinationUnitGraphObserver* observer : observers_) {
+    delete observer;
+  }
+}
 
 void CoordinationUnitManager::OnStart(
     service_manager::BinderRegistry* registry,
     service_manager::ServiceContextRefFactory* service_ref_factory) {
   registry->AddInterface(base::Bind(&CoordinationUnitProviderImpl::Create,
-                                    base::Unretained(service_ref_factory)));
+                                    base::Unretained(service_ref_factory),
+                                    base::Unretained(this)));
+}
+
+void CoordinationUnitManager::RegisterObserver(
+    CoordinationUnitGraphObserver* observer) {
+  observers_.push_back(observer);
+}
+
+void CoordinationUnitManager::NotifyObserversCoordinationUnitCreated(
+    CoordinationUnitImpl* coordination_unit) {
+  for (CoordinationUnitGraphObserver* observer : observers_) {
+    if (observer->filter() == coordination_unit->id().type ||
+        // CoordinationUnitType::kInvalidType overrides filtering.
+        observer->filter() == CoordinationUnitType::kInvalidType) {

[Zhen Wang, 2017/06/15 20:07:53]

This is not intuitive. See comments in CUGraphObserver.

+      observer->CoordinationUnitCreated(coordination_unit);
+    }
+  }
+}
+
+void CoordinationUnitManager::NotifyObserversCoordinationUnitWillBeDestroyed(
+    CoordinationUnitImpl* coordination_unit) {
+  coordination_unit->WillBeDestroyed();
+}
+
+void CoordinationUnitManager::UnregisterAllObserversForTesting() {
+  observers_.clear();
 }
 
 }  // namespace resource_coordinator