| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_REGISTRY_H_ | 5 #ifndef CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_REGISTRY_H_ |
| 6 #define CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_REGISTRY_H_ | 6 #define CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_REGISTRY_H_ |
| 7 | 7 |
| 8 #include <map> | 8 #include <map> |
| 9 #include <memory> | 9 #include <memory> |
| 10 #include <set> | 10 #include <set> |
| (...skipping 91 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 102 // Returns an embedded worker instance for given |embedded_worker_id|. | 102 // Returns an embedded worker instance for given |embedded_worker_id|. |
| 103 EmbeddedWorkerInstance* GetWorker(int embedded_worker_id); | 103 EmbeddedWorkerInstance* GetWorker(int embedded_worker_id); |
| 104 | 104 |
| 105 // Returns true if |embedded_worker_id| is managed by this registry. | 105 // Returns true if |embedded_worker_id| is managed by this registry. |
| 106 bool CanHandle(int embedded_worker_id) const; | 106 bool CanHandle(int embedded_worker_id) const; |
| 107 | 107 |
| 108 MessagePortMessageFilter* MessagePortMessageFilterForProcess(int process_id); | 108 MessagePortMessageFilter* MessagePortMessageFilterForProcess(int process_id); |
| 109 | 109 |
| 110 private: | 110 private: |
| 111 friend class base::RefCounted<EmbeddedWorkerRegistry>; | 111 friend class base::RefCounted<EmbeddedWorkerRegistry>; |
| 112 friend class MojoEmbeddedWorkerInstanceTest; |
| 112 friend class EmbeddedWorkerInstance; | 113 friend class EmbeddedWorkerInstance; |
| 113 friend class EmbeddedWorkerInstanceTest; | 114 friend class EmbeddedWorkerInstanceTest; |
| 114 FRIEND_TEST_ALL_PREFIXES(EmbeddedWorkerInstanceTest, | 115 FRIEND_TEST_ALL_PREFIXES(EmbeddedWorkerInstanceTest, |
| 115 RemoveWorkerInSharedProcess); | 116 RemoveWorkerInSharedProcess); |
| 117 FRIEND_TEST_ALL_PREFIXES(MojoEmbeddedWorkerInstanceTest, |
| 118 RemoveWorkerInSharedProcess); |
| 116 | 119 |
| 117 using WorkerInstanceMap = std::map<int, EmbeddedWorkerInstance*>; | 120 using WorkerInstanceMap = std::map<int, EmbeddedWorkerInstance*>; |
| 118 using ProcessToSenderMap = std::map<int, IPC::Sender*>; | 121 using ProcessToSenderMap = std::map<int, IPC::Sender*>; |
| 119 using ProcessToMessagePortMessageFilterMap = | 122 using ProcessToMessagePortMessageFilterMap = |
| 120 std::map<int, MessagePortMessageFilter*>; | 123 std::map<int, MessagePortMessageFilter*>; |
| 121 | 124 |
| 122 EmbeddedWorkerRegistry( | 125 EmbeddedWorkerRegistry( |
| 123 const base::WeakPtr<ServiceWorkerContextCore>& context, | 126 const base::WeakPtr<ServiceWorkerContextCore>& context, |
| 124 int initial_embedded_worker_id); | 127 int initial_embedded_worker_id); |
| 125 ~EmbeddedWorkerRegistry(); | 128 ~EmbeddedWorkerRegistry(); |
| 126 | 129 |
| 127 ServiceWorkerStatusCode Send(int process_id, IPC::Message* message); | 130 ServiceWorkerStatusCode Send(int process_id, IPC::Message* message); |
| 128 | 131 |
| 129 // Called when EmbeddedWorkerInstance gets ready for IPC. This function | 132 // Called when EmbeddedWorkerInstance gets ready for IPC. This function |
| 130 // prepares a route to the child worker thread. | 133 // prepares a route to the child worker thread. |
| 131 // TODO(shimazu): Remove this function once mojofication is completed. | 134 // TODO(shimazu): Remove this function once mojofication is completed. |
| 132 ServiceWorkerStatusCode AddWorker(int process_id, int embedded_worker_id); | 135 ServiceWorkerStatusCode AddWorker(int process_id, int embedded_worker_id); |
| 133 | 136 |
| 134 // RemoveWorker is called when EmbeddedWorkerInstance is destructed. | 137 // RemoveWorker is called when EmbeddedWorkerInstance is destructed. |
| 135 // |process_id| could be invalid (i.e. ChildProcessHost::kInvalidUniqueID) | 138 // |process_id| could be invalid (i.e. ChildProcessHost::kInvalidUniqueID) |
| 136 // if it's not running. | 139 // if it's not running. |
| 137 void RemoveWorker(int process_id, int embedded_worker_id); | 140 void RemoveWorker(int process_id, int embedded_worker_id); |
| 141 // DetachWorker is called when EmbeddedWorkerInstance releases a process. |
| 142 // |process_id| could be invalid (i.e. ChildProcessHost::kInvalidUniqueID) |
| 143 // if it's not running. |
| 144 void DetachWorker(int process_id, int embedded_worker_id); |
| 138 | 145 |
| 139 EmbeddedWorkerInstance* GetWorkerForMessage(int process_id, | 146 EmbeddedWorkerInstance* GetWorkerForMessage(int process_id, |
| 140 int embedded_worker_id); | 147 int embedded_worker_id); |
| 141 | 148 |
| 142 base::WeakPtr<ServiceWorkerContextCore> context_; | 149 base::WeakPtr<ServiceWorkerContextCore> context_; |
| 143 | 150 |
| 144 WorkerInstanceMap worker_map_; | 151 WorkerInstanceMap worker_map_; |
| 145 ProcessToSenderMap process_sender_map_; | 152 ProcessToSenderMap process_sender_map_; |
| 146 ProcessToMessagePortMessageFilterMap process_message_port_message_filter_map_; | 153 ProcessToMessagePortMessageFilterMap process_message_port_message_filter_map_; |
| 147 | 154 |
| 148 // Map from process_id to embedded_worker_id. | 155 // Map from process_id to embedded_worker_id. |
| 149 // This map only contains starting and running workers. | 156 // This map only contains starting and running workers. |
| 150 std::map<int, std::set<int> > worker_process_map_; | 157 std::map<int, std::set<int> > worker_process_map_; |
| 151 | 158 |
| 152 int next_embedded_worker_id_; | 159 int next_embedded_worker_id_; |
| 153 const int initial_embedded_worker_id_; | 160 const int initial_embedded_worker_id_; |
| 154 | 161 |
| 155 DISALLOW_COPY_AND_ASSIGN(EmbeddedWorkerRegistry); | 162 DISALLOW_COPY_AND_ASSIGN(EmbeddedWorkerRegistry); |
| 156 }; | 163 }; |
| 157 | 164 |
| 158 } // namespace content | 165 } // namespace content |
| 159 | 166 |
| 160 #endif // CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_REGISTRY_H_ | 167 #endif // CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_REGISTRY_H_ |
| OLD | NEW |