Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(82)

Unified Diff: content/browser/download/save_package.cc

Issue 2890853002: Downloads: replace BrowserThread::FILE with task scheduler. (Closed)
Patch Set: Address Avi's comments. Created 3 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « content/browser/download/save_package.h ('k') | content/public/browser/BUILD.gn » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: content/browser/download/save_package.cc
diff --git a/content/browser/download/save_package.cc b/content/browser/download/save_package.cc
index b6b6bd454ff1ab1cffd36f55fa8f70cc32dbe66c..fc2f04b810624d7dabf804d659b9bb343e2e5bd6 100644
--- a/content/browser/download/save_package.cc
+++ b/content/browser/download/save_package.cc
@@ -20,13 +20,16 @@
#include "base/strings/string_split.h"
#include "base/strings/sys_string_conversions.h"
#include "base/strings/utf_string_conversions.h"
+#include "base/task_runner_util.h"
#include "base/threading/thread.h"
+#include "base/threading/thread_restrictions.h"
#include "build/build_config.h"
#include "components/url_formatter/url_formatter.h"
#include "content/browser/bad_message.h"
#include "content/browser/download/download_item_impl.h"
#include "content/browser/download/download_manager_impl.h"
#include "content/browser/download/download_stats.h"
+#include "content/browser/download/download_task_runner.h"
#include "content/browser/download/save_file.h"
#include "content/browser/download/save_file_manager.h"
#include "content/browser/download/save_item.h"
@@ -600,8 +603,8 @@ bool SavePackage::UpdateSaveProgress(SaveItemId save_item_id,
return true;
}
-// Stop all page saving jobs that are in progress and instruct the FILE thread
-// to delete all saved files.
+// Stop all page saving jobs that are in progress and instruct the download
+// sequence to delete all saved files.
void SavePackage::Stop(bool cancel_download_item) {
DCHECK_CURRENTLY_ON(BrowserThread::UI);
// If we haven't moved out of the initial state, there's nothing to cancel and
@@ -632,10 +635,9 @@ void SavePackage::Stop(bool cancel_download_item) {
for (const auto& it : saved_failed_items_)
save_item_ids.push_back(it.first);
- BrowserThread::PostTask(
- BrowserThread::FILE, FROM_HERE,
- base::Bind(&SaveFileManager::RemoveSavedFileFromFileMap, file_manager_,
- save_item_ids));
+ GetDownloadTaskRunner()->PostTask(
+ FROM_HERE, base::Bind(&SaveFileManager::RemoveSavedFileFromFileMap,
+ file_manager_, save_item_ids));
finished_ = true;
wait_state_ = FAILED;
@@ -661,15 +663,11 @@ void SavePackage::CheckFinish() {
for (const auto& it : saved_success_items_)
final_names.insert(std::make_pair(it.first, it.second->full_path()));
- BrowserThread::PostTask(
- BrowserThread::FILE, FROM_HERE,
- base::Bind(&SaveFileManager::RenameAllFiles,
- file_manager_,
- final_names,
- dir,
- web_contents()->GetRenderProcessHost()->GetID(),
- web_contents()->GetMainFrame()->GetRoutingID(),
- id()));
+ GetDownloadTaskRunner()->PostTask(
+ FROM_HERE,
+ base::Bind(&SaveFileManager::RenameAllFiles, file_manager_, final_names,
+ dir, web_contents()->GetRenderProcessHost()->GetID(),
+ web_contents()->GetMainFrame()->GetRoutingID(), id()));
}
// Successfully finished all items of this SavePackage.
@@ -701,10 +699,9 @@ void SavePackage::Finish() {
list_of_failed_save_item_ids.push_back(save_item->id());
}
- BrowserThread::PostTask(
- BrowserThread::FILE, FROM_HERE,
- base::Bind(&SaveFileManager::RemoveSavedFileFromFileMap, file_manager_,
- list_of_failed_save_item_ids));
+ GetDownloadTaskRunner()->PostTask(
+ FROM_HERE, base::Bind(&SaveFileManager::RemoveSavedFileFromFileMap,
+ file_manager_, list_of_failed_save_item_ids));
if (download_) {
if (save_type_ != SAVE_PAGE_TYPE_AS_MHTML) {
@@ -768,8 +765,8 @@ void SavePackage::SaveFinished(SaveItemId save_item_id,
void SavePackage::SaveCanceled(const SaveItem* save_item) {
DCHECK_CURRENTLY_ON(BrowserThread::UI);
file_manager_->RemoveSaveFile(save_item->id(), this);
- BrowserThread::PostTask(
- BrowserThread::FILE, FROM_HERE,
+ GetDownloadTaskRunner()->PostTask(
+ FROM_HERE,
base::Bind(&SaveFileManager::CancelSave, file_manager_, save_item->id()));
}
@@ -929,10 +926,9 @@ void SavePackage::GetSerializedHtmlWithLocalLinks() {
number_of_frames_pending_response_++;
} else {
// Notify SaveFileManager about the failure to save this SaveItem.
- BrowserThread::PostTask(
- BrowserThread::FILE, FROM_HERE,
- base::Bind(&SaveFileManager::SaveFinished, file_manager_,
- save_item->id(), id(), false));
+ GetDownloadTaskRunner()->PostTask(
+ FROM_HERE, base::Bind(&SaveFileManager::SaveFinished, file_manager_,
+ save_item->id(), id(), false));
}
}
if (number_of_frames_pending_response_ == 0) {
@@ -1042,22 +1038,21 @@ void SavePackage::OnSerializedHtmlWithLocalLinksResponse(
scoped_refptr<net::IOBuffer> new_data(new net::IOBuffer(data.size()));
memcpy(new_data->data(), data.data(), data.size());
- // Call write file functionality in FILE thread.
- BrowserThread::PostTask(
- BrowserThread::FILE, FROM_HERE,
+ // Call write file functionality in download sequence.
+ GetDownloadTaskRunner()->PostTask(
+ FROM_HERE,
base::Bind(&SaveFileManager::UpdateSaveProgress, file_manager_,
save_item->id(), base::RetainedRef(new_data),
static_cast<int>(data.size())));
}
- // Current frame is completed saving, call finish in FILE thread.
+ // Current frame is completed saving, call finish in download sequence.
if (end_of_data) {
DVLOG(20) << __func__ << "() save_item_id = " << save_item->id()
<< " url = \"" << save_item->url().spec() << "\"";
- BrowserThread::PostTask(
- BrowserThread::FILE, FROM_HERE,
- base::Bind(&SaveFileManager::SaveFinished, file_manager_,
- save_item->id(), id(), true));
+ GetDownloadTaskRunner()->PostTask(
+ FROM_HERE, base::Bind(&SaveFileManager::SaveFinished, file_manager_,
+ save_item->id(), id(), true));
number_of_frames_pending_response_--;
DCHECK_LE(0, number_of_frames_pending_response_);
}
@@ -1289,7 +1284,7 @@ base::FilePath SavePackage::GetSuggestedNameForSaveAs(
// static
base::FilePath SavePackage::EnsureHtmlExtension(const base::FilePath& name) {
- DCHECK_CURRENTLY_ON(BrowserThread::FILE);
+ base::ThreadRestrictions::AssertIOAllowed();
base::FilePath::StringType ext = name.Extension();
if (!ext.empty())
@@ -1306,7 +1301,7 @@ base::FilePath SavePackage::EnsureHtmlExtension(const base::FilePath& name) {
// static
base::FilePath SavePackage::EnsureMimeExtension(const base::FilePath& name,
const std::string& contents_mime_type) {
- DCHECK_CURRENTLY_ON(BrowserThread::FILE);
+ base::ThreadRestrictions::AssertIOAllowed();
// Start extension at 1 to skip over period if non-empty.
base::FilePath::StringType ext = name.Extension();
@@ -1346,8 +1341,8 @@ const base::FilePath::CharType* SavePackage::ExtensionForMimeType(
void SavePackage::GetSaveInfo() {
DCHECK_CURRENTLY_ON(BrowserThread::UI);
- // Can't use |web_contents_| in the FILE thread, so get the data that we need
- // before calling to it.
+ // Can't use |web_contents_| in the download sequence, so get the data that we
+ // need before calling to it.
base::FilePath website_save_dir;
base::FilePath download_save_dir;
bool skip_dir_check = false;
@@ -1359,8 +1354,8 @@ void SavePackage::GetSaveInfo() {
}
std::string mime_type = web_contents()->GetContentsMimeType();
bool can_save_as_complete = CanSaveAsComplete(mime_type);
- BrowserThread::PostTaskAndReplyWithResult(
- BrowserThread::FILE, FROM_HERE,
+ base::PostTaskAndReplyWithResult(
+ GetDownloadTaskRunner().get(), FROM_HERE,
base::Bind(&SavePackage::CreateDirectoryOnFileThread, title_, page_url_,
can_save_as_complete, mime_type, website_save_dir,
download_save_dir, skip_dir_check),
@@ -1377,7 +1372,7 @@ base::FilePath SavePackage::CreateDirectoryOnFileThread(
const base::FilePath& website_save_dir,
const base::FilePath& download_save_dir,
bool skip_dir_check) {
- DCHECK_CURRENTLY_ON(BrowserThread::FILE);
+ DCHECK(GetDownloadTaskRunner()->RunsTasksInCurrentSequence());
base::FilePath suggested_filename = GetSuggestedNameForSaveAs(
title, page_url, can_save_as_complete, mime_type);
@@ -1420,7 +1415,7 @@ void SavePackage::ContinueGetSaveInfo(bool can_save_as_complete,
DCHECK_CURRENTLY_ON(BrowserThread::UI);
// The WebContents which owns this SavePackage may have disappeared during
- // the UI->FILE->UI thread hop of
+ // the UI->download sequence->UI thread hop of
// GetSaveInfo->CreateDirectoryOnFileThread->ContinueGetSaveInfo.
if (!web_contents() || !download_manager_->GetDelegate())
return;
« no previous file with comments | « content/browser/download/save_package.h ('k') | content/public/browser/BUILD.gn » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698