fake_attachment_store.cc revision 5d1f7b1de12d16ceb2c938c56701a3e8bfa558f7
15d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// Copyright 2014 The Chromium Authors. All rights reserved. 25d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// found in the LICENSE file. 45d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 55d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "sync/api/attachments/fake_attachment_store.h" 65d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 75d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/bind.h" 85d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/location.h" 95d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/memory/ref_counted_memory.h" 105d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/message_loop/message_loop_proxy.h" 115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/sequenced_task_runner.h" 125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/single_thread_task_runner.h" 135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "sync/api/attachments/attachment.h" 145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)namespace syncer { 165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 175d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// Backend is where all the work happens. 185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)class FakeAttachmentStore::Backend 195d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : public base::RefCountedThreadSafe<FakeAttachmentStore::Backend> { 205d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) public: 215d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // Construct a Backend that posts its results to |frontend_task_runner|. 225d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) Backend( 235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const scoped_refptr<base::SingleThreadTaskRunner>& frontend_task_runner); 245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 255d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void Read(const sync_pb::AttachmentId& id, const ReadCallback& callback); 265d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void Write(const sync_pb::AttachmentId& id, 275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const scoped_refptr<base::RefCountedMemory>& bytes, 285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const WriteCallback& callback); 295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void Drop(const sync_pb::AttachmentId& id, const DropCallback& callback); 305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 315d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) private: 325d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) friend class base::RefCountedThreadSafe<Backend>; 335d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) typedef std::string UniqueId; 345d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) typedef std::map<UniqueId, Attachment*> AttachmentMap; 355d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 365d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ~Backend(); 375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) Result RemoveAttachment(const sync_pb::AttachmentId& id); 385d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 395d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) scoped_refptr<base::SingleThreadTaskRunner> frontend_task_runner_; 405d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) AttachmentMap attachments_; 415d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) STLValueDeleter<AttachmentMap> attachments_value_deleter_; 425d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)}; 435d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 445d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)FakeAttachmentStore::Backend::Backend( 455d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const scoped_refptr<base::SingleThreadTaskRunner>& frontend_task_runner) 465d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : frontend_task_runner_(frontend_task_runner), 475d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) attachments_value_deleter_(&attachments_) {} 485d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 495d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)FakeAttachmentStore::Backend::~Backend() {} 505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void FakeAttachmentStore::Backend::Read(const sync_pb::AttachmentId& id, 525d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const ReadCallback& callback) { 535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) AttachmentMap::iterator iter = attachments_.find(id.unique_id()); 545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) scoped_ptr<Attachment> attachment; 555d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) Result result = NOT_FOUND; 565d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (iter != attachments_.end()) { 575d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) attachment.reset(new Attachment(*iter->second)); 585d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) result = SUCCESS; 595d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 605d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) frontend_task_runner_->PostTask( 615d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) FROM_HERE, base::Bind(callback, result, base::Passed(&attachment))); 625d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 635d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 645d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void FakeAttachmentStore::Backend::Write( 655d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const sync_pb::AttachmentId& id, 665d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const scoped_refptr<base::RefCountedMemory>& bytes, 675d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const WriteCallback& callback) { 685d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) scoped_ptr<Attachment> attachment = Attachment::CreateWithId(id, bytes); 695d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) RemoveAttachment(id); 705d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) attachments_.insert( 715d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) AttachmentMap::value_type(id.unique_id(), attachment.release())); 725d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) frontend_task_runner_->PostTask(FROM_HERE, base::Bind(callback, SUCCESS)); 735d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 745d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 755d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void FakeAttachmentStore::Backend::Drop(const sync_pb::AttachmentId& id, 765d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const DropCallback& callback) { 775d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) Result result = RemoveAttachment(id); 785d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) frontend_task_runner_->PostTask(FROM_HERE, base::Bind(callback, result)); 795d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 805d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 815d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)AttachmentStore::Result FakeAttachmentStore::Backend::RemoveAttachment( 825d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const sync_pb::AttachmentId& id) { 835d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) Result result = NOT_FOUND; 845d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) AttachmentMap::iterator iter = attachments_.find(id.unique_id()); 855d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (iter != attachments_.end()) { 865d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) delete iter->second; 875d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) attachments_.erase(iter); 885d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) result = SUCCESS; 895d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 905d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return result; 915d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 925d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 935d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)FakeAttachmentStore::FakeAttachmentStore( 945d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const scoped_refptr<base::SequencedTaskRunner>& backend_task_runner) 955d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : backend_(new Backend(base::MessageLoopProxy::current())), 965d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) backend_task_runner_(backend_task_runner) {} 975d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 985d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)FakeAttachmentStore::~FakeAttachmentStore() {} 995d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1005d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void FakeAttachmentStore::Read(const sync_pb::AttachmentId& id, 1015d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const ReadCallback& callback) { 1025d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) backend_task_runner_->PostTask( 1035d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) FROM_HERE, 1045d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::Bind(&FakeAttachmentStore::Backend::Read, backend_, id, callback)); 1055d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 1065d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1075d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void FakeAttachmentStore::Write( 1085d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const sync_pb::AttachmentId& id, 1095d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const scoped_refptr<base::RefCountedMemory>& bytes, 1105d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const WriteCallback& callback) { 1115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) backend_task_runner_->PostTask( 1125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) FROM_HERE, 1135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::Bind( 1145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) &FakeAttachmentStore::Backend::Write, backend_, id, bytes, callback)); 1155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 1165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1175d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void FakeAttachmentStore::Drop(const sync_pb::AttachmentId& id, 1185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const DropCallback& callback) { 1195d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) backend_task_runner_->PostTask( 1205d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) FROM_HERE, 1215d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::Bind(&FakeAttachmentStore::Backend::Drop, backend_, id, callback)); 1225d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 1235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} // namespace syncer 125