Add SharedWorkerServiceImpl::CheckWorkerDependency().

content/browser/shared_worker/shared_worker_service_impl.h

 // 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 <set>
+
 #include "base/compiler_specific.h"
 #include "base/containers/scoped_ptr_hash_map.h"
 #include "base/memory/singleton.h"
 #include "base/observer_list.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "content/public/browser/worker_service.h"
 
 struct ViewHostMsg_CreateWorker_Params;
 
@@ skipping 60 matching lines @@
                        SharedWorkerMessageFilter* filter);
   void AllowIndexedDB(int worker_route_id,
                       const GURL& url,
                       const base::string16& name,
                       bool* result,
                       SharedWorkerMessageFilter* filter);
 
   void OnSharedWorkerMessageFilterClosing(
       SharedWorkerMessageFilter* filter);
 
+  // Checks the worker dependency of renderer processes and calls
+  // IncrementWorkerRefCount and DecrementWorkerRefCount of
+  // RenderProcessHostImpl on UI thread if necessary.
+  void CheckWorkerDependency();
+
  private:
   friend struct DefaultSingletonTraits<SharedWorkerServiceImpl>;
+  friend class SharedWorkerServiceImplTest;
+
+  typedef void (*UpdateWorkerDependencyFunc)(const std::vector<int>&,
+                                             const std::vector<int>&);
 
   SharedWorkerServiceImpl();
   virtual ~SharedWorkerServiceImpl();
 
+  void ResetForTesting();
+
   SharedWorkerHost* FindSharedWorkerHost(
       SharedWorkerMessageFilter* filter,
       int worker_route_id);
 
   SharedWorkerInstance* FindSharedWorkerInstance(
       const GURL& url,
       const base::string16& name,
       const WorkerStoragePartition& worker_partition,
       ResourceContext* resource_context);
 
+  // Returns the IDs of the renderer processes which are executing
+  // SharedWorkers connected to other renderer processes.
+  const std::set<int> GetRenderersWithWorkerDependency();
+
+  void ChangeUpdateWorkerDependencyFuncForTesting(
+      UpdateWorkerDependencyFunc new_func);
+
+  std::set<int> last_worker_depended_renderers_;
+  UpdateWorkerDependencyFunc update_worker_dependency_;
+
   // Pair of render_process_id and worker_route_id.
   typedef std::pair<int, int> ProcessRouteIdPair;
   typedef base::ScopedPtrHashMap<ProcessRouteIdPair,
                                  SharedWorkerHost> WorkerHostMap;
   WorkerHostMap worker_hosts_;
 
   ObserverList<WorkerServiceObserver> observers_;
 
   DISALLOW_COPY_AND_ASSIGN(SharedWorkerServiceImpl);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_SHARED_WORKER_SHARED_WORKER_SERVICE_IMPL_H_