Index: mojo/public/cpp/bindings/interface_ptr_set.h |
diff --git a/mojo/public/cpp/bindings/interface_ptr_set.h b/mojo/public/cpp/bindings/interface_ptr_set.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..c5d402d7cfaf51571c418c3460c2ee7d71fc24f6 |
--- /dev/null |
+++ b/mojo/public/cpp/bindings/interface_ptr_set.h |
@@ -0,0 +1,83 @@ |
+// Copyright 2014 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. |
+ |
+#ifndef MOJO_PUBLIC_CPP_BINDINGS_INTERFACE_PTR_SET_H_ |
+#define MOJO_PUBLIC_CPP_BINDINGS_INTERFACE_PTR_SET_H_ |
+ |
+#include <utility> |
+#include <vector> |
+ |
+#include "base/macros.h" |
+#include "base/memory/weak_ptr.h" |
+#include "mojo/public/cpp/bindings/interface_ptr.h" |
+ |
+namespace mojo { |
+ |
+template <typename Interface> |
+class InterfacePtrSet { |
+ public: |
+ InterfacePtrSet() {} |
+ ~InterfacePtrSet() { CloseAll(); } |
+ |
+ void AddInterfacePtr(InterfacePtr<Interface> ptr) { |
+ auto weak_interface_ptr = new Element(std::move(ptr)); |
+ ptrs_.push_back(weak_interface_ptr->GetWeakPtr()); |
+ ClearNullInterfacePtrs(); |
+ } |
+ |
+ template <typename FunctionType> |
+ void ForAllPtrs(FunctionType function) { |
+ for (const auto& it : ptrs_) { |
+ if (it) |
+ function(it->get()); |
+ } |
+ ClearNullInterfacePtrs(); |
+ } |
+ |
+ void CloseAll() { |
+ for (const auto& it : ptrs_) { |
+ if (it) |
+ it->Close(); |
+ } |
+ ptrs_.clear(); |
+ } |
+ |
+ private: |
+ class Element { |
+ public: |
+ explicit Element(InterfacePtr<Interface> ptr) |
+ : ptr_(std::move(ptr)), weak_ptr_factory_(this) { |
+ ptr_.set_connection_error_handler([this]() { delete this; }); |
+ } |
+ ~Element() {} |
+ |
+ void Close() { ptr_.reset(); } |
+ |
+ Interface* get() { return ptr_.get(); } |
+ |
+ base::WeakPtr<Element> GetWeakPtr() { |
+ return weak_ptr_factory_.GetWeakPtr(); |
+ } |
+ |
+ private: |
+ InterfacePtr<Interface> ptr_; |
+ base::WeakPtrFactory<Element> weak_ptr_factory_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(Element); |
+ }; |
+ |
+ void ClearNullInterfacePtrs() { |
+ ptrs_.erase(std::remove_if(ptrs_.begin(), ptrs_.end(), |
+ [](const base::WeakPtr<Element>& p) { |
+ return p.get() == nullptr; |
+ }), |
+ ptrs_.end()); |
+ } |
+ |
+ std::vector<base::WeakPtr<Element>> ptrs_; |
+}; |
+ |
+} // namespace mojo |
+ |
+#endif // MOJO_PUBLIC_CPP_BINDINGS_INTERFACE_PTR_SET_H_ |