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 62 matching lines @@
                       const GURL& url,
                       const base::string16& name,
                       bool* result,
                       SharedWorkerMessageFilter* filter);
 
   void OnSharedWorkerMessageFilterClosing(
       SharedWorkerMessageFilter* filter);
 
  private:
   friend struct DefaultSingletonTraits<SharedWorkerServiceImpl>;
+  friend class SharedWorkerServiceImplTest;
+
+  typedef void (*UpdateWorkerDependencyFunc)(const std::vector<int>,
+                                             const std::vector<int>);

[kinuko, 2014/03/10 04:25:43]

const vector<int>& ?

[horo, 2014/03/10 11:48:31]

Done.

+
+  class CONTENT_EXPORT ScopedWorkerDependencyChecker {

[kinuko, 2014/03/10 04:25:43]

Does this class need to be exported?  In a quick look its definition can be
hidden in .cc.

[horo, 2014/03/10 11:48:31]

Done.
Moved to .cc.

+   public:
+    explicit ScopedWorkerDependencyChecker(SharedWorkerServiceImpl* service)
+        : service_(service) {}
+    ~ScopedWorkerDependencyChecker() { service_->CheckWorkerDependency(); }
+
+   private:
+    SharedWorkerServiceImpl* service_;
+    DISALLOW_COPY_AND_ASSIGN(ScopedWorkerDependencyChecker);
+  };
 
   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();
+
+  // Checks the worker dependency of renderer processes and calls
+  // IncrementWorkerRefCount and DecrementWorkerRefCount of
+  // RenderProcessHostImpl on UI thread if necessary.
+  void CheckWorkerDependency();
+
+  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_