| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "content/common/mojo/service_registry_impl.h" | 5 #include "content/common/mojo/service_registry_impl.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 | 8 |
| 9 #include "mojo/common/common_type_converters.h" | 9 #include "mojo/common/common_type_converters.h" |
| 10 | 10 |
| (...skipping 21 matching lines...) Expand all Loading... |
| 32 CHECK(!remote_provider_); | 32 CHECK(!remote_provider_); |
| 33 remote_provider_ = std::move(service_provider); | 33 remote_provider_ = std::move(service_provider); |
| 34 while (!pending_connects_.empty()) { | 34 while (!pending_connects_.empty()) { |
| 35 remote_provider_->ConnectToService( | 35 remote_provider_->ConnectToService( |
| 36 mojo::String::From(pending_connects_.front().first), | 36 mojo::String::From(pending_connects_.front().first), |
| 37 mojo::ScopedMessagePipeHandle(pending_connects_.front().second)); | 37 mojo::ScopedMessagePipeHandle(pending_connects_.front().second)); |
| 38 pending_connects_.pop(); | 38 pending_connects_.pop(); |
| 39 } | 39 } |
| 40 } | 40 } |
| 41 | 41 |
| 42 void ServiceRegistryImpl::AddServiceOverrideForTesting( | 42 void ServiceRegistryImpl::AddService( |
| 43 const std::string& service_name, | 43 const std::string& service_name, |
| 44 const ServiceFactory& factory) { | 44 const base::Callback<void(mojo::ScopedMessagePipeHandle)> service_factory) { |
| 45 service_overrides_[service_name] = factory; | |
| 46 } | |
| 47 | |
| 48 void ServiceRegistryImpl::AddService(const std::string& service_name, | |
| 49 const ServiceFactory service_factory) { | |
| 50 service_factories_[service_name] = service_factory; | 45 service_factories_[service_name] = service_factory; |
| 51 } | 46 } |
| 52 | 47 |
| 53 void ServiceRegistryImpl::RemoveService(const std::string& service_name) { | 48 void ServiceRegistryImpl::RemoveService(const std::string& service_name) { |
| 54 service_factories_.erase(service_name); | 49 service_factories_.erase(service_name); |
| 55 } | 50 } |
| 56 | 51 |
| 57 void ServiceRegistryImpl::ConnectToRemoteService( | 52 void ServiceRegistryImpl::ConnectToRemoteService( |
| 58 const base::StringPiece& service_name, | 53 const base::StringPiece& service_name, |
| 59 mojo::ScopedMessagePipeHandle handle) { | 54 mojo::ScopedMessagePipeHandle handle) { |
| 60 auto override_it = service_overrides_.find(service_name.as_string()); | |
| 61 if (override_it != service_overrides_.end()) { | |
| 62 override_it->second.Run(std::move(handle)); | |
| 63 return; | |
| 64 } | |
| 65 | |
| 66 if (!remote_provider_) { | 55 if (!remote_provider_) { |
| 67 pending_connects_.push( | 56 pending_connects_.push( |
| 68 std::make_pair(service_name.as_string(), handle.release())); | 57 std::make_pair(service_name.as_string(), handle.release())); |
| 69 return; | 58 return; |
| 70 } | 59 } |
| 71 remote_provider_->ConnectToService( | 60 remote_provider_->ConnectToService( |
| 72 mojo::String::From(service_name.as_string()), std::move(handle)); | 61 mojo::String::From(service_name.as_string()), std::move(handle)); |
| 73 } | 62 } |
| 74 | 63 |
| 75 bool ServiceRegistryImpl::IsBound() const { | 64 bool ServiceRegistryImpl::IsBound() const { |
| 76 return binding_.is_bound(); | 65 return binding_.is_bound(); |
| 77 } | 66 } |
| 78 | 67 |
| 79 base::WeakPtr<ServiceRegistry> ServiceRegistryImpl::GetWeakPtr() { | 68 base::WeakPtr<ServiceRegistry> ServiceRegistryImpl::GetWeakPtr() { |
| 80 return weak_factory_.GetWeakPtr(); | 69 return weak_factory_.GetWeakPtr(); |
| 81 } | 70 } |
| 82 | 71 |
| 83 void ServiceRegistryImpl::ConnectToService( | 72 void ServiceRegistryImpl::ConnectToService( |
| 84 const mojo::String& name, | 73 const mojo::String& name, |
| 85 mojo::ScopedMessagePipeHandle client_handle) { | 74 mojo::ScopedMessagePipeHandle client_handle) { |
| 86 auto it = service_factories_.find(name); | 75 std::map<std::string, |
| 76 base::Callback<void(mojo::ScopedMessagePipeHandle)> >::iterator it = |
| 77 service_factories_.find(name); |
| 87 if (it == service_factories_.end()) | 78 if (it == service_factories_.end()) |
| 88 return; | 79 return; |
| 89 | 80 |
| 90 // It's possible and effectively unavoidable that under certain conditions | 81 // It's possible and effectively unavoidable that under certain conditions |
| 91 // an invalid handle may be received. Don't invoke the factory in that case. | 82 // an invalid handle may be received. Don't invoke the factory in that case. |
| 92 if (!client_handle.is_valid()) { | 83 if (!client_handle.is_valid()) { |
| 93 DVLOG(2) << "Invalid pipe handle for " << name << " interface request."; | 84 DVLOG(2) << "Invalid pipe handle for " << name << " interface request."; |
| 94 return; | 85 return; |
| 95 } | 86 } |
| 96 | 87 |
| 97 it->second.Run(std::move(client_handle)); | 88 it->second.Run(std::move(client_handle)); |
| 98 } | 89 } |
| 99 | 90 |
| 100 void ServiceRegistryImpl::OnConnectionError() { | 91 void ServiceRegistryImpl::OnConnectionError() { |
| 101 binding_.Close(); | 92 binding_.Close(); |
| 102 } | 93 } |
| 103 | 94 |
| 104 } // namespace content | 95 } // namespace content |
| OLD | NEW |