Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/loader/upload_data_stream_builder.h" | 5 #include "content/browser/loader/upload_data_stream_builder.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "content/browser/fileapi/upload_file_system_file_element_reader.h" | 8 #include "content/browser/fileapi/upload_file_system_file_element_reader.h" |
| 9 #include "content/common/resource_request_body.h" | 9 #include "content/common/resource_request_body.h" |
| 10 #include "net/base/elements_upload_data_stream.h" | 10 #include "net/base/elements_upload_data_stream.h" |
| 11 #include "net/base/upload_bytes_element_reader.h" | 11 #include "net/base/upload_bytes_element_reader.h" |
| 12 #include "net/base/upload_file_element_reader.h" | 12 #include "net/base/upload_file_element_reader.h" |
| 13 #include "storage/browser/blob/blob_data_handle.h" | 13 #include "storage/browser/blob/blob_data_handle.h" |
| 14 #include "storage/browser/blob/blob_storage_context.h" | 14 #include "storage/browser/blob/blob_storage_context.h" |
| 15 | 15 #include "storage/common/blob/blob_data.h" |
| 16 using storage::BlobData; | |
| 17 using storage::BlobDataHandle; | |
| 18 using storage::BlobStorageContext; | |
| 19 | 16 |
| 20 namespace content { | 17 namespace content { |
| 21 namespace { | 18 namespace { |
| 22 | 19 |
| 23 // A subclass of net::UploadBytesElementReader which owns ResourceRequestBody. | 20 // A subclass of net::UploadBytesElementReader which owns ResourceRequestBody. |
| 24 class BytesElementReader : public net::UploadBytesElementReader { | 21 class BytesElementReader : public net::UploadBytesElementReader { |
| 25 public: | 22 public: |
| 26 BytesElementReader(ResourceRequestBody* resource_request_body, | 23 BytesElementReader(ResourceRequestBody* resource_request_body, |
| 27 const ResourceRequestBody::Element& element) | 24 const ResourceRequestBody::Element& element) |
| 28 : net::UploadBytesElementReader(element.bytes(), element.length()), | 25 : net::UploadBytesElementReader(element.bytes(), element.length()), |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 67 storage::BlobStorageContext* blob_context, | 64 storage::BlobStorageContext* blob_context, |
| 68 const ResourceRequestBody::Element& element, | 65 const ResourceRequestBody::Element& element, |
| 69 std::vector<const ResourceRequestBody::Element*>* resolved_elements) { | 66 std::vector<const ResourceRequestBody::Element*>* resolved_elements) { |
| 70 DCHECK(blob_context); | 67 DCHECK(blob_context); |
| 71 scoped_ptr<storage::BlobDataHandle> handle = | 68 scoped_ptr<storage::BlobDataHandle> handle = |
| 72 blob_context->GetBlobDataFromUUID(element.blob_uuid()); | 69 blob_context->GetBlobDataFromUUID(element.blob_uuid()); |
| 73 DCHECK(handle); | 70 DCHECK(handle); |
| 74 if (!handle) | 71 if (!handle) |
| 75 return; | 72 return; |
| 76 | 73 |
| 74 // TODO(dmurph): store this snapshot. This works for now, as the items | |
|
mmenke
2015/01/20 22:03:57
nit: Capitalize S
dmurph
2015/01/21 22:40:11
Done.
| |
| 75 // that the snapshot points to are refcounted and kept in memory. But as | |
|
mmenke
2015/01/20 22:03:57
Should mention that they're attached to the reques
dmurph
2015/01/21 22:40:11
Done.
| |
| 76 // soon as we implement evicting to disk this will break, so we need a reader | |
| 77 // that can store this snapshot and have it used below instead of this | |
| 78 // decomposition. | |
|
michaeln
2015/01/21 01:46:41
It would be really good to store snapshots in this
dmurph
2015/01/21 22:40:11
Done.
| |
| 79 scoped_ptr<storage::BlobDataSnapshot> snapshot = handle->CreateSnapshot(); | |
| 77 // If there is no element in the referred blob data, just return. | 80 // If there is no element in the referred blob data, just return. |
| 78 if (handle->data()->items().empty()) | 81 if (snapshot->items().empty()) |
| 79 return; | 82 return; |
| 80 | 83 |
| 81 // Append the elements in the referenced blob data. | 84 // Append the elements in the referenced blob data. |
| 82 for (size_t i = 0; i < handle->data()->items().size(); ++i) { | 85 for (const auto& item : snapshot->items()) { |
| 83 const BlobData::Item& item = handle->data()->items().at(i); | 86 DCHECK_NE(storage::DataElement::TYPE_BLOB, item->type()); |
| 84 DCHECK_NE(BlobData::Item::TYPE_BLOB, item.type()); | 87 resolved_elements->push_back(item->data_element_ptr()); |
| 85 resolved_elements->push_back(&item); | |
| 86 } | 88 } |
| 87 } | 89 } |
| 88 | 90 |
| 89 } // namespace | 91 } // namespace |
| 90 | 92 |
| 91 scoped_ptr<net::UploadDataStream> UploadDataStreamBuilder::Build( | 93 scoped_ptr<net::UploadDataStream> UploadDataStreamBuilder::Build( |
| 92 ResourceRequestBody* body, | 94 ResourceRequestBody* body, |
| 93 BlobStorageContext* blob_context, | 95 storage::BlobStorageContext* blob_context, |
| 94 storage::FileSystemContext* file_system_context, | 96 storage::FileSystemContext* file_system_context, |
| 95 base::TaskRunner* file_task_runner) { | 97 base::TaskRunner* file_task_runner) { |
| 96 // Resolve all blob elements. | 98 // Resolve all blob elements. |
| 97 std::vector<const ResourceRequestBody::Element*> resolved_elements; | 99 std::vector<const ResourceRequestBody::Element*> resolved_elements; |
| 98 for (size_t i = 0; i < body->elements()->size(); ++i) { | 100 for (size_t i = 0; i < body->elements()->size(); ++i) { |
| 99 const ResourceRequestBody::Element& element = (*body->elements())[i]; | 101 const ResourceRequestBody::Element& element = (*body->elements())[i]; |
| 100 if (element.type() == ResourceRequestBody::Element::TYPE_BLOB) | 102 if (element.type() == ResourceRequestBody::Element::TYPE_BLOB) |
| 101 ResolveBlobReference(blob_context, element, &resolved_elements); | 103 ResolveBlobReference(blob_context, element, &resolved_elements); |
| 102 else | 104 else |
| 103 resolved_elements.push_back(&element); | 105 resolved_elements.push_back(&element); |
| (...skipping 17 matching lines...) Expand all Loading... | |
| 121 element_readers.push_back( | 123 element_readers.push_back( |
| 122 new content::UploadFileSystemFileElementReader( | 124 new content::UploadFileSystemFileElementReader( |
| 123 file_system_context, | 125 file_system_context, |
| 124 element.filesystem_url(), | 126 element.filesystem_url(), |
| 125 element.offset(), | 127 element.offset(), |
| 126 element.length(), | 128 element.length(), |
| 127 element.expected_modification_time())); | 129 element.expected_modification_time())); |
| 128 break; | 130 break; |
| 129 case ResourceRequestBody::Element::TYPE_BLOB: | 131 case ResourceRequestBody::Element::TYPE_BLOB: |
| 130 // Blob elements should be resolved beforehand. | 132 // Blob elements should be resolved beforehand. |
| 133 // TODO(dmurph): Create blob reader and store the snapshot in there. | |
| 131 NOTREACHED(); | 134 NOTREACHED(); |
| 132 break; | 135 break; |
| 133 case ResourceRequestBody::Element::TYPE_UNKNOWN: | 136 case ResourceRequestBody::Element::TYPE_UNKNOWN: |
| 134 NOTREACHED(); | 137 NOTREACHED(); |
| 135 break; | 138 break; |
| 136 } | 139 } |
| 137 } | 140 } |
| 138 | 141 |
| 139 return make_scoped_ptr( | 142 return make_scoped_ptr( |
| 140 new net::ElementsUploadDataStream(element_readers.Pass(), | 143 new net::ElementsUploadDataStream(element_readers.Pass(), |
| 141 body->identifier())); | 144 body->identifier())); |
| 142 } | 145 } |
| 143 | 146 |
| 144 } // namespace content | 147 } // namespace content |
| OLD | NEW |