Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "sync/internal_api/public/attachments/attachment_uploader_impl.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/message_loop/message_loop.h" | |
| 9 #include "base/threading/non_thread_safe.h" | |
| 10 #include "net/base/load_flags.h" | |
| 11 #include "net/http/http_status_code.h" | |
| 12 #include "net/url_request/url_fetcher.h" | |
| 13 #include "net/url_request/url_fetcher_delegate.h" | |
| 14 #include "sync/api/attachments/attachment.h" | |
| 15 #include "sync/protocol/sync.pb.h" | |
| 16 #include "url/gurl.h" | |
| 17 | |
| 18 namespace { | |
| 19 | |
| 20 const char kContentType[] = "application/octet-stream"; | |
| 21 | |
| 22 } // namespace | |
| 23 | |
| 24 namespace syncer { | |
| 25 | |
| 26 // Encapsulates all the state associated with a single upload. | |
| 27 class AttachmentUploaderImpl::UploadState : public net::URLFetcherDelegate, | |
| 28 public base::NonThreadSafe { | |
| 29 public: | |
| 30 // Construct an UploadState. | |
| 31 // | |
| 32 // |owner| is a pointer to the object that will own (and must outlive!) this | |
| 33 // |UploadState. | |
| 34 UploadState(const GURL& upload_url, | |
| 35 const scoped_refptr<net::URLRequestContextGetter>& | |
| 36 url_request_context_getter, | |
| 37 const Attachment& attachment, | |
| 38 const UploadCallback& user_callback, | |
| 39 AttachmentUploaderImpl* owner); | |
| 40 | |
| 41 virtual ~UploadState(); | |
| 42 | |
| 43 // Add |user_callback| to the list of callbacks to be invoked when this upload | |
| 44 // completed. | |
| 45 void AddUserCallback(const UploadCallback& user_callback); | |
| 46 | |
| 47 // Return the Attachment this object is uploading. | |
| 48 const Attachment& GetAttachment(); | |
| 49 | |
| 50 // URLFetcher implementation. | |
| 51 virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE; | |
| 52 | |
| 53 private: | |
| 54 typedef std::vector<UploadCallback> UploadCallbackList; | |
| 55 | |
| 56 GURL upload_url_; | |
| 57 const scoped_refptr<net::URLRequestContextGetter>& | |
| 58 url_request_context_getter_; | |
| 59 Attachment attachment_; | |
| 60 UploadCallbackList user_callbacks_; | |
| 61 scoped_ptr<net::URLFetcher> fetcher_; | |
| 62 // Pointer to the AttachmentUploaderImpl that owns this object. | |
| 63 AttachmentUploaderImpl* owner_; | |
| 64 | |
| 65 DISALLOW_COPY_AND_ASSIGN(UploadState); | |
| 66 }; | |
| 67 | |
| 68 AttachmentUploaderImpl::UploadState::UploadState( | |
| 69 const GURL& upload_url, | |
| 70 const scoped_refptr<net::URLRequestContextGetter>& | |
| 71 url_request_context_getter, | |
| 72 const Attachment& attachment, | |
| 73 const UploadCallback& user_callback, | |
| 74 AttachmentUploaderImpl* owner) | |
| 75 : upload_url_(upload_url), | |
| 76 url_request_context_getter_(url_request_context_getter), | |
| 77 attachment_(attachment), | |
| 78 user_callbacks_(1, user_callback), | |
| 79 owner_(owner) { | |
| 80 DCHECK(url_request_context_getter_); | |
| 81 DCHECK(upload_url_.is_valid()); | |
| 82 DCHECK(owner_); | |
| 83 fetcher_.reset( | |
| 84 net::URLFetcher::Create(upload_url_, net::URLFetcher::POST, this)); | |
| 85 fetcher_->SetRequestContext(url_request_context_getter_.get()); | |
| 86 // TODO(maniscalco): Is there a better way? Copying the attachment data into | |
| 87 // a string feels wrong given how large attachments may be (several MBs). If | |
| 88 // we may end up switching from URLFetcher to URLRequest, this copy won't be | |
| 89 // necessary. | |
| 90 scoped_refptr<base::RefCountedMemory> memory = attachment.GetData(); | |
| 91 const std::string upload_content(memory->front_as<char>(), memory->size()); | |
| 92 fetcher_->SetUploadData(kContentType, upload_content); | |
| 93 // TODO(maniscalco): Add authentication support (bug 371516). | |
| 94 fetcher_->SetLoadFlags(net::LOAD_DO_NOT_SAVE_COOKIES | | |
| 95 net::LOAD_DO_NOT_SEND_COOKIES | | |
| 96 net::LOAD_DISABLE_CACHE); | |
| 97 // TODO(maniscalco): Set an appropriate headers (User-Agent, Content-type, and | |
| 98 // Content-length) on the request and include the content's MD5, | |
| 99 // AttachmentId's unique_id and the "sync birthday" (bug 371521). | |
| 100 fetcher_->Start(); | |
| 101 } | |
| 102 | |
| 103 AttachmentUploaderImpl::UploadState::~UploadState() { | |
| 104 } | |
| 105 | |
| 106 void AttachmentUploaderImpl::UploadState::AddUserCallback( | |
| 107 const UploadCallback& user_callback) { | |
| 108 DCHECK(CalledOnValidThread()); | |
| 109 user_callbacks_.push_back(user_callback); | |
| 110 } | |
| 111 | |
| 112 const Attachment& AttachmentUploaderImpl::UploadState::GetAttachment() { | |
| 113 DCHECK(CalledOnValidThread()); | |
| 114 return attachment_; | |
| 115 } | |
| 116 | |
| 117 void AttachmentUploaderImpl::UploadState::OnURLFetchComplete( | |
| 118 const net::URLFetcher* source) { | |
| 119 DCHECK(CalledOnValidThread()); | |
| 120 // TODO(maniscalco): Once the protocol is better defined, deal with the | |
| 121 // various HTTP response code we may encounter. | |
| 122 UploadResult result = UPLOAD_UNSPECIFIED_ERROR; | |
| 123 if (source->GetResponseCode() == net::HTTP_OK) { | |
| 124 result = UPLOAD_SUCCESS; | |
| 125 } | |
| 126 // TODO(maniscalco): Update the attachment id with server address information | |
| 127 // before passing it to the callback (bug 371522). | |
| 128 AttachmentId updated_id = attachment_.GetId(); | |
| 129 UploadCallbackList::const_iterator iter = user_callbacks_.begin(); | |
| 130 UploadCallbackList::const_iterator end = user_callbacks_.end(); | |
| 131 for (; iter != end; ++iter) { | |
| 132 base::MessageLoop::current()->PostTask( | |
| 133 FROM_HERE, base::Bind(*iter, result, updated_id)); | |
| 134 } | |
| 135 // Destroy this object and return immediately. | |
|
pavely
2014/05/15 23:27:28
This is unfortunate gentlemen's agreement. I try t
| |
| 136 owner_->DeleteUploadStateFor(attachment_.GetId().GetProto().unique_id()); | |
| 137 return; | |
| 138 } | |
| 139 | |
| 140 AttachmentUploaderImpl::AttachmentUploaderImpl( | |
| 141 const std::string& url_prefix, | |
| 142 const scoped_refptr<net::URLRequestContextGetter>& | |
| 143 url_request_context_getter) | |
| 144 : url_prefix_(url_prefix), | |
| 145 url_request_context_getter_(url_request_context_getter) { | |
| 146 DCHECK(CalledOnValidThread()); | |
| 147 } | |
| 148 | |
| 149 AttachmentUploaderImpl::~AttachmentUploaderImpl() { | |
| 150 DCHECK(CalledOnValidThread()); | |
| 151 } | |
| 152 | |
| 153 void AttachmentUploaderImpl::UploadAttachment(const Attachment& attachment, | |
| 154 const UploadCallback& callback) { | |
| 155 DCHECK(CalledOnValidThread()); | |
| 156 const AttachmentId attachment_id = attachment.GetId(); | |
| 157 const std::string unique_id = attachment_id.GetProto().unique_id(); | |
| 158 DCHECK(!unique_id.empty()); | |
| 159 StateMap::iterator iter = state_map_.find(unique_id); | |
| 160 if (iter == state_map_.end()) { | |
| 161 const GURL url = GetUploadURLForAttachmentId(attachment_id); | |
| 162 scoped_ptr<UploadState> upload_state(new UploadState( | |
| 163 url, url_request_context_getter_, attachment, callback, this)); | |
| 164 state_map_.add(unique_id, upload_state.Pass()); | |
| 165 } else { | |
| 166 DCHECK( | |
| 167 attachment.GetData()->Equals(iter->second->GetAttachment().GetData())); | |
| 168 // We already have an upload for this attachment. "Join" it. | |
| 169 iter->second->AddUserCallback(callback); | |
| 170 } | |
| 171 } | |
| 172 | |
| 173 GURL AttachmentUploaderImpl::GetUploadURLForAttachmentId( | |
| 174 const AttachmentId& attachment_id) const { | |
| 175 return GURL(url_prefix_ + attachment_id.GetProto().unique_id()); | |
| 176 } | |
| 177 | |
| 178 void AttachmentUploaderImpl::DeleteUploadStateFor(const UniqueId& unique_id) { | |
| 179 state_map_.erase(unique_id); | |
| 180 } | |
| 181 | |
| 182 } // namespace syncer | |
| OLD | NEW |