| OLD | NEW |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/download/download_worker.h" | 5 #include "content/browser/download/download_worker.h" |
| 6 | 6 |
| 7 #include "content/browser/download/download_create_info.h" | 7 #include "content/browser/download/download_create_info.h" |
| 8 #include "content/public/browser/download_interrupt_reasons.h" | 8 #include "content/public/browser/download_interrupt_reasons.h" |
| 9 #include "content/public/browser/web_contents.h" | 9 #include "content/public/browser/web_contents.h" |
| 10 | 10 |
| 11 namespace content { | 11 namespace content { |
| 12 namespace { | 12 namespace { |
| 13 | 13 |
| 14 const int kVerboseLevel = 1; | 14 const int kWorkerVerbosityLevel = 1; |
| 15 | 15 |
| 16 class CompletedByteStreamReader : public ByteStreamReader { | 16 class CompletedByteStreamReader : public ByteStreamReader { |
| 17 public: | 17 public: |
| 18 CompletedByteStreamReader(int status) : status_(status) {}; | 18 CompletedByteStreamReader(int status) : status_(status) {}; |
| 19 ~CompletedByteStreamReader() override = default; | 19 ~CompletedByteStreamReader() override = default; |
| 20 | 20 |
| 21 // ByteStreamReader implementations: | 21 // ByteStreamReader implementations: |
| 22 ByteStreamReader::StreamState Read(scoped_refptr<net::IOBuffer>* data, | 22 ByteStreamReader::StreamState Read(scoped_refptr<net::IOBuffer>* data, |
| 23 size_t* length) override { | 23 size_t* length) override { |
| 24 return ByteStreamReader::STREAM_COMPLETE; | 24 return ByteStreamReader::STREAM_COMPLETE; |
| (...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 97 | 97 |
| 98 void DownloadWorker::OnUrlDownloadStarted( | 98 void DownloadWorker::OnUrlDownloadStarted( |
| 99 std::unique_ptr<DownloadCreateInfo> create_info, | 99 std::unique_ptr<DownloadCreateInfo> create_info, |
| 100 std::unique_ptr<DownloadManager::InputStream> input_stream, | 100 std::unique_ptr<DownloadManager::InputStream> input_stream, |
| 101 const DownloadUrlParameters::OnStartedCallback& callback) { | 101 const DownloadUrlParameters::OnStartedCallback& callback) { |
| 102 // |callback| is not used in subsequent requests. | 102 // |callback| is not used in subsequent requests. |
| 103 DCHECK(callback.is_null()); | 103 DCHECK(callback.is_null()); |
| 104 | 104 |
| 105 // Destroy the request if user canceled. | 105 // Destroy the request if user canceled. |
| 106 if (is_canceled_) { | 106 if (is_canceled_) { |
| 107 VLOG(kVerboseLevel) << "Byte stream arrived after user cancel the request."; | 107 VLOG(kWorkerVerbosityLevel) << "Byte stream arrived after user cancel the re
quest."; |
| 108 create_info->request_handle->CancelRequest(is_user_cancel_); | 108 create_info->request_handle->CancelRequest(is_user_cancel_); |
| 109 return; | 109 return; |
| 110 } | 110 } |
| 111 | 111 |
| 112 // TODO(xingliu): Add metric for error handling. | 112 // TODO(xingliu): Add metric for error handling. |
| 113 if (create_info->result != | 113 if (create_info->result != |
| 114 DownloadInterruptReason::DOWNLOAD_INTERRUPT_REASON_NONE) { | 114 DownloadInterruptReason::DOWNLOAD_INTERRUPT_REASON_NONE) { |
| 115 VLOG(kVerboseLevel) << "Parallel download sub-request failed. reason = " | 115 VLOG(kWorkerVerbosityLevel) << "Parallel download sub-request failed. reason
= " |
| 116 << create_info->result; | 116 << create_info->result; |
| 117 input_stream->stream_reader_.reset( | 117 input_stream->stream_reader_.reset( |
| 118 new CompletedByteStreamReader(create_info->result)); | 118 new CompletedByteStreamReader(create_info->result)); |
| 119 } | 119 } |
| 120 | 120 |
| 121 request_handle_ = std::move(create_info->request_handle); | 121 request_handle_ = std::move(create_info->request_handle); |
| 122 | 122 |
| 123 // Pause the stream if user paused, still push the stream reader to the sink. | 123 // Pause the stream if user paused, still push the stream reader to the sink. |
| 124 if (is_paused_) { | 124 if (is_paused_) { |
| 125 VLOG(kVerboseLevel) << "Byte stream arrived after user pause the request."; | 125 VLOG(kWorkerVerbosityLevel) << "Byte stream arrived after user pause the req
uest."; |
| 126 Pause(); | 126 Pause(); |
| 127 } | 127 } |
| 128 | 128 |
| 129 delegate_->OnByteStreamReady(this, std::move(input_stream->stream_reader_)); | 129 delegate_->OnByteStreamReady(this, std::move(input_stream->stream_reader_)); |
| 130 } | 130 } |
| 131 | 131 |
| 132 void DownloadWorker::OnUrlDownloadStopped(UrlDownloadHandler* downloader) { | 132 void DownloadWorker::OnUrlDownloadStopped(UrlDownloadHandler* downloader) { |
| 133 // Release the |url_download_handler_|, the object will be deleted on IO | 133 // Release the |url_download_handler_|, the object will be deleted on IO |
| 134 // thread. | 134 // thread. |
| 135 url_download_handler_.reset(); | 135 url_download_handler_.reset(); |
| 136 } | 136 } |
| 137 | 137 |
| 138 void DownloadWorker::AddUrlDownloadHandler( | 138 void DownloadWorker::AddUrlDownloadHandler( |
| 139 std::unique_ptr<UrlDownloadHandler, BrowserThread::DeleteOnIOThread> | 139 std::unique_ptr<UrlDownloadHandler, BrowserThread::DeleteOnIOThread> |
| 140 downloader) { | 140 downloader) { |
| 141 url_download_handler_ = std::move(downloader); | 141 url_download_handler_ = std::move(downloader); |
| 142 } | 142 } |
| 143 | 143 |
| 144 } // namespace content | 144 } // namespace content |
| OLD | NEW |