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

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

Issue 2526983002: Refactor ResourceHandler API. (Closed)
Patch Set: Response to comments Created 3 years, 11 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
Index: content/browser/download/download_resource_handler.cc
diff --git a/content/browser/download/download_resource_handler.cc b/content/browser/download/download_resource_handler.cc
index 30bcd5e8027da76ddbec1acfeac84aa82cd88c41..30cf6fab78a4aae523dbf622206843204702e5e7 100644
--- a/content/browser/download/download_resource_handler.cc
+++ b/content/browser/download/download_resource_handler.cc
@@ -139,24 +139,34 @@ std::unique_ptr<ResourceHandler> DownloadResourceHandler::Create(
return handler;
}
-bool DownloadResourceHandler::OnRequestRedirected(
+void DownloadResourceHandler::OnRequestRedirected(
const net::RedirectInfo& redirect_info,
ResourceResponse* response,
- bool* defer) {
- return core_.OnRequestRedirected();
+ std::unique_ptr<ResourceController> controller) {
+ if (core_.OnRequestRedirected()) {
+ controller->Resume();
+ } else {
+ controller->Cancel();
+ }
}
// Send the download creation information to the download thread.
-bool DownloadResourceHandler::OnResponseStarted(
+void DownloadResourceHandler::OnResponseStarted(
ResourceResponse* response,
- bool* defer) {
+ std::unique_ptr<ResourceController> controller) {
// The MIME type in ResourceResponse is the product of
// MimeTypeResourceHandler.
- return core_.OnResponseStarted(response->head.mime_type);
+ if (core_.OnResponseStarted(response->head.mime_type)) {
+ controller->Resume();
+ } else {
+ controller->Cancel();
+ }
}
-bool DownloadResourceHandler::OnWillStart(const GURL& url, bool* defer) {
- return true;
+void DownloadResourceHandler::OnWillStart(
+ const GURL& url,
+ std::unique_ptr<ResourceController> controller) {
+ controller->Resume();
}
// Create a new buffer, which will be handed to the download thread for file
@@ -168,14 +178,29 @@ bool DownloadResourceHandler::OnWillRead(scoped_refptr<net::IOBuffer>* buf,
}
// Pass the buffer to the download file writer.
-bool DownloadResourceHandler::OnReadCompleted(int bytes_read, bool* defer) {
- return core_.OnReadCompleted(bytes_read, defer);
+void DownloadResourceHandler::OnReadCompleted(
+ int bytes_read,
+ std::unique_ptr<ResourceController> controller) {
+ DCHECK(!has_controller());
+
+ bool defer = false;
+ if (!core_.OnReadCompleted(bytes_read, &defer)) {
+ controller->Cancel();
+ return;
+ }
+
+ if (defer) {
+ HoldController(std::move(controller));
+ } else {
+ controller->Resume();
+ }
}
void DownloadResourceHandler::OnResponseCompleted(
const net::URLRequestStatus& status,
- bool* defer) {
+ std::unique_ptr<ResourceController> controller) {
core_.OnResponseCompleted(status);
+ controller->Resume();
}
void DownloadResourceHandler::OnDataDownloaded(int bytes_downloaded) {
@@ -226,7 +251,7 @@ void DownloadResourceHandler::OnStart(
void DownloadResourceHandler::OnReadyToRead() {
DCHECK_CURRENTLY_ON(BrowserThread::IO);
- controller()->Resume();
+ Resume();
}
void DownloadResourceHandler::CancelRequest() {
« no previous file with comments | « content/browser/download/download_resource_handler.h ('k') | content/browser/download/save_file_resource_handler.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698