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 |