Introduce ServiceWorkerHandle for tracking WebServiceWorkerImpl reference

content/browser/service_worker/service_worker_dispatcher_host.h

 // Copyright 2013 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_SERVICE_WORKER_SERVICE_WORKER_DISPATCHER_HOST_H_
 #define CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_DISPATCHER_HOST_H_
 
+#include "base/id_map.h"
 #include "base/memory/weak_ptr.h"
 #include "base/strings/string16.h"
 #include "content/browser/service_worker/service_worker_registration_status.h"
 #include "content/public/browser/browser_message_filter.h"
 
 class GURL;
 
 namespace content {
 
 class MessagePortMessageFilter;
 class ServiceWorkerContextCore;
 class ServiceWorkerContextWrapper;
+class ServiceWorkerHandle;
 class ServiceWorkerProviderHost;
 class ServiceWorkerRegistration;
 
 class CONTENT_EXPORT ServiceWorkerDispatcherHost : public BrowserMessageFilter {
  public:
   ServiceWorkerDispatcherHost(
       int render_process_id,
       MessagePortMessageFilter* message_port_message_filter);
 
   void Init(ServiceWorkerContextWrapper* context_wrapper);
 
   // BrowserIOMessageFilter implementation
   virtual void OnDestruct() const OVERRIDE;
   virtual bool OnMessageReceived(const IPC::Message& message,
                                  bool* message_was_ok) OVERRIDE;
 
+  // Returns a new handle id.
+  int RegisterServiceWorkerHandle(scoped_ptr<ServiceWorkerHandle> handle);
+
  protected:
   virtual ~ServiceWorkerDispatcherHost();
 
  private:
   friend class BrowserThread;
   friend class base::DeleteHelper<ServiceWorkerDispatcherHost>;
   friend class TestingServiceWorkerDispatcherHost;
 
   // IPC Message handlers
   void OnRegisterServiceWorker(int32 thread_id,
@@ skipping 12 matching lines @@
                        int embedded_worker_id);
   void OnWorkerStopped(int embedded_worker_id);
   void OnSendMessageToBrowser(int embedded_worker_id,
                               int request_id,
                               const IPC::Message& message);
   void OnReportException(int embedded_worker_id,
                          const base::string16& error_message,
                          int line_number,
                          int column_number,
                          const GURL& source_url);
-  void OnPostMessage(int64 version_id,
+  void OnPostMessage(int handle_id,
                      const base::string16& message,
                      const std::vector<int>& sent_message_port_ids);
+  void OnServiceWorkerObjectDestroyed(int handle_id);
 
   // Callbacks from ServiceWorkerContextCore
   void RegistrationComplete(int32 thread_id,
                             int32 request_id,
                             ServiceWorkerStatusCode status,
                             int64 registration_id,
                             int64 version_id);
 
   void UnregistrationComplete(int32 thread_id,
                               int32 request_id,
                               ServiceWorkerStatusCode status);
 
   void SendRegistrationError(int32 thread_id,
                              int32 request_id,
                              ServiceWorkerStatusCode status);
 
   int render_process_id_;
   MessagePortMessageFilter* const message_port_message_filter_;
+  base::WeakPtr<ServiceWorkerContextCore> context_;
 
-  base::WeakPtr<ServiceWorkerContextCore> context_;
+  IDMap<ServiceWorkerHandle, IDMapOwnPointer> handles_;
+
+  DISALLOW_COPY_AND_ASSIGN(ServiceWorkerDispatcherHost);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_DISPATCHER_HOST_H_