OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 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 "sync/internal_api/public/attachments/attachment_uploader_impl.h" | 5 #include "sync/internal_api/public/attachments/attachment_uploader_impl.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
9 #include "base/threading/non_thread_safe.h" | 9 #include "base/threading/non_thread_safe.h" |
10 #include "google_apis/gaia/gaia_constants.h" | 10 #include "google_apis/gaia/gaia_constants.h" |
11 #include "net/base/load_flags.h" | 11 #include "net/base/load_flags.h" |
12 #include "net/http/http_status_code.h" | 12 #include "net/http/http_status_code.h" |
13 #include "net/url_request/url_fetcher.h" | 13 #include "net/url_request/url_fetcher.h" |
14 #include "net/url_request/url_fetcher_delegate.h" | 14 #include "net/url_request/url_fetcher_delegate.h" |
15 #include "sync/api/attachments/attachment.h" | 15 #include "sync/api/attachments/attachment.h" |
16 #include "sync/protocol/sync.pb.h" | 16 #include "sync/protocol/sync.pb.h" |
| 17 #include "url/gurl.h" |
17 | 18 |
18 namespace { | 19 namespace { |
19 | 20 |
20 const char kContentType[] = "application/octet-stream"; | 21 const char kContentType[] = "application/octet-stream"; |
21 const char kAttachments[] = "attachments/"; | |
22 | 22 |
23 } // namespace | 23 } // namespace |
24 | 24 |
25 namespace syncer { | 25 namespace syncer { |
26 | 26 |
27 // Encapsulates all the state associated with a single upload. | 27 // Encapsulates all the state associated with a single upload. |
28 class AttachmentUploaderImpl::UploadState : public net::URLFetcherDelegate, | 28 class AttachmentUploaderImpl::UploadState : public net::URLFetcherDelegate, |
29 public OAuth2TokenService::Consumer, | 29 public OAuth2TokenService::Consumer, |
30 public base::NonThreadSafe { | 30 public base::NonThreadSafe { |
31 public: | 31 public: |
(...skipping 170 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
202 for (; iter != end; ++iter) { | 202 for (; iter != end; ++iter) { |
203 base::MessageLoop::current()->PostTask( | 203 base::MessageLoop::current()->PostTask( |
204 FROM_HERE, base::Bind(*iter, result, attachment_id)); | 204 FROM_HERE, base::Bind(*iter, result, attachment_id)); |
205 } | 205 } |
206 // Destroy this object and return immediately. | 206 // Destroy this object and return immediately. |
207 owner_->DeleteUploadStateFor(attachment_.GetId().GetProto().unique_id()); | 207 owner_->DeleteUploadStateFor(attachment_.GetId().GetProto().unique_id()); |
208 return; | 208 return; |
209 } | 209 } |
210 | 210 |
211 AttachmentUploaderImpl::AttachmentUploaderImpl( | 211 AttachmentUploaderImpl::AttachmentUploaderImpl( |
212 const GURL& sync_service_url, | 212 const std::string& url_prefix, |
213 const scoped_refptr<net::URLRequestContextGetter>& | 213 const scoped_refptr<net::URLRequestContextGetter>& |
214 url_request_context_getter, | 214 url_request_context_getter, |
215 const std::string& account_id, | 215 const std::string& account_id, |
216 const OAuth2TokenService::ScopeSet& scopes, | 216 const OAuth2TokenService::ScopeSet& scopes, |
217 scoped_ptr<OAuth2TokenServiceRequest::TokenServiceProvider> | 217 scoped_ptr<OAuth2TokenServiceRequest::TokenServiceProvider> |
218 token_service_provider) | 218 token_service_provider) |
219 : sync_service_url_(sync_service_url), | 219 : url_prefix_(url_prefix), |
220 url_request_context_getter_(url_request_context_getter), | 220 url_request_context_getter_(url_request_context_getter), |
221 account_id_(account_id), | 221 account_id_(account_id), |
222 scopes_(scopes), | 222 scopes_(scopes), |
223 token_service_provider_(token_service_provider.Pass()) { | 223 token_service_provider_(token_service_provider.Pass()) { |
224 DCHECK(CalledOnValidThread()); | 224 DCHECK(CalledOnValidThread()); |
225 DCHECK(token_service_provider_); | 225 DCHECK(token_service_provider_); |
226 } | 226 } |
227 | 227 |
228 AttachmentUploaderImpl::~AttachmentUploaderImpl() { | 228 AttachmentUploaderImpl::~AttachmentUploaderImpl() { |
229 DCHECK(CalledOnValidThread()); | 229 DCHECK(CalledOnValidThread()); |
230 } | 230 } |
231 | 231 |
232 void AttachmentUploaderImpl::UploadAttachment(const Attachment& attachment, | 232 void AttachmentUploaderImpl::UploadAttachment(const Attachment& attachment, |
233 const UploadCallback& callback) { | 233 const UploadCallback& callback) { |
234 DCHECK(CalledOnValidThread()); | 234 DCHECK(CalledOnValidThread()); |
235 const AttachmentId attachment_id = attachment.GetId(); | 235 const AttachmentId attachment_id = attachment.GetId(); |
236 const std::string unique_id = attachment_id.GetProto().unique_id(); | 236 const std::string unique_id = attachment_id.GetProto().unique_id(); |
237 DCHECK(!unique_id.empty()); | 237 DCHECK(!unique_id.empty()); |
238 StateMap::iterator iter = state_map_.find(unique_id); | 238 StateMap::iterator iter = state_map_.find(unique_id); |
239 if (iter == state_map_.end()) { | 239 if (iter == state_map_.end()) { |
240 const GURL url = GetURLForAttachmentId(sync_service_url_, attachment_id); | 240 const GURL url = GetUploadURLForAttachmentId(attachment_id); |
241 scoped_ptr<UploadState> upload_state( | 241 scoped_ptr<UploadState> upload_state( |
242 new UploadState(url, | 242 new UploadState(url, |
243 url_request_context_getter_, | 243 url_request_context_getter_, |
244 attachment, | 244 attachment, |
245 callback, | 245 callback, |
246 account_id_, | 246 account_id_, |
247 scopes_, | 247 scopes_, |
248 token_service_provider_.get(), | 248 token_service_provider_.get(), |
249 this)); | 249 this)); |
250 state_map_.add(unique_id, upload_state.Pass()); | 250 state_map_.add(unique_id, upload_state.Pass()); |
251 } else { | 251 } else { |
252 DCHECK( | 252 DCHECK( |
253 attachment.GetData()->Equals(iter->second->GetAttachment().GetData())); | 253 attachment.GetData()->Equals(iter->second->GetAttachment().GetData())); |
254 // We already have an upload for this attachment. "Join" it. | 254 // We already have an upload for this attachment. "Join" it. |
255 iter->second->AddUserCallback(callback); | 255 iter->second->AddUserCallback(callback); |
256 } | 256 } |
257 } | 257 } |
258 | 258 |
259 // Static. | 259 GURL AttachmentUploaderImpl::GetUploadURLForAttachmentId( |
260 GURL AttachmentUploaderImpl::GetURLForAttachmentId( | 260 const AttachmentId& attachment_id) const { |
261 const GURL& sync_service_url, | 261 return GURL(url_prefix_ + attachment_id.GetProto().unique_id()); |
262 const AttachmentId& attachment_id) { | |
263 std::string path = sync_service_url.path(); | |
264 if (path.empty() || *path.rbegin() != '/') { | |
265 path += '/'; | |
266 } | |
267 path += kAttachments; | |
268 path += attachment_id.GetProto().unique_id(); | |
269 GURL::Replacements replacements; | |
270 replacements.SetPathStr(path); | |
271 return sync_service_url.ReplaceComponents(replacements); | |
272 } | 262 } |
273 | 263 |
274 void AttachmentUploaderImpl::DeleteUploadStateFor(const UniqueId& unique_id) { | 264 void AttachmentUploaderImpl::DeleteUploadStateFor(const UniqueId& unique_id) { |
275 state_map_.erase(unique_id); | 265 state_map_.erase(unique_id); |
276 } | 266 } |
277 | 267 |
278 } // namespace syncer | 268 } // namespace syncer |
OLD | NEW |