1 // Copyright 2013 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 "components/webdata/common/web_data_service_backend.h"
6
7 #include "base/bind.h"
8 #include "base/location.h"
9 #include "components/webdata/common/web_data_request_manager.h"
10 #include "components/webdata/common/web_database.h"
11 #include "components/webdata/common/web_database_table.h"
12
13 using base::Bind;
14 using base::FilePath;
15
WebDataServiceBackend(const FilePath & path,Delegate * delegate,const scoped_refptr<base::MessageLoopProxy> & db_thread)16 WebDataServiceBackend::WebDataServiceBackend(
17 const FilePath& path,
18 Delegate* delegate,
19 const scoped_refptr<base::MessageLoopProxy>& db_thread)
20 : base::RefCountedDeleteOnMessageLoop<WebDataServiceBackend>(db_thread),
21 db_path_(path),
22 request_manager_(new WebDataRequestManager()),
23 init_status_(sql::INIT_FAILURE),
24 init_complete_(false),
25 delegate_(delegate) {
26 }
27
AddTable(scoped_ptr<WebDatabaseTable> table)28 void WebDataServiceBackend::AddTable(scoped_ptr<WebDatabaseTable> table) {
29 DCHECK(!db_.get());
30 tables_.push_back(table.release());
31 }
32
InitDatabase()33 void WebDataServiceBackend::InitDatabase() {
34 LoadDatabaseIfNecessary();
35 if (delegate_) {
36 delegate_->DBLoaded(init_status_);
37 }
38 }
39
LoadDatabaseIfNecessary()40 sql::InitStatus WebDataServiceBackend::LoadDatabaseIfNecessary() {
41 if (init_complete_ || db_path_.empty()) {
42 return init_status_;
43 }
44 init_complete_ = true;
45 db_.reset(new WebDatabase());
46
47 for (ScopedVector<WebDatabaseTable>::iterator it = tables_.begin();
48 it != tables_.end();
49 ++it) {
50 db_->AddTable(*it);
51 }
52
53 init_status_ = db_->Init(db_path_);
54 if (init_status_ != sql::INIT_OK) {
55 LOG(ERROR) << "Cannot initialize the web database: " << init_status_;
56 db_.reset(NULL);
57 return init_status_;
58 }
59
60 db_->BeginTransaction();
61 return init_status_;
62 }
63
ShutdownDatabase(bool should_reinit)64 void WebDataServiceBackend::ShutdownDatabase(bool should_reinit) {
65 if (db_ && init_status_ == sql::INIT_OK)
66 db_->CommitTransaction();
67 db_.reset(NULL);
68 init_complete_ = !should_reinit; // Setting init_complete_ to true will ensure
69 // that the init sequence is not re-run.
70
71 init_status_ = sql::INIT_FAILURE;
72 }
73
DBWriteTaskWrapper(const WebDatabaseService::WriteTask & task,scoped_ptr<WebDataRequest> request)74 void WebDataServiceBackend::DBWriteTaskWrapper(
75 const WebDatabaseService::WriteTask& task,
76 scoped_ptr<WebDataRequest> request) {
77 if (request->IsCancelled())
78 return;
79
80 ExecuteWriteTask(task);
81 request_manager_->RequestCompleted(request.Pass());
82 }
83
ExecuteWriteTask(const WebDatabaseService::WriteTask & task)84 void WebDataServiceBackend::ExecuteWriteTask(
85 const WebDatabaseService::WriteTask& task) {
86 LoadDatabaseIfNecessary();
87 if (db_ && init_status_ == sql::INIT_OK) {
88 WebDatabase::State state = task.Run(db_.get());
89 if (state == WebDatabase::COMMIT_NEEDED)
90 Commit();
91 }
92 }
93
DBReadTaskWrapper(const WebDatabaseService::ReadTask & task,scoped_ptr<WebDataRequest> request)94 void WebDataServiceBackend::DBReadTaskWrapper(
95 const WebDatabaseService::ReadTask& task,
96 scoped_ptr<WebDataRequest> request) {
97 if (request->IsCancelled())
98 return;
99
100 request->SetResult(ExecuteReadTask(task).Pass());
101 request_manager_->RequestCompleted(request.Pass());
102 }
103
ExecuteReadTask(const WebDatabaseService::ReadTask & task)104 scoped_ptr<WDTypedResult> WebDataServiceBackend::ExecuteReadTask(
105 const WebDatabaseService::ReadTask& task) {
106 LoadDatabaseIfNecessary();
107 if (db_ && init_status_ == sql::INIT_OK) {
108 return task.Run(db_.get());
109 }
110 return scoped_ptr<WDTypedResult>();
111 }
112
~WebDataServiceBackend()113 WebDataServiceBackend::~WebDataServiceBackend() {
114 ShutdownDatabase(false);
115 }
116
Commit()117 void WebDataServiceBackend::Commit() {
118 if (db_ && init_status_ == sql::INIT_OK) {
119 db_->CommitTransaction();
120 db_->BeginTransaction();
121 } else {
122 NOTREACHED() << "Commit scheduled after Shutdown()";
123 }
124 }
125