content/browser/service_worker/embedded_worker_registry.h

   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
   3 // found in the LICENSE file.
   4
   5 #ifndef CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_REGISTRY_H_
   6 #define CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_REGISTRY_H_
   7
   8 #include <map>
   9 #include <set>
  10 #include <vector>
  11
  12 #include "base/basictypes.h"
  13 #include "base/memory/ref_counted.h"
  14 #include "base/memory/scoped_ptr.h"
  15 #include "base/memory/weak_ptr.h"
  16 #include "base/strings/string16.h"
  17 #include "content/common/content_export.h"
  18 #include "content/common/service_worker/service_worker_status_code.h"
  19
  20 struct EmbeddedWorkerMsg_StartWorker_Params;
  21 class GURL;
  22
  23 namespace IPC {
  24 class Message;
  25 class Sender;
  26 }
  27
  28 namespace content {
  29
  30 class EmbeddedWorkerInstance;
  31 class ServiceWorkerContextCore;
  32
  33 // Acts as a thin stub between MessageFilter and each EmbeddedWorkerInstance,
  34 // which sends/receives messages to/from each EmbeddedWorker in child process.
  35 //
  36 // Hangs off ServiceWorkerContextCore (its reference is also held by each
  37 // EmbeddedWorkerInstance).  Operated only on IO thread.
  38 class CONTENT_EXPORT EmbeddedWorkerRegistry
  39     : public NON_EXPORTED_BASE(base::RefCounted<EmbeddedWorkerRegistry>) {
  40  public:
  41   typedef base::Callback<void(ServiceWorkerStatusCode)> StatusCallback;
  42
  43   explicit EmbeddedWorkerRegistry(
  44       base::WeakPtr<ServiceWorkerContextCore> context);
  45
  46   bool OnMessageReceived(const IPC::Message& message);
  47
  48   // Creates and removes a new worker instance entry for bookkeeping.
  49   // This doesn't actually start or stop the worker.
  50   scoped_ptr<EmbeddedWorkerInstance> CreateWorker();
  51
  52   // Called from EmbeddedWorkerInstance, relayed to the child process.
  53   void StartWorker(const std::vector<int>& process_ids,
  54                    int embedded_worker_id,
  55                    int64 service_worker_version_id,
  56                    const GURL& scope,
  57                    const GURL& script_url,
  58                    const StatusCallback& callback);
  59   ServiceWorkerStatusCode StopWorker(int process_id,
  60                                      int embedded_worker_id);
  61
  62   // Stop all active workers, even if they're handling events.
  63   void Shutdown();
  64
  65   // Called back from EmbeddedWorker in the child process, relayed via
  66   // ServiceWorkerDispatcherHost.
  67   void OnWorkerScriptLoaded(int process_id, int embedded_worker_id);
  68   void OnWorkerScriptLoadFailed(int process_id, int embedded_worker_id);
  69   void OnWorkerStarted(int process_id, int thread_id, int embedded_worker_id);
  70   void OnWorkerStopped(int process_id, int embedded_worker_id);
  71   void OnReportException(int embedded_worker_id,
  72                          const base::string16& error_message,
  73                          int line_number,
  74                          int column_number,
  75                          const GURL& source_url);
  76   void OnReportConsoleMessage(int embedded_worker_id,
  77                               int source_identifier,
  78                               int message_level,
  79                               const base::string16& message,
  80                               int line_number,
  81                               const GURL& source_url);
  82
  83   // Keeps a map from process_id to sender information.
  84   void AddChildProcessSender(int process_id, IPC::Sender* sender);
  85   void RemoveChildProcessSender(int process_id);
  86
  87   // Returns an embedded worker instance for given |embedded_worker_id|.
  88   EmbeddedWorkerInstance* GetWorker(int embedded_worker_id);
  89
  90  private:
  91   friend class base::RefCounted<EmbeddedWorkerRegistry>;
  92   friend class EmbeddedWorkerInstance;
  93
  94   typedef std::map<int, EmbeddedWorkerInstance*> WorkerInstanceMap;
  95   typedef std::map<int, IPC::Sender*> ProcessToSenderMap;
  96
  97   ~EmbeddedWorkerRegistry();
  98
  99   void StartWorkerWithProcessId(
 100       int embedded_worker_id,
 101       scoped_ptr<EmbeddedWorkerMsg_StartWorker_Params> params,
 102       const StatusCallback& callback,
 103       ServiceWorkerStatusCode status,
 104       int process_id);
 105
 106   ServiceWorkerStatusCode Send(int process_id, IPC::Message* message);
 107
 108   // RemoveWorker is called when EmbeddedWorkerInstance is destructed.
 109   // |process_id| could be invalid (i.e. -1) if it's not running.
 110   void RemoveWorker(int process_id, int embedded_worker_id);
 111
 112   base::WeakPtr<ServiceWorkerContextCore> context_;
 113
 114   WorkerInstanceMap worker_map_;
 115   ProcessToSenderMap process_sender_map_;
 116
 117   // Map from process_id to embedded_worker_id.
 118   // This map only contains running workers.
 119   std::map<int, std::set<int> > worker_process_map_;
 120
 121   int next_embedded_worker_id_;
 122
 123   DISALLOW_COPY_AND_ASSIGN(EmbeddedWorkerRegistry);
 124 };
 125
 126 }  // namespace content
 127
 128 #endif  // CONTENT_BROWSER_SERVICE_WORKER_EMBEDDED_WORKER_REGISTRY_H_