| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 #ifndef MOJO_PUBLIC_CPP_BINDINGS_STRONG_ASSOCIATED_BINDING_H_ | 5 #ifndef MOJO_PUBLIC_CPP_BINDINGS_STRONG_ASSOCIATED_BINDING_H_ |
| 6 #define MOJO_PUBLIC_CPP_BINDINGS_STRONG_ASSOCIATED_BINDING_H_ | 6 #define MOJO_PUBLIC_CPP_BINDINGS_STRONG_ASSOCIATED_BINDING_H_ |
| 7 | 7 |
| 8 #include <memory> | 8 #include <memory> |
| 9 #include <string> | 9 #include <string> |
| 10 #include <utility> | 10 #include <utility> |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 51 AssociatedInterfaceRequest<Interface> request) { | 51 AssociatedInterfaceRequest<Interface> request) { |
| 52 StrongAssociatedBinding* binding = | 52 StrongAssociatedBinding* binding = |
| 53 new StrongAssociatedBinding(std::move(impl), std::move(request)); | 53 new StrongAssociatedBinding(std::move(impl), std::move(request)); |
| 54 return binding->weak_factory_.GetWeakPtr(); | 54 return binding->weak_factory_.GetWeakPtr(); |
| 55 } | 55 } |
| 56 | 56 |
| 57 // Note: The error handler must not delete the interface implementation. | 57 // Note: The error handler must not delete the interface implementation. |
| 58 // | 58 // |
| 59 // This method may only be called after this StrongAssociatedBinding has been | 59 // This method may only be called after this StrongAssociatedBinding has been |
| 60 // bound to a message pipe. | 60 // bound to a message pipe. |
| 61 void set_connection_error_handler(const base::Closure& error_handler) { | 61 void set_connection_error_handler(base::OnceClosure error_handler) { |
| 62 DCHECK(binding_.is_bound()); | 62 DCHECK(binding_.is_bound()); |
| 63 connection_error_handler_ = error_handler; | 63 connection_error_handler_ = std::move(error_handler); |
| 64 connection_error_with_reason_handler_.Reset(); | 64 connection_error_with_reason_handler_.Reset(); |
| 65 } | 65 } |
| 66 | 66 |
| 67 void set_connection_error_with_reason_handler( | 67 void set_connection_error_with_reason_handler( |
| 68 const ConnectionErrorWithReasonCallback& error_handler) { | 68 ConnectionErrorWithReasonCallback error_handler) { |
| 69 DCHECK(binding_.is_bound()); | 69 DCHECK(binding_.is_bound()); |
| 70 connection_error_with_reason_handler_ = error_handler; | 70 connection_error_with_reason_handler_ = std::move(error_handler); |
| 71 connection_error_handler_.Reset(); | 71 connection_error_handler_.Reset(); |
| 72 } | 72 } |
| 73 | 73 |
| 74 // Forces the binding to close. This destroys the StrongBinding instance. | 74 // Forces the binding to close. This destroys the StrongBinding instance. |
| 75 void Close() { delete this; } | 75 void Close() { delete this; } |
| 76 | 76 |
| 77 Interface* impl() { return impl_.get(); } | 77 Interface* impl() { return impl_.get(); } |
| 78 | 78 |
| 79 // Sends a message on the underlying message pipe and runs the current | 79 // Sends a message on the underlying message pipe and runs the current |
| 80 // message loop until its response is received. This can be used in tests to | 80 // message loop until its response is received. This can be used in tests to |
| 81 // verify that no message was sent on a message pipe in response to some | 81 // verify that no message was sent on a message pipe in response to some |
| 82 // stimulus. | 82 // stimulus. |
| 83 void FlushForTesting() { binding_.FlushForTesting(); } | 83 void FlushForTesting() { binding_.FlushForTesting(); } |
| 84 | 84 |
| 85 private: | 85 private: |
| 86 StrongAssociatedBinding(std::unique_ptr<Interface> impl, | 86 StrongAssociatedBinding(std::unique_ptr<Interface> impl, |
| 87 AssociatedInterfaceRequest<Interface> request) | 87 AssociatedInterfaceRequest<Interface> request) |
| 88 : impl_(std::move(impl)), | 88 : impl_(std::move(impl)), |
| 89 binding_(impl_.get(), std::move(request)), | 89 binding_(impl_.get(), std::move(request)), |
| 90 weak_factory_(this) { | 90 weak_factory_(this) { |
| 91 binding_.set_connection_error_with_reason_handler(base::Bind( | 91 binding_.set_connection_error_with_reason_handler(base::Bind( |
| 92 &StrongAssociatedBinding::OnConnectionError, base::Unretained(this))); | 92 &StrongAssociatedBinding::OnConnectionError, base::Unretained(this))); |
| 93 } | 93 } |
| 94 | 94 |
| 95 ~StrongAssociatedBinding() {} | 95 ~StrongAssociatedBinding() {} |
| 96 | 96 |
| 97 void OnConnectionError(uint32_t custom_reason, | 97 void OnConnectionError(uint32_t custom_reason, |
| 98 const std::string& description) { | 98 const std::string& description) { |
| 99 if (!connection_error_handler_.is_null()) | 99 if (connection_error_handler_) { |
| 100 connection_error_handler_.Run(); | 100 std::move(connection_error_handler_).Run(); |
| 101 else if (!connection_error_with_reason_handler_.is_null()) | 101 } else if (connection_error_with_reason_handler_) { |
| 102 connection_error_with_reason_handler_.Run(custom_reason, description); | 102 std::move(connection_error_with_reason_handler_) |
| 103 .Run(custom_reason, description); |
| 104 } |
| 103 Close(); | 105 Close(); |
| 104 } | 106 } |
| 105 | 107 |
| 106 std::unique_ptr<Interface> impl_; | 108 std::unique_ptr<Interface> impl_; |
| 107 base::Closure connection_error_handler_; | 109 base::OnceClosure connection_error_handler_; |
| 108 ConnectionErrorWithReasonCallback connection_error_with_reason_handler_; | 110 ConnectionErrorWithReasonCallback connection_error_with_reason_handler_; |
| 109 AssociatedBinding<Interface> binding_; | 111 AssociatedBinding<Interface> binding_; |
| 110 base::WeakPtrFactory<StrongAssociatedBinding> weak_factory_; | 112 base::WeakPtrFactory<StrongAssociatedBinding> weak_factory_; |
| 111 | 113 |
| 112 DISALLOW_COPY_AND_ASSIGN(StrongAssociatedBinding); | 114 DISALLOW_COPY_AND_ASSIGN(StrongAssociatedBinding); |
| 113 }; | 115 }; |
| 114 | 116 |
| 115 template <typename Interface, typename Impl> | 117 template <typename Interface, typename Impl> |
| 116 StrongAssociatedBindingPtr<Interface> MakeStrongAssociatedBinding( | 118 StrongAssociatedBindingPtr<Interface> MakeStrongAssociatedBinding( |
| 117 std::unique_ptr<Impl> impl, | 119 std::unique_ptr<Impl> impl, |
| 118 AssociatedInterfaceRequest<Interface> request) { | 120 AssociatedInterfaceRequest<Interface> request) { |
| 119 return StrongAssociatedBinding<Interface>::Create(std::move(impl), | 121 return StrongAssociatedBinding<Interface>::Create(std::move(impl), |
| 120 std::move(request)); | 122 std::move(request)); |
| 121 } | 123 } |
| 122 | 124 |
| 123 } // namespace mojo | 125 } // namespace mojo |
| 124 | 126 |
| 125 #endif // MOJO_PUBLIC_CPP_BINDINGS_STRONG_ASSOCIATED_BINDING_H_ | 127 #endif // MOJO_PUBLIC_CPP_BINDINGS_STRONG_ASSOCIATED_BINDING_H_ |
| OLD | NEW |