OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "net/base/elements_upload_data_stream.h" | 5 #include "net/base/elements_upload_data_stream.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/logging.h" | 8 #include "base/logging.h" |
9 #include "net/base/completion_callback.h" | 9 #include "net/base/completion_callback.h" |
10 #include "net/base/io_buffer.h" | 10 #include "net/base/io_buffer.h" |
11 #include "net/base/net_errors.h" | 11 #include "net/base/net_errors.h" |
12 #include "net/base/upload_bytes_element_reader.h" | 12 #include "net/base/upload_bytes_element_reader.h" |
13 #include "net/base/upload_element_reader.h" | 13 #include "net/base/upload_element_reader.h" |
14 | 14 |
15 namespace net { | 15 namespace net { |
16 | 16 |
17 ElementsUploadDataStream::ElementsUploadDataStream( | 17 ElementsUploadDataStream::ElementsUploadDataStream( |
18 ScopedVector<UploadElementReader> element_readers, | 18 std::vector<scoped_ptr<UploadElementReader>> element_readers, |
19 int64_t identifier) | 19 int64_t identifier) |
20 : UploadDataStream(false, identifier), | 20 : UploadDataStream(false, identifier), |
21 element_readers_(element_readers.Pass()), | 21 element_readers_(std::move(element_readers)), |
22 element_index_(0), | 22 element_index_(0), |
23 read_failed_(false), | 23 read_failed_(false), |
24 weak_ptr_factory_(this) { | 24 weak_ptr_factory_(this) {} |
25 } | |
26 | 25 |
27 ElementsUploadDataStream::~ElementsUploadDataStream() { | 26 ElementsUploadDataStream::~ElementsUploadDataStream() { |
28 } | 27 } |
29 | 28 |
30 scoped_ptr<UploadDataStream> ElementsUploadDataStream::CreateWithReader( | 29 scoped_ptr<UploadDataStream> ElementsUploadDataStream::CreateWithReader( |
31 scoped_ptr<UploadElementReader> reader, | 30 scoped_ptr<UploadElementReader> reader, |
32 int64_t identifier) { | 31 int64_t identifier) { |
33 ScopedVector<UploadElementReader> readers; | 32 std::vector<scoped_ptr<UploadElementReader>> readers; |
34 readers.push_back(reader.Pass()); | 33 readers.push_back(std::move(reader)); |
35 return scoped_ptr<UploadDataStream>( | 34 return scoped_ptr<UploadDataStream>( |
36 new ElementsUploadDataStream(readers.Pass(), identifier)); | 35 new ElementsUploadDataStream(std::move(readers), identifier)); |
37 } | 36 } |
38 | 37 |
39 int ElementsUploadDataStream::InitInternal() { | 38 int ElementsUploadDataStream::InitInternal() { |
40 return InitElements(0); | 39 return InitElements(0); |
41 } | 40 } |
42 | 41 |
43 int ElementsUploadDataStream::ReadInternal( | 42 int ElementsUploadDataStream::ReadInternal( |
44 IOBuffer* buf, | 43 IOBuffer* buf, |
45 int buf_len) { | 44 int buf_len) { |
46 DCHECK_GT(buf_len, 0); | 45 DCHECK_GT(buf_len, 0); |
47 return ReadElements(new DrainableIOBuffer(buf, buf_len)); | 46 return ReadElements(new DrainableIOBuffer(buf, buf_len)); |
48 } | 47 } |
49 | 48 |
50 bool ElementsUploadDataStream::IsInMemory() const { | 49 bool ElementsUploadDataStream::IsInMemory() const { |
51 for (size_t i = 0; i < element_readers_.size(); ++i) { | 50 for (const scoped_ptr<UploadElementReader>& it : element_readers_) { |
52 if (!element_readers_[i]->IsInMemory()) | 51 if (!it->IsInMemory()) |
53 return false; | 52 return false; |
54 } | 53 } |
55 return true; | 54 return true; |
56 } | 55 } |
57 | 56 |
58 const ScopedVector<UploadElementReader>* | 57 const std::vector<scoped_ptr<UploadElementReader>>* |
59 ElementsUploadDataStream::GetElementReaders() const { | 58 ElementsUploadDataStream::GetElementReaders() const { |
60 return &element_readers_; | 59 return &element_readers_; |
61 } | 60 } |
62 | 61 |
63 void ElementsUploadDataStream::ResetInternal() { | 62 void ElementsUploadDataStream::ResetInternal() { |
64 weak_ptr_factory_.InvalidateWeakPtrs(); | 63 weak_ptr_factory_.InvalidateWeakPtrs(); |
65 read_failed_ = false; | 64 read_failed_ = false; |
66 element_index_ = 0; | 65 element_index_ = 0; |
67 } | 66 } |
68 | 67 |
69 int ElementsUploadDataStream::InitElements(size_t start_index) { | 68 int ElementsUploadDataStream::InitElements(size_t start_index) { |
70 // Call Init() for all elements. | 69 // Call Init() for all elements. |
71 for (size_t i = start_index; i < element_readers_.size(); ++i) { | 70 for (size_t i = start_index; i < element_readers_.size(); ++i) { |
72 UploadElementReader* reader = element_readers_[i]; | 71 UploadElementReader* reader = element_readers_[i].get(); |
73 // When new_result is ERR_IO_PENDING, InitInternal() will be called | 72 // When new_result is ERR_IO_PENDING, InitInternal() will be called |
74 // with start_index == i + 1 when reader->Init() finishes. | 73 // with start_index == i + 1 when reader->Init() finishes. |
75 int result = reader->Init( | 74 int result = reader->Init( |
76 base::Bind(&ElementsUploadDataStream::OnInitElementCompleted, | 75 base::Bind(&ElementsUploadDataStream::OnInitElementCompleted, |
77 weak_ptr_factory_.GetWeakPtr(), | 76 weak_ptr_factory_.GetWeakPtr(), |
78 i)); | 77 i)); |
79 DCHECK(result != ERR_IO_PENDING || !reader->IsInMemory()); | 78 DCHECK(result != ERR_IO_PENDING || !reader->IsInMemory()); |
80 DCHECK_LE(result, OK); | 79 DCHECK_LE(result, OK); |
81 if (result != OK) | 80 if (result != OK) |
82 return result; | 81 return result; |
83 } | 82 } |
84 | 83 |
85 uint64_t total_size = 0; | 84 uint64_t total_size = 0; |
86 for (size_t i = 0; i < element_readers_.size(); ++i) { | 85 for (const scoped_ptr<UploadElementReader>& it : element_readers_) { |
87 total_size += element_readers_[i]->GetContentLength(); | 86 total_size += it->GetContentLength(); |
88 } | 87 } |
89 SetSize(total_size); | 88 SetSize(total_size); |
90 return OK; | 89 return OK; |
91 } | 90 } |
92 | 91 |
93 void ElementsUploadDataStream::OnInitElementCompleted(size_t index, | 92 void ElementsUploadDataStream::OnInitElementCompleted(size_t index, |
94 int result) { | 93 int result) { |
95 DCHECK_NE(ERR_IO_PENDING, result); | 94 DCHECK_NE(ERR_IO_PENDING, result); |
96 | 95 |
97 // Check the last result. | 96 // Check the last result. |
98 if (result == OK) | 97 if (result == OK) |
99 result = InitElements(index + 1); | 98 result = InitElements(index + 1); |
100 | 99 |
101 if (result != ERR_IO_PENDING) | 100 if (result != ERR_IO_PENDING) |
102 OnInitCompleted(result); | 101 OnInitCompleted(result); |
103 } | 102 } |
104 | 103 |
105 int ElementsUploadDataStream::ReadElements( | 104 int ElementsUploadDataStream::ReadElements( |
106 const scoped_refptr<DrainableIOBuffer>& buf) { | 105 const scoped_refptr<DrainableIOBuffer>& buf) { |
107 while (!read_failed_ && element_index_ < element_readers_.size()) { | 106 while (!read_failed_ && element_index_ < element_readers_.size()) { |
108 UploadElementReader* reader = element_readers_[element_index_]; | 107 UploadElementReader* reader = element_readers_[element_index_].get(); |
109 | 108 |
110 if (reader->BytesRemaining() == 0) { | 109 if (reader->BytesRemaining() == 0) { |
111 ++element_index_; | 110 ++element_index_; |
112 continue; | 111 continue; |
113 } | 112 } |
114 | 113 |
115 if (buf->BytesRemaining() == 0) | 114 if (buf->BytesRemaining() == 0) |
116 break; | 115 break; |
117 | 116 |
118 int result = reader->Read( | 117 int result = reader->Read( |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
157 DCHECK(!read_failed_); | 156 DCHECK(!read_failed_); |
158 | 157 |
159 if (result >= 0) { | 158 if (result >= 0) { |
160 buf->DidConsume(result); | 159 buf->DidConsume(result); |
161 } else { | 160 } else { |
162 read_failed_ = true; | 161 read_failed_ = true; |
163 } | 162 } |
164 } | 163 } |
165 | 164 |
166 } // namespace net | 165 } // namespace net |
OLD | NEW |