[IndexedDB] Refactoring to remove ref ptrs and host transaction ids.

content/browser/indexed_db/indexed_db_connection.cc

 // Copyright 2013 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 "content/browser/indexed_db/indexed_db_connection.h"
 
 #include "base/logging.h"
+#include "base/stl_util.h"
+#include "content/browser/indexed_db/indexed_db_class_factory.h"
+#include "content/browser/indexed_db/indexed_db_database_callbacks.h"
+#include "content/browser/indexed_db/indexed_db_observer.h"
+#include "content/browser/indexed_db/indexed_db_tracing.h"
+#include "content/browser/indexed_db/indexed_db_transaction.h"
 
 namespace content {
 
 namespace {
 
 static int32_t next_id;
 
 }  // namespace
 
 IndexedDBConnection::IndexedDBConnection(
+    int child_process_id,
+    const url::Origin& origin,
     scoped_refptr<IndexedDBDatabase> database,
     scoped_refptr<IndexedDBDatabaseCallbacks> callbacks)
     : id_(next_id++),
+      child_process_id_(child_process_id),
+      origin_(origin),
       database_(database),
       callbacks_(callbacks),
       weak_factory_(this) {}
 
 IndexedDBConnection::~IndexedDBConnection() {}
 
 void IndexedDBConnection::Close() {
   if (!callbacks_.get())
     return;
   base::WeakPtr<IndexedDBConnection> this_obj = weak_factory_.GetWeakPtr();
@@ skipping 47 matching lines @@
     const auto& it = std::find_if(
         active_observers_.begin(), active_observers_.end(),
         [&id_to_remove](const std::unique_ptr<IndexedDBObserver>& o) {
           return o->id() == id_to_remove;
         });
     if (it != active_observers_.end())
       active_observers_.erase(it);
     else
       pending_observer_ids.push_back(id_to_remove);
   }
-  if (!pending_observer_ids.empty())
-    database_->RemovePendingObservers(this, pending_observer_ids);
+  if (pending_observer_ids.empty())
+    return;
+
+  for (const auto& it : transactions_) {
+    it.second->RemovePendingObservers(pending_observer_ids);
+  }
+}
+
+IndexedDBTransaction* IndexedDBConnection::CreateTransaction(
+    int64_t id,
+    const std::set<int64_t>& scope,
+    blink::WebIDBTransactionMode mode,
+    IndexedDBBackingStore::Transaction* backing_store_transaction) {
+  DCHECK_EQ(GetTransaction(id), nullptr);
+  std::unique_ptr<IndexedDBTransaction> transaction =
+      IndexedDBClassFactory::Get()->CreateIndexedDBTransaction(
+          id, this, scope, mode, backing_store_transaction);
+  IndexedDBTransaction* transaction_ptr = transaction.get();
+  transactions_[id] = std::move(transaction);
+  return transaction_ptr;
+}
+
+void IndexedDBConnection::AbortTransaction(IndexedDBTransaction* transaction) {
+  IDB_TRACE1("IndexedDBDatabase::Abort", "txn.id", transaction->id());
+  transaction->Abort();
+}
+
+void IndexedDBConnection::AbortTransaction(
+    IndexedDBTransaction* transaction,
+    const IndexedDBDatabaseError& error) {
+  IDB_TRACE1("IndexedDBDatabase::Abort(error)", "txn.id", transaction->id());
+  transaction->Abort(error);
+}
+
+void IndexedDBConnection::AbortAllTransactions(
+    const IndexedDBDatabaseError& error) {
+  std::unordered_map<int64_t, std::unique_ptr<IndexedDBTransaction>> temp_map;
+  std::swap(temp_map, transactions_);
+  for (const auto& pair : temp_map) {
+    IDB_TRACE1("IndexedDBDatabase::Abort(error)", "txn.id", pair.second->id());
+    pair.second->Abort(error);
+  }
+}
+
+IndexedDBTransaction* IndexedDBConnection::GetTransaction(int64_t id) {
+  auto it = transactions_.find(id);
+  if (it == transactions_.end())
+    return nullptr;
+  return it->second.get();
+}
+
+IndexedDBTransaction* IndexedDBConnection::StoreTransactionForTesting(
+    std::unique_ptr<IndexedDBTransaction> transaction) {
+  DCHECK(!base::ContainsKey(transactions_, transaction->id()));
+  IndexedDBTransaction* transaction_ptr = transaction.get();
+  transactions_[transaction->id()] = std::move(transaction);
+  return transaction_ptr;
+}
+
+void IndexedDBConnection::EraseTransaction(int64_t id) {
+  transactions_.erase(id);
 }
 
 }  // namespace content