OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 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 "components/webdata/common/web_database_service.h" | 5 #include "components/webdata/common/web_database_service.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/location.h" | 8 #include "base/location.h" |
9 #include "components/webdata/common/web_data_request_manager.h" | 9 #include "components/webdata/common/web_data_request_manager.h" |
10 #include "components/webdata/common/web_data_results.h" | 10 #include "components/webdata/common/web_data_results.h" |
11 #include "components/webdata/common/web_data_service_backend.h" | 11 #include "components/webdata/common/web_data_service_backend.h" |
12 #include "components/webdata/common/web_data_service_consumer.h" | 12 #include "components/webdata/common/web_data_service_consumer.h" |
13 | 13 |
14 using base::Bind; | 14 using base::Bind; |
15 using base::FilePath; | 15 using base::FilePath; |
16 using content::BrowserThread; | |
17 | 16 |
18 // Receives messages from the backend on the DB thread, posts them to | 17 // Receives messages from the backend on the DB thread, posts them to |
19 // WebDatabaseService on the UI thread. | 18 // WebDatabaseService on the UI thread. |
20 class WebDatabaseService::BackendDelegate : | 19 class WebDatabaseService::BackendDelegate : |
21 public WebDataServiceBackend::Delegate { | 20 public WebDataServiceBackend::Delegate { |
22 public: | 21 public: |
23 BackendDelegate( | 22 BackendDelegate( |
24 const base::WeakPtr<WebDatabaseService>& web_database_service) | 23 const base::WeakPtr<WebDatabaseService>& web_database_service) |
25 : web_database_service_(web_database_service) { | 24 : web_database_service_(web_database_service), |
| 25 callback_thread_(base::MessageLoopProxy::current()) { |
26 } | 26 } |
27 | 27 |
28 virtual void DBLoaded(sql::InitStatus status) OVERRIDE { | 28 virtual void DBLoaded(sql::InitStatus status) OVERRIDE { |
29 BrowserThread::PostTask( | 29 callback_thread_->PostTask( |
30 BrowserThread::UI, | |
31 FROM_HERE, | 30 FROM_HERE, |
32 base::Bind(&WebDatabaseService::OnDatabaseLoadDone, | 31 base::Bind(&WebDatabaseService::OnDatabaseLoadDone, |
33 web_database_service_, | 32 web_database_service_, |
34 status)); | 33 status)); |
35 } | 34 } |
36 private: | 35 private: |
37 const base::WeakPtr<WebDatabaseService> web_database_service_; | 36 const base::WeakPtr<WebDatabaseService> web_database_service_; |
| 37 scoped_refptr<base::MessageLoopProxy> callback_thread_; |
38 }; | 38 }; |
39 | 39 |
40 WebDatabaseService::WebDatabaseService( | 40 WebDatabaseService::WebDatabaseService( |
41 const base::FilePath& path, | 41 const base::FilePath& path, |
42 const scoped_refptr<base::MessageLoopProxy>& ui_thread) | 42 const scoped_refptr<base::MessageLoopProxy>& ui_thread, |
| 43 const scoped_refptr<base::MessageLoopProxy>& db_thread) |
43 : base::RefCountedDeleteOnMessageLoop<WebDatabaseService>(ui_thread), | 44 : base::RefCountedDeleteOnMessageLoop<WebDatabaseService>(ui_thread), |
44 path_(path), | 45 path_(path), |
45 weak_ptr_factory_(this), | 46 weak_ptr_factory_(this), |
46 db_loaded_(false) { | 47 db_loaded_(false), |
| 48 db_thread_(db_thread) { |
47 // WebDatabaseService should be instantiated on UI thread. | 49 // WebDatabaseService should be instantiated on UI thread. |
48 DCHECK(ui_thread->BelongsToCurrentThread()); | 50 DCHECK(ui_thread->BelongsToCurrentThread()); |
49 // WebDatabaseService requires DB thread if instantiated. | 51 // WebDatabaseService requires DB thread if instantiated. |
50 DCHECK(BrowserThread::IsWellKnownThread(BrowserThread::DB)); | 52 DCHECK(db_thread); |
51 } | 53 } |
52 | 54 |
53 WebDatabaseService::~WebDatabaseService() { | 55 WebDatabaseService::~WebDatabaseService() { |
54 } | 56 } |
55 | 57 |
56 void WebDatabaseService::AddTable(scoped_ptr<WebDatabaseTable> table) { | 58 void WebDatabaseService::AddTable(scoped_ptr<WebDatabaseTable> table) { |
57 if (!wds_backend_.get()) { | 59 if (!wds_backend_.get()) { |
58 wds_backend_ = new WebDataServiceBackend( | 60 wds_backend_ = new WebDataServiceBackend( |
59 path_, new BackendDelegate(weak_ptr_factory_.GetWeakPtr())); | 61 path_, new BackendDelegate(weak_ptr_factory_.GetWeakPtr()), |
| 62 db_thread_); |
60 } | 63 } |
61 wds_backend_->AddTable(table.Pass()); | 64 wds_backend_->AddTable(table.Pass()); |
62 } | 65 } |
63 | 66 |
64 void WebDatabaseService::LoadDatabase() { | 67 void WebDatabaseService::LoadDatabase() { |
65 DCHECK(wds_backend_.get()); | 68 DCHECK(wds_backend_.get()); |
66 | 69 |
67 BrowserThread::PostTask( | 70 db_thread_->PostTask( |
68 BrowserThread::DB, | |
69 FROM_HERE, | 71 FROM_HERE, |
70 Bind(&WebDataServiceBackend::InitDatabase, wds_backend_)); | 72 Bind(&WebDataServiceBackend::InitDatabase, wds_backend_)); |
71 } | 73 } |
72 | 74 |
73 void WebDatabaseService::UnloadDatabase() { | 75 void WebDatabaseService::UnloadDatabase() { |
74 db_loaded_ = false; | 76 db_loaded_ = false; |
75 if (!wds_backend_.get()) | 77 if (!wds_backend_.get()) |
76 return; | 78 return; |
77 BrowserThread::PostTask(BrowserThread::DB, FROM_HERE, | 79 db_thread_->PostTask(FROM_HERE, |
78 Bind(&WebDataServiceBackend::ShutdownDatabase, | 80 Bind(&WebDataServiceBackend::ShutdownDatabase, |
79 wds_backend_, true)); | 81 wds_backend_, true)); |
80 } | 82 } |
81 | 83 |
82 void WebDatabaseService::ShutdownDatabase() { | 84 void WebDatabaseService::ShutdownDatabase() { |
83 db_loaded_ = false; | 85 db_loaded_ = false; |
84 weak_ptr_factory_.InvalidateWeakPtrs(); | 86 weak_ptr_factory_.InvalidateWeakPtrs(); |
85 loaded_callbacks_.clear(); | 87 loaded_callbacks_.clear(); |
86 error_callbacks_.clear(); | 88 error_callbacks_.clear(); |
87 if (!wds_backend_.get()) | 89 if (!wds_backend_.get()) |
88 return; | 90 return; |
89 BrowserThread::PostTask(BrowserThread::DB, FROM_HERE, | 91 db_thread_->PostTask(FROM_HERE, |
90 Bind(&WebDataServiceBackend::ShutdownDatabase, | 92 Bind(&WebDataServiceBackend::ShutdownDatabase, |
91 wds_backend_, false)); | 93 wds_backend_, false)); |
92 } | 94 } |
93 | 95 |
94 WebDatabase* WebDatabaseService::GetDatabaseOnDB() const { | 96 WebDatabase* WebDatabaseService::GetDatabaseOnDB() const { |
95 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::DB)); | 97 DCHECK(db_thread_->BelongsToCurrentThread()); |
96 if (!wds_backend_.get()) | 98 if (!wds_backend_.get()) |
97 return NULL; | 99 return NULL; |
98 return wds_backend_->database(); | 100 return wds_backend_->database(); |
99 } | 101 } |
100 | 102 |
101 scoped_refptr<WebDataServiceBackend> WebDatabaseService::GetBackend() const { | 103 scoped_refptr<WebDataServiceBackend> WebDatabaseService::GetBackend() const { |
102 return wds_backend_; | 104 return wds_backend_; |
103 } | 105 } |
104 | 106 |
105 void WebDatabaseService::ScheduleDBTask( | 107 void WebDatabaseService::ScheduleDBTask( |
106 const tracked_objects::Location& from_here, | 108 const tracked_objects::Location& from_here, |
107 const WriteTask& task) { | 109 const WriteTask& task) { |
108 if (!wds_backend_.get()) { | 110 if (!wds_backend_.get()) { |
109 NOTREACHED() << "Task scheduled after Shutdown()"; | 111 NOTREACHED() << "Task scheduled after Shutdown()"; |
110 return; | 112 return; |
111 } | 113 } |
112 | 114 |
113 scoped_ptr<WebDataRequest> request( | 115 scoped_ptr<WebDataRequest> request( |
114 new WebDataRequest(NULL, wds_backend_->request_manager().get())); | 116 new WebDataRequest(NULL, wds_backend_->request_manager().get())); |
115 | 117 |
116 BrowserThread::PostTask(BrowserThread::DB, from_here, | 118 db_thread_->PostTask(from_here, |
117 Bind(&WebDataServiceBackend::DBWriteTaskWrapper, wds_backend_, | 119 Bind(&WebDataServiceBackend::DBWriteTaskWrapper, wds_backend_, |
118 task, base::Passed(&request))); | 120 task, base::Passed(&request))); |
119 } | 121 } |
120 | 122 |
121 WebDataServiceBase::Handle WebDatabaseService::ScheduleDBTaskWithResult( | 123 WebDataServiceBase::Handle WebDatabaseService::ScheduleDBTaskWithResult( |
122 const tracked_objects::Location& from_here, | 124 const tracked_objects::Location& from_here, |
123 const ReadTask& task, | 125 const ReadTask& task, |
124 WebDataServiceConsumer* consumer) { | 126 WebDataServiceConsumer* consumer) { |
125 DCHECK(consumer); | 127 DCHECK(consumer); |
126 WebDataServiceBase::Handle handle = 0; | 128 WebDataServiceBase::Handle handle = 0; |
127 | 129 |
128 if (!wds_backend_.get()) { | 130 if (!wds_backend_.get()) { |
129 NOTREACHED() << "Task scheduled after Shutdown()"; | 131 NOTREACHED() << "Task scheduled after Shutdown()"; |
130 return handle; | 132 return handle; |
131 } | 133 } |
132 | 134 |
133 scoped_ptr<WebDataRequest> request( | 135 scoped_ptr<WebDataRequest> request( |
134 new WebDataRequest(consumer, wds_backend_->request_manager().get())); | 136 new WebDataRequest(consumer, wds_backend_->request_manager().get())); |
135 handle = request->GetHandle(); | 137 handle = request->GetHandle(); |
136 | 138 |
137 BrowserThread::PostTask(BrowserThread::DB, from_here, | 139 db_thread_->PostTask(from_here, |
138 Bind(&WebDataServiceBackend::DBReadTaskWrapper, wds_backend_, | 140 Bind(&WebDataServiceBackend::DBReadTaskWrapper, wds_backend_, |
139 task, base::Passed(&request))); | 141 task, base::Passed(&request))); |
140 | 142 |
141 return handle; | 143 return handle; |
142 } | 144 } |
143 | 145 |
144 void WebDatabaseService::CancelRequest(WebDataServiceBase::Handle h) { | 146 void WebDatabaseService::CancelRequest(WebDataServiceBase::Handle h) { |
145 if (!wds_backend_.get()) | 147 if (!wds_backend_.get()) |
146 return; | 148 return; |
147 wds_backend_->request_manager()->CancelRequest(h); | 149 wds_backend_->request_manager()->CancelRequest(h); |
(...skipping 22 matching lines...) Expand all Loading... |
170 } else { | 172 } else { |
171 // Notify that the database load failed. | 173 // Notify that the database load failed. |
172 for (size_t i = 0; i < error_callbacks_.size(); i++) { | 174 for (size_t i = 0; i < error_callbacks_.size(); i++) { |
173 if (!error_callbacks_[i].is_null()) | 175 if (!error_callbacks_[i].is_null()) |
174 error_callbacks_[i].Run(status); | 176 error_callbacks_[i].Run(status); |
175 } | 177 } |
176 | 178 |
177 error_callbacks_.clear(); | 179 error_callbacks_.clear(); |
178 } | 180 } |
179 } | 181 } |
OLD | NEW |