Add a ResourceScheduler to ResourceDispatcherHost.

content/browser/renderer_host/resource_dispatcher_host_impl.h

 // Copyright (c) 2012 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.
 
 // This is the browser side of the resource dispatcher, it receives requests
 // from the child process (i.e. [Renderer, Plugin, Worker]ProcessHost), and
 // dispatches them to URLRequests. It then forwards the messages from the
 // URLRequests back to the correct process for handling.
 //
 // See http://dev.chromium.org/developers/design-documents/multi-process-resource-loading
 
 #ifndef CONTENT_BROWSER_RENDERER_HOST_RESOURCE_DISPATCHER_HOST_IMPL_H_
 #define CONTENT_BROWSER_RENDERER_HOST_RESOURCE_DISPATCHER_HOST_IMPL_H_
 
 #include <map>
 #include <set>
 #include <string>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/gtest_prod_util.h"
 #include "base/memory/linked_ptr.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/time.h"
 #include "base/timer.h"
 #include "content/browser/download/download_resource_handler.h"
-#include "content/browser/renderer_host/resource_loader.h"
-#include "content/browser/renderer_host/resource_loader_delegate.h"
+#include "content/browser/renderer_host/resource_loader_impl.h"
+#include "content/browser/renderer_host/resource_loader_impl_delegate.h"
 #include "content/common/content_export.h"
 #include "content/public/browser/child_process_data.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/resource_dispatcher_host.h"
 #include "ipc/ipc_message.h"
 #include "net/cookies/canonical_cookie.h"
 #include "net/url_request/url_request.h"
 #include "webkit/glue/resource_type.h"
 
 class ResourceHandler;
 class SaveFileManager;
 struct ResourceHostMsg_Request;
 struct ViewMsg_SwapOut_Params;
 
 namespace net {
 class URLRequestJobFactory;
 }
 
 namespace webkit_blob {
 class ShareableFileReference;
 }
 
 namespace content {
 class ResourceContext;
 class ResourceDispatcherHostDelegate;
 class ResourceMessageFilter;
 class ResourceRequestInfoImpl;
+class ResourceScheduler;
 class WebContentsImpl;
 struct DownloadSaveInfo;
 struct GlobalRequestID;
 struct Referrer;
 
 class CONTENT_EXPORT ResourceDispatcherHostImpl
     : public ResourceDispatcherHost,
-      public ResourceLoaderDelegate {
+      public ResourceLoaderImplDelegate {
  public:
   ResourceDispatcherHostImpl();
   virtual ~ResourceDispatcherHostImpl();
 
   // Returns the current ResourceDispatcherHostImpl. May return NULL if it
   // hasn't been created yet.
   static ResourceDispatcherHostImpl* Get();
 
   // ResourceDispatcherHost implementation:
   virtual void SetDelegate(ResourceDispatcherHostDelegate* delegate) OVERRIDE;
@@ skipping 118 matching lines @@
   bool Send(IPC::Message* message);
 
   // Indicates whether third-party sub-content can pop-up HTTP basic auth
   // dialog boxes.
   bool allow_cross_origin_auth_prompt();
 
   ResourceDispatcherHostDelegate* delegate() {
     return delegate_;
   }
 
+  ResourceScheduler* scheduler() {
+    return resource_scheduler_.get();
+  }
+
   // Must be called after the ResourceRequestInfo has been created
   // and associated with the request.
   scoped_ptr<ResourceHandler> CreateResourceHandlerForDownload(
       net::URLRequest* request,
       bool is_content_initiated,
       scoped_ptr<DownloadSaveInfo> save_info,
       const DownloadResourceHandler::OnStartedCallback& started_cb);
 
   void ClearSSLClientAuthHandlerForRequest(net::URLRequest* request);
 
  private:
   FRIEND_TEST_ALL_PREFIXES(ResourceDispatcherHostTest,
                            TestBlockedRequestsProcessDies);
   FRIEND_TEST_ALL_PREFIXES(ResourceDispatcherHostTest,
                            IncrementOutstandingRequestsMemoryCost);
   FRIEND_TEST_ALL_PREFIXES(ResourceDispatcherHostTest,
                            CalculateApproximateMemoryCost);
 
   class ShutdownTask;
 
   friend class ShutdownTask;
 
-  // ResourceLoaderDelegate implementation:
+  // ResourceLoaderImplDelegate implementation:
   virtual ResourceDispatcherHostLoginDelegate* CreateLoginDelegate(
-      ResourceLoader* loader,
+      ResourceLoaderImpl* loader,
       net::AuthChallengeInfo* auth_info) OVERRIDE;
   virtual bool AcceptAuthRequest(
-      ResourceLoader* loader,
+      ResourceLoaderImpl* loader,
       net::AuthChallengeInfo* auth_info) OVERRIDE;
   virtual bool AcceptSSLClientCertificateRequest(
-      ResourceLoader* loader,
+      ResourceLoaderImpl* loader,
       net::SSLCertRequestInfo* cert_info) OVERRIDE;
-  virtual bool HandleExternalProtocol(ResourceLoader* loader,
+  virtual bool HandleExternalProtocol(ResourceLoaderImpl* loader,
                                       const GURL& url) OVERRIDE;
-  virtual void DidStartRequest(ResourceLoader* loader) OVERRIDE;
-  virtual void DidReceiveRedirect(ResourceLoader* loader,
+  virtual void DidStartRequest(ResourceLoaderImpl* loader) OVERRIDE;
+  virtual void DidReceiveRedirect(ResourceLoaderImpl* loader,
                                   const GURL& new_url) OVERRIDE;
-  virtual void DidReceiveResponse(ResourceLoader* loader) OVERRIDE;
-  virtual void DidFinishLoading(ResourceLoader* loader) OVERRIDE;
+  virtual void DidReceiveResponse(ResourceLoaderImpl* loader) OVERRIDE;
+  virtual void DidFinishLoading(ResourceLoaderImpl* loader) OVERRIDE;
 
   // Extracts the render view/process host's identifiers from the given request
   // and places them in the given out params (both required). If there are no
   // such IDs associated with the request (such as non-page-related requests),
   // this function will return false and both out params will be -1.
   static bool RenderViewForRequest(const net::URLRequest* request,
                                    int* render_process_host_id,
                                    int* render_view_host_id);
 
   // A shutdown helper that runs on the IO thread.
   void OnShutdown();
 
   // The real implementation of the OnSwapOutACK logic. OnSwapOutACK and
   // OnSimulateSwapOutACK just call this method, supplying the |timed_out|
   // parameter, which indicates whether the call is due to a timeout while
   // waiting for SwapOut acknowledgement from the renderer process.
   void HandleSwapOutACK(const ViewMsg_SwapOut_Params& params, bool timed_out);
 
   // Helper function for regular and download requests.
   void BeginRequestInternal(scoped_ptr<net::URLRequest> request,
                             scoped_ptr<ResourceHandler> handler);
 
   void StartLoading(ResourceRequestInfoImpl* info,
-                    const linked_ptr<ResourceLoader>& loader);
+                    const linked_ptr<ResourceLoaderImpl>& loader);
 
   // Updates the "cost" of outstanding requests for |child_id|.
   // The "cost" approximates how many bytes are consumed by all the in-memory
   // data structures supporting this request (net::URLRequest object,
   // HttpNetworkTransaction, etc...).
   // The value of |cost| is added to the running total, and the resulting
   // sum is returned.
   int IncrementOutstandingRequestsMemoryCost(int cost,
                                              int child_id);
 
   // Estimate how much heap space |request| will consume to run.
   static int CalculateApproximateMemoryCost(net::URLRequest* request);
 
   // The list of all requests that we have pending. This list is not really
   // optimized, and assumes that we have relatively few requests pending at once
   // since some operations require brute-force searching of the list.
   //
   // It may be enhanced in the future to provide some kind of prioritization
   // mechanism. We should also consider a hashtable or binary tree if it turns
   // out we have a lot of things here.
-  typedef std::map<GlobalRequestID, linked_ptr<ResourceLoader> > LoaderMap;
+  typedef std::map<GlobalRequestID, linked_ptr<ResourceLoaderImpl> > LoaderMap;
 
   // Deletes the pending request identified by the iterator passed in.
   // This function will invalidate the iterator passed in. Callers should
   // not rely on this iterator being valid on return.
   void RemovePendingLoader(const LoaderMap::iterator& iter);
 
   // Checks all pending requests and updates the load states and upload
   // progress if necessary.
   void UpdateLoadStates();
 
@@ skipping 40 matching lines @@
 
   HttpAuthResourceType HttpAuthResourceTypeOf(net::URLRequest* request);
 
   // Returns whether the URLRequest identified by |transferred_request_id| is
   // currently in the process of being transferred to a different renderer.
   // This happens if a request is redirected cross-site and needs to be resumed
   // by a new render view.
   bool IsTransferredNavigation(
       const GlobalRequestID& transferred_request_id) const;
 
-  ResourceLoader* GetLoader(const GlobalRequestID& id) const;
-  ResourceLoader* GetLoader(int child_id, int request_id) const;
+  ResourceLoaderImpl* GetLoader(const GlobalRequestID& id) const;
+  ResourceLoaderImpl* GetLoader(int child_id, int request_id) const;
 
   LoaderMap pending_loaders_;
 
   // Collection of temp files downloaded for child processes via
   // the download_to_file mechanism. We avoid deleting them until
   // the client no longer needs them.
   typedef std::map<int, scoped_refptr<webkit_blob::ShareableFileReference> >
       DeletableFilesMap;  // key is request id
   typedef std::map<int, DeletableFilesMap>
       RegisteredTempFiles;  // key is child process id
@@ skipping 12 matching lines @@
   // start at -2 and go down from there. (We need to start at -2 because -1 is
   // used as a special value all over the resource_dispatcher_host for
   // uninitialized variables.) This way, we no longer have the unlikely (but
   // observed in the real world!) event where we have two requests with the same
   // request_id_.
   int request_id_;
 
   // True if the resource dispatcher host has been shut down.
   bool is_shutdown_;
 
-  typedef std::vector<linked_ptr<ResourceLoader> > BlockedLoadersList;
+  typedef std::vector<linked_ptr<ResourceLoaderImpl> > BlockedLoadersList;
   typedef std::pair<int, int> ProcessRouteIDs;
   typedef std::map<ProcessRouteIDs, BlockedLoadersList*> BlockedLoadersMap;
   BlockedLoadersMap blocked_loaders_map_;
 
   // Maps the child_ids to the approximate number of bytes
   // being used to service its resource requests. No entry implies 0 cost.
   typedef std::map<int, int> OutstandingRequestsMemoryCostMap;
   OutstandingRequestsMemoryCostMap outstanding_requests_memory_cost_map_;
 
   // |max_outstanding_requests_cost_per_process_| is the upper bound on how
@@ skipping 15 matching lines @@
   ResourceMessageFilter* filter_;
 
   ResourceDispatcherHostDelegate* delegate_;
 
   bool allow_cross_origin_auth_prompt_;
 
   // http://crbug.com/90971 - Assists in tracking down use-after-frees on
   // shutdown.
   std::set<const ResourceContext*> active_resource_contexts_;
 
+  scoped_ptr<ResourceScheduler> resource_scheduler_;
+
   DISALLOW_COPY_AND_ASSIGN(ResourceDispatcherHostImpl);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_RENDERER_HOST_RESOURCE_DISPATCHER_HOST_IMPL_H_