Mandoline: Introduce ApplicationConnection::CloseConnection

mojo/application/public/cpp/lib/application_impl.cc

Index: mojo/application/public/cpp/lib/application_impl.cc
diff --git a/mojo/application/public/cpp/lib/application_impl.cc b/mojo/application/public/cpp/lib/application_impl.cc
index bafabe276fbc04eb61da9300ebc3412fbd1577e8..6ff7cc1aaeae2718c24780a5415aed0b21fbf968 100644
--- a/mojo/application/public/cpp/lib/application_impl.cc
+++ b/mojo/application/public/cpp/lib/application_impl.cc
@@ -4,6 +4,8 @@
 
 #include "mojo/application/public/cpp/application_impl.h"
 
+#include <algorithm>
+
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "mojo/application/public/cpp/application_delegate.h"
@@ -77,6 +79,23 @@ ApplicationConnection* ApplicationImpl::ConnectToApplication(
   return registry;
 }
 
+void ApplicationImpl::CloseConnection(ApplicationConnection* connection) {
+  scoped_ptr<ApplicationConnection> connection_owner(connection);
+  delegate_->OnWillCloseConnection(connection);
+  auto outgoing_it = std::find(outgoing_service_registries_.begin(),
+                               outgoing_service_registries_.end(),
+                               connection);
+  if (outgoing_it != outgoing_service_registries_.end()) {
+    outgoing_service_registries_.erase(outgoing_it);
+    return;
+  }
+  auto incoming_it = std::find(incoming_service_registries_.begin(),
+                               incoming_service_registries_.end(),
+                               connection);
+ if (incoming_it != incoming_service_registries_.end())
+    incoming_service_registries_.erase(incoming_it);
+}
+
 void ApplicationImpl::Initialize(ShellPtr shell, const mojo::String& url) {
   shell_ = shell.Pass();
   shell_.set_error_handler(this);