Pass calls to GetDocuments through the scheduler

chrome/browser/chromeos/drive/drive_feed_loader.cc

 // 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.
 
 #include "chrome/browser/chromeos/drive/drive_feed_loader.h"
 
 #include <set>
 
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/format_macros.h"
 #include "base/json/json_reader.h"
 #include "base/json/json_writer.h"
 #include "base/metrics/histogram.h"
 #include "base/stringprintf.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "base/values.h"
 #include "chrome/browser/chromeos/drive/drive_cache.h"
 #include "chrome/browser/chromeos/drive/drive_feed_loader_observer.h"
 #include "chrome/browser/chromeos/drive/drive_feed_processor.h"
 #include "chrome/browser/chromeos/drive/drive_file_system_util.h"
+#include "chrome/browser/chromeos/drive/drive_scheduler.h"
 #include "chrome/browser/chromeos/drive/drive_webapps_registry.h"
 #include "chrome/browser/google_apis/drive_api_parser.h"
 #include "chrome/browser/google_apis/drive_api_util.h"
 #include "chrome/browser/google_apis/drive_service_interface.h"
 #include "chrome/common/chrome_switches.h"
 #include "content/public/browser/browser_thread.h"
 
 using content::BrowserThread;
 
 namespace drive {
@@ skipping 215 matching lines @@
 
   // Time elapsed since the feed fetching was started.
   base::TimeDelta feed_fetching_elapsed_time;
 
   base::WeakPtrFactory<GetDocumentsUiState> weak_ptr_factory;
 };
 
 DriveFeedLoader::DriveFeedLoader(
     DriveResourceMetadata* resource_metadata,
     google_apis::DriveServiceInterface* drive_service,
+    DriveScheduler* scheduler,
     DriveWebAppsRegistryInterface* webapps_registry,
     DriveCache* cache,
     scoped_refptr<base::SequencedTaskRunner> blocking_task_runner)
     : resource_metadata_(resource_metadata),
       drive_service_(drive_service),
+      scheduler_(scheduler),
       webapps_registry_(webapps_registry),
       cache_(cache),
       blocking_task_runner_(blocking_task_runner),
       refreshing_(false),
       ALLOW_THIS_IN_INITIALIZER_LIST(weak_ptr_factory_(this)) {
 }
 
 DriveFeedLoader::~DriveFeedLoader() {
 }
 
@@ skipping 126 matching lines @@
 }
 
 void DriveFeedLoader::LoadFromServer(scoped_ptr<LoadFeedParams> params) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   const base::TimeTicks start_time = base::TimeTicks::Now();
 
   // base::Passed() may get evaluated first, so get a pointer to params.
   LoadFeedParams* params_ptr = params.get();
   if (google_apis::util::IsDriveV2ApiEnabled()) {
-    drive_service_->GetDocuments(
+    scheduler_->GetDocuments(
         params_ptr->feed_to_load,
         params_ptr->start_changestamp,
         std::string(),  // No search query.
         params_ptr->shared_with_me,
         std::string(),  // No directory resource ID.
         base::Bind(&DriveFeedLoader::OnGetChangelist,
                    weak_ptr_factory_.GetWeakPtr(),
                    base::Passed(&params),
                    start_time));
   } else {
-    drive_service_->GetDocuments(
+    scheduler_->GetDocuments(
         params_ptr->feed_to_load,
         params_ptr->start_changestamp,
         params_ptr->search_query,
         params_ptr->shared_with_me,
         params_ptr->directory_resource_id,
         base::Bind(&DriveFeedLoader::OnGetDocuments,
                    weak_ptr_factory_.GetWeakPtr(),
                    base::Passed(&params),
                    start_time));
   }
@@ skipping 122 matching lines @@
                      weak_ptr_factory_.GetWeakPtr(),
                      ui_state->weak_ptr_factory.GetWeakPtr()));
     }
     ui_state->num_fetched_documents = num_accumulated_entries;
     ui_state->feed_fetching_elapsed_time = base::TimeTicks::Now() - start_time;
 
     // |params| will be passed to the callback and thus nulled. Extract the
     // pointer so we can use it bellow.
     LoadFeedParams* params_ptr = params.get();
     // Kick off the remaining part of the feeds.
-    drive_service_->GetDocuments(
+    scheduler_->GetDocuments(
         next_feed_url,
         params_ptr->start_changestamp,
         params_ptr->search_query,
         params_ptr->shared_with_me,
         params_ptr->directory_resource_id,
         base::Bind(&DriveFeedLoader::OnGetDocuments,
                    weak_ptr_factory_.GetWeakPtr(),
                    base::Passed(&params),
                    start_time));
     return;
@@ skipping 82 matching lines @@
           base::Bind(&DriveFeedLoader::OnNotifyDocumentFeedFetched,
                      weak_ptr_factory_.GetWeakPtr(),
                      ui_state->weak_ptr_factory.GetWeakPtr()));
     }
     ui_state->num_fetched_documents = num_accumulated_entries;
     ui_state->feed_fetching_elapsed_time = base::TimeTicks::Now() - start_time;
 
     // Kick off the remaining part of the feeds.
     // Extract the pointer so we can use it bellow.
     LoadFeedParams* params_ptr = params.get();
-    drive_service_->GetDocuments(
+    scheduler_->GetDocuments(
         current_feed->next_link(),
         params_ptr->start_changestamp,
         std::string(),  // No search query.
         false,  // Not shared with me.
         std::string(),  // No directory resource ID.
         base::Bind(&DriveFeedLoader::OnGetChangelist,
                    weak_ptr_factory_.GetWeakPtr(),
                    base::Passed(&params),
                    start_time));
     return;
@@ skipping 217 matching lines @@
 
   // Run the callback now that the filesystem is ready.
   load_finished_callback.Run(DRIVE_FILE_OK);
 
   FOR_EACH_OBSERVER(DriveFeedLoaderObserver,
                     observers_,
                     OnFeedFromServerLoaded());
 }
 
 }  // namespace drive