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/api/attachments/fake_attachment_store.h" |
| 6 |
| 7 #include "base/bind.h" |
| 8 #include "base/location.h" |
| 9 #include "base/memory/ref_counted_memory.h" |
| 10 #include "base/sequenced_task_runner.h" |
| 11 #include "base/single_thread_task_runner.h" |
| 12 #include "base/thread_task_runner_handle.h" |
| 13 #include "sync/api/attachments/attachment.h" |
| 14 |
| 15 namespace syncer { |
| 16 |
| 17 // Backend is where all the work happens. |
| 18 class FakeAttachmentStore::Backend |
| 19 : public base::RefCountedThreadSafe<FakeAttachmentStore::Backend> { |
| 20 public: |
| 21 // Construct a Backend that posts its results to |frontend_task_runner|. |
| 22 Backend( |
| 23 const scoped_refptr<base::SingleThreadTaskRunner>& frontend_task_runner); |
| 24 |
| 25 void Read(const AttachmentIdList& ids, const ReadCallback& callback); |
| 26 void Write(const AttachmentList& attachments, const WriteCallback& callback); |
| 27 void Drop(const AttachmentIdList& ids, const DropCallback& callback); |
| 28 |
| 29 private: |
| 30 friend class base::RefCountedThreadSafe<Backend>; |
| 31 |
| 32 ~Backend(); |
| 33 |
| 34 scoped_refptr<base::SingleThreadTaskRunner> frontend_task_runner_; |
| 35 AttachmentMap attachments_; |
| 36 }; |
| 37 |
| 38 FakeAttachmentStore::Backend::Backend( |
| 39 const scoped_refptr<base::SingleThreadTaskRunner>& frontend_task_runner) |
| 40 : frontend_task_runner_(frontend_task_runner) {} |
| 41 |
| 42 FakeAttachmentStore::Backend::~Backend() {} |
| 43 |
| 44 void FakeAttachmentStore::Backend::Read(const AttachmentIdList& ids, |
| 45 const ReadCallback& callback) { |
| 46 Result result_code = SUCCESS; |
| 47 AttachmentIdList::const_iterator id_iter = ids.begin(); |
| 48 AttachmentIdList::const_iterator id_end = ids.end(); |
| 49 scoped_ptr<AttachmentMap> result_map(new AttachmentMap); |
| 50 scoped_ptr<AttachmentIdList> unavailable_attachments(new AttachmentIdList); |
| 51 for (; id_iter != id_end; ++id_iter) { |
| 52 const AttachmentId& id = *id_iter; |
| 53 syncer::AttachmentMap::iterator attachment_iter = |
| 54 attachments_.find(*id_iter); |
| 55 if (attachment_iter != attachments_.end()) { |
| 56 const Attachment& attachment = attachment_iter->second; |
| 57 result_map->insert(std::make_pair(id, attachment)); |
| 58 } else { |
| 59 unavailable_attachments->push_back(id); |
| 60 } |
| 61 } |
| 62 if (!unavailable_attachments->empty()) { |
| 63 result_code = UNSPECIFIED_ERROR; |
| 64 } |
| 65 frontend_task_runner_->PostTask( |
| 66 FROM_HERE, |
| 67 base::Bind(callback, |
| 68 result_code, |
| 69 base::Passed(&result_map), |
| 70 base::Passed(&unavailable_attachments))); |
| 71 } |
| 72 |
| 73 void FakeAttachmentStore::Backend::Write(const AttachmentList& attachments, |
| 74 const WriteCallback& callback) { |
| 75 AttachmentList::const_iterator iter = attachments.begin(); |
| 76 AttachmentList::const_iterator end = attachments.end(); |
| 77 for (; iter != end; ++iter) { |
| 78 attachments_.insert(std::make_pair(iter->GetId(), *iter)); |
| 79 } |
| 80 frontend_task_runner_->PostTask(FROM_HERE, base::Bind(callback, SUCCESS)); |
| 81 } |
| 82 |
| 83 void FakeAttachmentStore::Backend::Drop(const AttachmentIdList& ids, |
| 84 const DropCallback& callback) { |
| 85 Result result = SUCCESS; |
| 86 AttachmentIdList::const_iterator ids_iter = ids.begin(); |
| 87 AttachmentIdList::const_iterator ids_end = ids.end(); |
| 88 for (; ids_iter != ids_end; ++ids_iter) { |
| 89 AttachmentMap::iterator attachments_iter = attachments_.find(*ids_iter); |
| 90 if (attachments_iter != attachments_.end()) { |
| 91 attachments_.erase(attachments_iter); |
| 92 } |
| 93 } |
| 94 frontend_task_runner_->PostTask(FROM_HERE, base::Bind(callback, result)); |
| 95 } |
| 96 |
| 97 FakeAttachmentStore::FakeAttachmentStore( |
| 98 const scoped_refptr<base::SequencedTaskRunner>& backend_task_runner) |
| 99 : backend_(new Backend(base::ThreadTaskRunnerHandle::Get())), |
| 100 backend_task_runner_(backend_task_runner) {} |
| 101 |
| 102 FakeAttachmentStore::~FakeAttachmentStore() {} |
| 103 |
| 104 void FakeAttachmentStore::Read(const AttachmentIdList& ids, |
| 105 const ReadCallback& callback) { |
| 106 backend_task_runner_->PostTask( |
| 107 FROM_HERE, |
| 108 base::Bind(&FakeAttachmentStore::Backend::Read, backend_, ids, callback)); |
| 109 } |
| 110 |
| 111 void FakeAttachmentStore::Write(const AttachmentList& attachments, |
| 112 const WriteCallback& callback) { |
| 113 backend_task_runner_->PostTask( |
| 114 FROM_HERE, |
| 115 base::Bind(&FakeAttachmentStore::Backend::Write, |
| 116 backend_, |
| 117 attachments, |
| 118 callback)); |
| 119 } |
| 120 |
| 121 void FakeAttachmentStore::Drop(const AttachmentIdList& ids, |
| 122 const DropCallback& callback) { |
| 123 backend_task_runner_->PostTask( |
| 124 FROM_HERE, |
| 125 base::Bind(&FakeAttachmentStore::Backend::Drop, backend_, ids, callback)); |
| 126 } |
| 127 |
| 128 } // namespace syncer |
OLD | NEW |