| // 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 CONTENT_BROWSER_SHARED_WORKER_SHARED_WORKER_SERVICE_IMPL_H_ |
| #define CONTENT_BROWSER_SHARED_WORKER_SHARED_WORKER_SERVICE_IMPL_H_ |
| |
| #include <map> |
| #include <memory> |
| #include <set> |
| #include <utility> |
| #include <vector> |
| |
| #include "base/compiler_specific.h" |
| #include "base/macros.h" |
| #include "base/memory/singleton.h" |
| #include "content/browser/shared_worker/shared_worker_host.h" |
| #include "content/common/shared_worker/shared_worker_connector.mojom.h" |
| #include "content/common/shared_worker/shared_worker_factory.mojom.h" |
| #include "content/public/browser/shared_worker_service.h" |
| |
| namespace blink { |
| class MessagePortChannel; |
| } |
| |
| namespace content { |
| |
| class SharedWorkerInstance; |
| class SharedWorkerHost; |
| class StoragePartition; |
| class ResourceContext; |
| class WorkerStoragePartitionId; |
| |
| class CONTENT_EXPORT SharedWorkerServiceImpl : public SharedWorkerService { |
| public: |
| // SharedWorkerService implementation. |
| bool TerminateWorker(const GURL& url, |
| const std::string& name, |
| const url::Origin& constructor_origin, |
| StoragePartition* storage_partition, |
| ResourceContext* resource_context) override; |
| |
| // Terminates the given worker. Returns true if the process was found. |
| bool TerminateWorkerById(int process_id, int route_id); |
| void TerminateAllWorkersForTesting(base::OnceClosure callback); |
| |
| // Creates the worker if necessary or connects to an already existing worker. |
| void ConnectToWorker( |
| int process_id, |
| int frame_id, |
| mojom::SharedWorkerInfoPtr info, |
| mojom::SharedWorkerClientPtr client, |
| blink::mojom::SharedWorkerCreationContextType creation_context_type, |
| const blink::MessagePortChannel& port, |
| ResourceContext* resource_context, |
| const WorkerStoragePartitionId& partition_id); |
| |
| void DestroyHost(int process_id, int route_id); |
| |
| private: |
| friend struct base::DefaultSingletonTraits<SharedWorkerServiceImpl>; |
| friend class SharedWorkerServiceImplTest; |
| friend class SharedWorkerService; |
| |
| using WorkerID = std::pair<int /* process_id */, int /* route_id */>; |
| using WorkerHostMap = std::map<WorkerID, std::unique_ptr<SharedWorkerHost>>; |
| |
| SharedWorkerServiceImpl(); |
| ~SharedWorkerServiceImpl() override; |
| |
| void ResetForTesting(); |
| |
| void CreateWorker(std::unique_ptr<SharedWorkerInstance> instance, |
| mojom::SharedWorkerClientPtr client, |
| int process_id, |
| int frame_id, |
| const blink::MessagePortChannel& message_port); |
| |
| // Returns nullptr if there is no such host. |
| SharedWorkerHost* FindSharedWorkerHost(int process_id, int route_id); |
| SharedWorkerHost* FindAvailableSharedWorkerHost( |
| const SharedWorkerInstance& instance); |
| |
| WorkerHostMap worker_hosts_; |
| base::OnceClosure terminate_all_workers_callback_; |
| |
| DISALLOW_COPY_AND_ASSIGN(SharedWorkerServiceImpl); |
| }; |
| |
| } // namespace content |
| |
| #endif // CONTENT_BROWSER_SHARED_WORKER_SHARED_WORKER_SERVICE_IMPL_H_ |