| 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 |