Index: content/browser/indexed_db/database_impl.cc |
diff --git a/content/browser/indexed_db/database_impl.cc b/content/browser/indexed_db/database_impl.cc |
index 4a22cda522deb3481d94c0ef788de132722c2970..feaec66892c446dba18123239ae5f6de3d58f510 100644 |
--- a/content/browser/indexed_db/database_impl.cc |
+++ b/content/browser/indexed_db/database_impl.cc |
@@ -6,6 +6,7 @@ |
#include "base/memory/ptr_util.h" |
#include "base/metrics/histogram_macros.h" |
+#include "base/sequenced_task_runner.h" |
#include "base/threading/thread_task_runner_handle.h" |
#include "content/browser/bad_message.h" |
#include "content/browser/child_process_security_policy_impl.h" |
@@ -28,13 +29,16 @@ const char kInvalidBlobUuid[] = "Blob does not exist"; |
const char kInvalidBlobFilePath[] = "Blob file path is invalid"; |
} // namespace |
+// Expect to be created on IO thread, and called/destroyed on IDB thread. |
jsbell
2017/04/11 16:58:03
Can you use DCHECK_CURRENTLY_ON and DCHECK(idb_run
dmurph
2017/04/11 19:32:49
Done.
|
class DatabaseImpl::IDBThreadHelper { |
public: |
IDBThreadHelper(std::unique_ptr<IndexedDBConnection> connection, |
const url::Origin& origin, |
- scoped_refptr<IndexedDBDispatcherHost> dispatcher_host); |
+ scoped_refptr<IndexedDBContextImpl> indexed_db_context); |
~IDBThreadHelper(); |
+ void ConnectionOpened(); |
+ |
void CreateObjectStore(int64_t transaction_id, |
int64_t object_store_id, |
const base::string16& name, |
@@ -130,21 +134,25 @@ class DatabaseImpl::IDBThreadHelper { |
void AckReceivedBlobs(const std::vector<std::string>& uuids); |
private: |
- scoped_refptr<IndexedDBDispatcherHost> dispatcher_host_; |
+ scoped_refptr<IndexedDBContextImpl> indexed_db_context_; |
std::unique_ptr<IndexedDBConnection> connection_; |
const url::Origin origin_; |
base::WeakPtrFactory<IDBThreadHelper> weak_factory_; |
}; |
-DatabaseImpl::DatabaseImpl( |
- std::unique_ptr<IndexedDBConnection> connection, |
- const url::Origin& origin, |
- scoped_refptr<IndexedDBDispatcherHost> dispatcher_host) |
+DatabaseImpl::DatabaseImpl(std::unique_ptr<IndexedDBConnection> connection, |
+ const url::Origin& origin, |
+ IndexedDBDispatcherHost* dispatcher_host, |
+ scoped_refptr<base::SequencedTaskRunner> idb_runner) |
: dispatcher_host_(dispatcher_host), |
origin_(origin), |
- idb_runner_(base::ThreadTaskRunnerHandle::Get()) { |
+ idb_runner_(std::move(idb_runner)) { |
+ DCHECK(connection); |
helper_ = new IDBThreadHelper(std::move(connection), origin, |
- std::move(dispatcher_host)); |
+ dispatcher_host->context()); |
+ idb_runner_->PostTask(FROM_HERE, |
+ base::Bind(&IDBThreadHelper::ConnectionOpened, |
+ base::Unretained(helper_))); |
} |
DatabaseImpl::~DatabaseImpl() { |
@@ -226,8 +234,9 @@ void DatabaseImpl::Get( |
const IndexedDBKeyRange& key_range, |
bool key_only, |
::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info) { |
- scoped_refptr<IndexedDBCallbacks> callbacks(new IndexedDBCallbacks( |
- dispatcher_host_.get(), origin_, std::move(callbacks_info))); |
+ scoped_refptr<IndexedDBCallbacks> callbacks( |
+ new IndexedDBCallbacks(dispatcher_host_->AsWeakPtr(), origin_, |
+ std::move(callbacks_info), idb_runner_)); |
idb_runner_->PostTask( |
FROM_HERE, base::Bind(&IDBThreadHelper::Get, base::Unretained(helper_), |
transaction_id, object_store_id, index_id, |
@@ -242,8 +251,9 @@ void DatabaseImpl::GetAll( |
bool key_only, |
int64_t max_count, |
::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info) { |
- scoped_refptr<IndexedDBCallbacks> callbacks(new IndexedDBCallbacks( |
- dispatcher_host_.get(), origin_, std::move(callbacks_info))); |
+ scoped_refptr<IndexedDBCallbacks> callbacks( |
+ new IndexedDBCallbacks(dispatcher_host_->AsWeakPtr(), origin_, |
+ std::move(callbacks_info), idb_runner_)); |
idb_runner_->PostTask( |
FROM_HERE, |
base::Bind(&IDBThreadHelper::GetAll, base::Unretained(helper_), |
@@ -262,8 +272,9 @@ void DatabaseImpl::Put( |
ChildProcessSecurityPolicyImpl* policy = |
ChildProcessSecurityPolicyImpl::GetInstance(); |
- scoped_refptr<IndexedDBCallbacks> callbacks(new IndexedDBCallbacks( |
- dispatcher_host_.get(), origin_, std::move(callbacks_info))); |
+ scoped_refptr<IndexedDBCallbacks> callbacks( |
+ new IndexedDBCallbacks(dispatcher_host_->AsWeakPtr(), origin_, |
+ std::move(callbacks_info), idb_runner_)); |
std::vector<std::unique_ptr<storage::BlobDataHandle>> handles( |
value->blob_or_file_info.size()); |
@@ -351,8 +362,9 @@ void DatabaseImpl::OpenCursor( |
bool key_only, |
blink::WebIDBTaskType task_type, |
::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info) { |
- scoped_refptr<IndexedDBCallbacks> callbacks(new IndexedDBCallbacks( |
- dispatcher_host_.get(), origin_, std::move(callbacks_info))); |
+ scoped_refptr<IndexedDBCallbacks> callbacks( |
+ new IndexedDBCallbacks(dispatcher_host_->AsWeakPtr(), origin_, |
+ std::move(callbacks_info), idb_runner_)); |
idb_runner_->PostTask( |
FROM_HERE, |
base::Bind(&IDBThreadHelper::OpenCursor, base::Unretained(helper_), |
@@ -366,8 +378,9 @@ void DatabaseImpl::Count( |
int64_t index_id, |
const IndexedDBKeyRange& key_range, |
::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info) { |
- scoped_refptr<IndexedDBCallbacks> callbacks(new IndexedDBCallbacks( |
- dispatcher_host_.get(), origin_, std::move(callbacks_info))); |
+ scoped_refptr<IndexedDBCallbacks> callbacks( |
+ new IndexedDBCallbacks(dispatcher_host_->AsWeakPtr(), origin_, |
+ std::move(callbacks_info), idb_runner_)); |
idb_runner_->PostTask( |
FROM_HERE, base::Bind(&IDBThreadHelper::Count, base::Unretained(helper_), |
transaction_id, object_store_id, index_id, |
@@ -379,8 +392,9 @@ void DatabaseImpl::DeleteRange( |
int64_t object_store_id, |
const IndexedDBKeyRange& key_range, |
::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info) { |
- scoped_refptr<IndexedDBCallbacks> callbacks(new IndexedDBCallbacks( |
- dispatcher_host_.get(), origin_, std::move(callbacks_info))); |
+ scoped_refptr<IndexedDBCallbacks> callbacks( |
+ new IndexedDBCallbacks(dispatcher_host_->AsWeakPtr(), origin_, |
+ std::move(callbacks_info), idb_runner_)); |
idb_runner_->PostTask( |
FROM_HERE, |
base::Bind(&IDBThreadHelper::DeleteRange, base::Unretained(helper_), |
@@ -392,8 +406,9 @@ void DatabaseImpl::Clear( |
int64_t transaction_id, |
int64_t object_store_id, |
::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info) { |
- scoped_refptr<IndexedDBCallbacks> callbacks(new IndexedDBCallbacks( |
- dispatcher_host_.get(), origin_, std::move(callbacks_info))); |
+ scoped_refptr<IndexedDBCallbacks> callbacks( |
+ new IndexedDBCallbacks(dispatcher_host_->AsWeakPtr(), origin_, |
+ std::move(callbacks_info), idb_runner_)); |
idb_runner_->PostTask( |
FROM_HERE, |
base::Bind(&IDBThreadHelper::Clear, base::Unretained(helper_), |
@@ -453,18 +468,20 @@ void DatabaseImpl::AckReceivedBlobs(const std::vector<std::string>& uuids) { |
DatabaseImpl::IDBThreadHelper::IDBThreadHelper( |
std::unique_ptr<IndexedDBConnection> connection, |
const url::Origin& origin, |
- scoped_refptr<IndexedDBDispatcherHost> dispatcher_host) |
- : dispatcher_host_(std::move(dispatcher_host)), |
+ scoped_refptr<IndexedDBContextImpl> indexed_db_context) |
+ : indexed_db_context_(indexed_db_context), |
connection_(std::move(connection)), |
origin_(origin), |
- weak_factory_(this) { |
- dispatcher_host_->context()->ConnectionOpened(origin_, connection.get()); |
-} |
+ weak_factory_(this) {} |
DatabaseImpl::IDBThreadHelper::~IDBThreadHelper() { |
if (connection_->IsConnected()) |
connection_->Close(); |
- dispatcher_host_->context()->ConnectionClosed(origin_, connection_.get()); |
+ indexed_db_context_->ConnectionClosed(origin_, connection_.get()); |
+} |
+ |
+void DatabaseImpl::IDBThreadHelper::ConnectionOpened() { |
+ indexed_db_context_->ConnectionOpened(origin_, connection_.get()); |
} |
void DatabaseImpl::IDBThreadHelper::CreateObjectStore( |
@@ -841,8 +858,8 @@ void DatabaseImpl::IDBThreadHelper::Commit(int64_t transaction_id) { |
return; |
} |
- dispatcher_host_->context()->quota_manager_proxy()->GetUsageAndQuota( |
- dispatcher_host_->context()->TaskRunner(), origin_.GetURL(), |
+ indexed_db_context_->quota_manager_proxy()->GetUsageAndQuota( |
+ indexed_db_context_->TaskRunner(), origin_.GetURL(), |
storage::kStorageTypeTemporary, |
base::Bind(&IDBThreadHelper::OnGotUsageAndQuotaForCommit, |
weak_factory_.GetWeakPtr(), transaction_id)); |