| Index: chrome/browser/webdata/web_data_service_base.cc
|
| diff --git a/chrome/browser/webdata/web_data_service_base.cc b/chrome/browser/webdata/web_data_service_base.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..57891aa14de24c57797554b5410367fb517dbe7b
|
| --- /dev/null
|
| +++ b/chrome/browser/webdata/web_data_service_base.cc
|
| @@ -0,0 +1,135 @@
|
| +// Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#include "chrome/browser/api/webdata/web_data_service_base.h"
|
| +
|
| +#include "base/bind.h"
|
| +#include "base/message_loop.h"
|
| +#include "base/stl_util.h"
|
| +#include "base/threading/thread.h"
|
| +#include "chrome/browser/profiles/profile.h"
|
| +#include "chrome/browser/webdata/web_database_service.h"
|
| +#include "chrome/common/chrome_constants.h"
|
| +#include "chrome/common/chrome_notification_types.h"
|
| +#ifdef DEBUG
|
| +#include "content/public/browser/browser_thread.h"
|
| +#endif
|
| +#include "content/public/browser/notification_details.h"
|
| +#include "content/public/browser/notification_service.h"
|
| +#include "content/public/browser/notification_source.h"
|
| +
|
| +////////////////////////////////////////////////////////////////////////////////
|
| +//
|
| +// WebDataServiceBase implementation.
|
| +//
|
| +////////////////////////////////////////////////////////////////////////////////
|
| +
|
| +using base::Bind;
|
| +using base::Time;
|
| +using content::BrowserThread;
|
| +
|
| +WebDataServiceBase::WebDataServiceBase(const ProfileErrorCallback& callback)
|
| + : db_loaded_(false),
|
| + profile_error_callback_(callback) {
|
| + // WebDataService requires DB thread if instantiated.
|
| + // Set WebDataServiceFactory::GetInstance()->SetTestingFactory(&profile, NULL)
|
| + // if you do not want to instantiate WebDataService in your test.
|
| + DCHECK(BrowserThread::IsWellKnownThread(BrowserThread::DB));
|
| +}
|
| +
|
| +void WebDataServiceBase::ShutdownOnUIThread() {
|
| + db_loaded_ = false;
|
| + BrowserThread::PostTask(
|
| + BrowserThread::DB, FROM_HERE,
|
| + base::Bind(&WebDataServiceBase::ShutdownOnDBThread, this));
|
| + ShutdownDatabase();
|
| +}
|
| +
|
| +void WebDataServiceBase::Init(const base::FilePath& path) {
|
| + wdbs_.reset(new WebDatabaseService(path));
|
| + wdbs_->LoadDatabase(Bind(&WebDataServiceBase::DatabaseInitOnDB, this));
|
| +}
|
| +
|
| +void WebDataServiceBase::UnloadDatabase() {
|
| + if (!wdbs_)
|
| + return;
|
| + wdbs_->UnloadDatabase();
|
| +}
|
| +
|
| +void WebDataServiceBase::ShutdownDatabase() {
|
| + if (!wdbs_)
|
| + return;
|
| + wdbs_->ShutdownDatabase();
|
| +}
|
| +
|
| +void WebDataServiceBase::CancelRequest(Handle h) {
|
| + if (!wdbs_)
|
| + return;
|
| + wdbs_->CancelRequest(h);
|
| +}
|
| +
|
| +content::NotificationSource WebDataServiceBase::GetNotificationSource() {
|
| + return content::Source<WebDataServiceBase>(this);
|
| +}
|
| +
|
| +bool WebDataServiceBase::IsDatabaseLoaded() {
|
| + return db_loaded_;
|
| +}
|
| +
|
| +WebDatabase* WebDataServiceBase::GetDatabase() {
|
| + if (!wdbs_)
|
| + return NULL;
|
| + return wdbs_->GetDatabaseOnDB();
|
| +}
|
| +
|
| +base::SupportsUserData* WebDataServiceBase::GetDBUserData() {
|
| + DCHECK(BrowserThread::CurrentlyOn(BrowserThread::DB));
|
| + if (!db_thread_user_data_)
|
| + db_thread_user_data_.reset(new SupportsUserDataAggregatable());
|
| + return db_thread_user_data_.get();
|
| +}
|
| +
|
| +WebDataServiceBase::~WebDataServiceBase() {
|
| + DCHECK(!db_thread_user_data_.get()) << "Forgot to call ShutdownOnUIThread?";
|
| + wdbs_.reset();
|
| +}
|
| +
|
| +void WebDataServiceBase::ShutdownOnDBThread() {
|
| + DCHECK(BrowserThread::CurrentlyOn(BrowserThread::DB));
|
| + db_thread_user_data_.reset();
|
| +}
|
| +
|
| +////////////////////////////////////////////////////////////////////////////////
|
| +//
|
| +// The following methods are executed on the DB thread.
|
| +//
|
| +////////////////////////////////////////////////////////////////////////////////
|
| +
|
| +void WebDataServiceBase::DBInitFailed(sql::InitStatus sql_status) {
|
| + if (!profile_error_callback_.is_null())
|
| + profile_error_callback_.Run(sql_status);
|
| +}
|
| +
|
| +void WebDataServiceBase::NotifyDatabaseLoadedOnUIThread() {
|
| + db_loaded_ = true;
|
| + // Notify that the database has been initialized.
|
| + content::NotificationService::current()->Notify(
|
| + chrome::NOTIFICATION_WEB_DATABASE_LOADED,
|
| + content::Source<WebDataServiceBase>(this),
|
| + content::NotificationService::NoDetails());
|
| +}
|
| +
|
| +void WebDataServiceBase::DatabaseInitOnDB(sql::InitStatus status) {
|
| + DCHECK(BrowserThread::CurrentlyOn(BrowserThread::DB));
|
| + if (status == sql::INIT_OK) {
|
| + BrowserThread::PostTask(
|
| + BrowserThread::UI, FROM_HERE,
|
| + base::Bind(&WebDataServiceBase::NotifyDatabaseLoadedOnUIThread, this));
|
| + } else {
|
| + BrowserThread::PostTask(
|
| + BrowserThread::UI, FROM_HERE,
|
| + base::Bind(&WebDataServiceBase::DBInitFailed, this, status));
|
| + }
|
| +}
|
| +
|
|
|