| OLD | NEW |
| 1 // Copyright (c) 2015 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2015 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 #ifndef STORAGE_BROWSER_BLOB_BLOB_DATA_BUILDER_H_ | 5 #ifndef STORAGE_BROWSER_BLOB_BLOB_DATA_BUILDER_H_ |
| 6 #define STORAGE_BROWSER_BLOB_BLOB_DATA_BUILDER_H_ | 6 #define STORAGE_BROWSER_BLOB_BLOB_DATA_BUILDER_H_ |
| 7 | 7 |
| 8 #include <stdint.h> | 8 #include <stdint.h> |
| 9 #include <ostream> |
| 9 #include <string> | 10 #include <string> |
| 10 #include <vector> | 11 #include <vector> |
| 11 | 12 |
| 12 #include "base/basictypes.h" | 13 #include "base/basictypes.h" |
| 13 #include "base/files/file_path.h" | 14 #include "base/files/file_path.h" |
| 14 #include "base/memory/ref_counted.h" | 15 #include "base/memory/ref_counted.h" |
| 15 #include "storage/browser/blob/blob_data_item.h" | 16 #include "storage/browser/blob/blob_data_item.h" |
| 16 #include "storage/browser/blob/blob_data_snapshot.h" | 17 #include "storage/browser/blob/blob_data_snapshot.h" |
| 17 #include "storage/browser/storage_browser_export.h" | 18 #include "storage/browser/storage_browser_export.h" |
| 18 | 19 |
| 19 namespace disk_cache { | 20 namespace disk_cache { |
| 20 class Entry; | 21 class Entry; |
| 21 } | 22 } |
| 22 | 23 |
| 23 namespace storage { | 24 namespace storage { |
| 24 class BlobStorageContext; | 25 class BlobStorageContext; |
| 25 | 26 |
| 26 class STORAGE_EXPORT BlobDataBuilder { | 27 class STORAGE_EXPORT BlobDataBuilder { |
| 27 public: | 28 public: |
| 28 using DataHandle = BlobDataItem::DataHandle; | 29 using DataHandle = BlobDataItem::DataHandle; |
| 29 | 30 |
| 30 explicit BlobDataBuilder(const std::string& uuid); | 31 explicit BlobDataBuilder(const std::string& uuid); |
| 31 ~BlobDataBuilder(); | 32 ~BlobDataBuilder(); |
| 32 | 33 |
| 33 const std::string& uuid() const { return uuid_; } | 34 const std::string& uuid() const { return uuid_; } |
| 34 | 35 |
| 36 // Validates the data element, and copies the data if it's a 'bytes' element. |
| 37 // Cannot append elements of BYTES_DESCRIPTION or DISK_CACHE_ENTRY types. |
| 38 void AppendDataElement(const DataElement& element); |
| 39 |
| 35 void AppendData(const std::string& data) { | 40 void AppendData(const std::string& data) { |
| 36 AppendData(data.c_str(), data.size()); | 41 AppendData(data.c_str(), data.size()); |
| 37 } | 42 } |
| 38 | 43 |
| 39 void AppendData(const char* data, size_t length); | 44 void AppendData(const char* data, size_t length); |
| 40 | 45 |
| 46 // ***** Async construction methods ***** |
| 47 void AppendFutureData(size_t length); |
| 48 bool PopulateFutureData(size_t index, |
| 49 const char* data, |
| 50 size_t offset, |
| 51 size_t length); |
| 52 // ************************************** |
| 53 |
| 41 // You must know the length of the file, you cannot use kuint64max to specify | 54 // You must know the length of the file, you cannot use kuint64max to specify |
| 42 // the whole file. This method creates a ShareableFileReference to the given | 55 // the whole file. This method creates a ShareableFileReference to the given |
| 43 // file, which is stored in this builder. | 56 // file, which is stored in this builder. |
| 44 void AppendFile(const base::FilePath& file_path, | 57 void AppendFile(const base::FilePath& file_path, |
| 45 uint64_t offset, | 58 uint64_t offset, |
| 46 uint64_t length, | 59 uint64_t length, |
| 47 const base::Time& expected_modification_time); | 60 const base::Time& expected_modification_time); |
| 48 | 61 |
| 49 void AppendBlob(const std::string& uuid, uint64_t offset, uint64_t length); | 62 void AppendBlob(const std::string& uuid, uint64_t offset, uint64_t length); |
| 50 | 63 |
| 51 void AppendBlob(const std::string& uuid); | 64 void AppendBlob(const std::string& uuid); |
| 52 | 65 |
| 53 void AppendFileSystemFile(const GURL& url, | 66 void AppendFileSystemFile(const GURL& url, |
| 54 uint64_t offset, | 67 uint64_t offset, |
| 55 uint64_t length, | 68 uint64_t length, |
| 56 const base::Time& expected_modification_time); | 69 const base::Time& expected_modification_time); |
| 57 | 70 |
| 58 void AppendDiskCacheEntry(const scoped_refptr<DataHandle>& data_handle, | 71 void AppendDiskCacheEntry(const scoped_refptr<DataHandle>& data_handle, |
| 59 disk_cache::Entry* disk_cache_entry, | 72 disk_cache::Entry* disk_cache_entry, |
| 60 int disk_cache_stream_index); | 73 int disk_cache_stream_index); |
| 61 | 74 |
| 62 void set_content_type(const std::string& content_type) { | 75 void set_content_type(const std::string& content_type) { |
| 63 content_type_ = content_type; | 76 content_type_ = content_type; |
| 64 } | 77 } |
| 65 | 78 |
| 66 void set_content_disposition(const std::string& content_disposition) { | 79 void set_content_disposition(const std::string& content_disposition) { |
| 67 content_disposition_ = content_disposition; | 80 content_disposition_ = content_disposition; |
| 68 } | 81 } |
| 69 | 82 |
| 83 void Clear(); |
| 84 |
| 70 private: | 85 private: |
| 71 friend class BlobStorageContext; | 86 friend class BlobStorageContext; |
| 87 friend class BlobAsyncBuilderHostTest; |
| 72 friend bool operator==(const BlobDataBuilder& a, const BlobDataBuilder& b); | 88 friend bool operator==(const BlobDataBuilder& a, const BlobDataBuilder& b); |
| 73 friend bool operator==(const BlobDataSnapshot& a, const BlobDataBuilder& b); | 89 friend bool operator==(const BlobDataSnapshot& a, const BlobDataBuilder& b); |
| 90 friend STORAGE_EXPORT std::ostream& operator<<(std::ostream& os, |
| 91 const BlobDataBuilder& x); |
| 74 | 92 |
| 75 std::string uuid_; | 93 std::string uuid_; |
| 76 std::string content_type_; | 94 std::string content_type_; |
| 77 std::string content_disposition_; | 95 std::string content_disposition_; |
| 78 std::vector<scoped_refptr<BlobDataItem>> items_; | 96 std::vector<scoped_refptr<BlobDataItem>> items_; |
| 79 | 97 |
| 80 DISALLOW_COPY_AND_ASSIGN(BlobDataBuilder); | 98 DISALLOW_COPY_AND_ASSIGN(BlobDataBuilder); |
| 81 }; | 99 }; |
| 82 | 100 |
| 101 std::ostream& operator<<(std::ostream& os, const BlobDataBuilder& x); |
| 102 |
| 83 #if defined(UNIT_TEST) | 103 #if defined(UNIT_TEST) |
| 84 inline bool operator==(const BlobDataBuilder& a, const BlobDataBuilder& b) { | 104 inline bool operator==(const BlobDataBuilder& a, const BlobDataBuilder& b) { |
| 85 if (a.content_type_ != b.content_type_) | 105 if (a.content_type_ != b.content_type_) |
| 86 return false; | 106 return false; |
| 87 if (a.content_disposition_ != b.content_disposition_) | 107 if (a.content_disposition_ != b.content_disposition_) |
| 88 return false; | 108 return false; |
| 89 if (a.items_.size() != b.items_.size()) | 109 if (a.items_.size() != b.items_.size()) |
| 90 return false; | 110 return false; |
| 91 for (size_t i = 0; i < a.items_.size(); ++i) { | 111 for (size_t i = 0; i < a.items_.size(); ++i) { |
| 92 if (a.items_[i] != b.items_[i]) | 112 if (*(a.items_[i]) != *(b.items_[i])) |
| 93 return false; | 113 return false; |
| 94 } | 114 } |
| 95 return true; | 115 return true; |
| 96 } | 116 } |
| 97 | 117 |
| 98 inline bool operator==(const BlobDataSnapshot& a, const BlobDataBuilder& b) { | 118 inline bool operator==(const BlobDataSnapshot& a, const BlobDataBuilder& b) { |
| 99 if (a.content_type() != b.content_type_) { | 119 if (a.content_type() != b.content_type_) { |
| 100 return false; | 120 return false; |
| 101 } | 121 } |
| 102 if (a.content_disposition() != b.content_disposition_) { | 122 if (a.content_disposition() != b.content_disposition_) { |
| 103 return false; | 123 return false; |
| 104 } | 124 } |
| 105 if (a.items().size() != b.items_.size()) { | 125 if (a.items().size() != b.items_.size()) { |
| 106 return false; | 126 return false; |
| 107 } | 127 } |
| 108 for (size_t i = 0; i < a.items().size(); ++i) { | 128 for (size_t i = 0; i < a.items().size(); ++i) { |
| 109 if (*(a.items()[i]) != *(b.items_[i])) | 129 if (*(a.items()[i]) != *(b.items_[i])) |
| 110 return false; | 130 return false; |
| 111 } | 131 } |
| 112 return true; | 132 return true; |
| 113 } | 133 } |
| 114 | 134 |
| 115 inline bool operator!=(const BlobDataSnapshot& a, const BlobDataBuilder& b) { | 135 inline bool operator!=(const BlobDataSnapshot& a, const BlobDataBuilder& b) { |
| 116 return !(a == b); | 136 return !(a == b); |
| 117 } | 137 } |
| 118 | 138 |
| 119 inline bool operator!=(const BlobDataBuilder& a, const BlobDataBuilder& b) { | 139 inline bool operator!=(const BlobDataBuilder& a, const BlobDataBuilder& b) { |
| 120 return !(a == b); | 140 return !(a == b); |
| 121 } | 141 } |
| 142 |
| 122 #endif // defined(UNIT_TEST) | 143 #endif // defined(UNIT_TEST) |
| 123 | 144 |
| 124 } // namespace storage | 145 } // namespace storage |
| 125 #endif // STORAGE_BROWSER_BLOB_BLOB_DATA_BUILDER_H_ | 146 #endif // STORAGE_BROWSER_BLOB_BLOB_DATA_BUILDER_H_ |
| OLD | NEW |