[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 "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_transaction.h"
 
 namespace content {
 
 namespace {
 
 static int32_t next_id;
 
 }  // namespace
 
 IndexedDBConnection::IndexedDBConnection(
+    int child_process_id,
     scoped_refptr<IndexedDBDatabase> database,
     scoped_refptr<IndexedDBDatabaseCallbacks> callbacks)
     : id_(next_id++),
+      child_process_id_(child_process_id),
       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 =
+      base::MakeUnique<IndexedDBTransaction>(id, this, scope, mode,
+                                             backing_store_transaction);
+  IndexedDBTransaction* transaction_ptr = transaction.get();
+  transactions_[id] = std::move(transaction);
+  return transaction_ptr;
+}
+
+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(transactions_.find(transaction->id()) == transactions_.end());

[jsbell, 2016/11/04 17:48:19]

base::ContainsKey

[dmurph, 2016/11/04 22:52:24]

Done.

+  IndexedDBTransaction* transaction_ptr = transaction.get();
+  transactions_[transaction->id()] = std::move(transaction);
+  return transaction_ptr;
+}
+
+void IndexedDBConnection::EraseTransaction(int64_t id) {
+  transactions_.erase(id);
+}
+
+void IndexedDBConnection::AbortAllTransactions(
+    const IndexedDBDatabaseError& error) {
+  std::unordered_map<int64_t, std::unique_ptr<IndexedDBTransaction>> temp_map;

[cmumford, 2016/11/04 23:33:12]

Couldn't you also do this?

  const auto temp_map = std::move(transactions_);

[dmurph, 2016/11/07 20:05:22]

I could, but that leaves transactions_ in a technically undefined state. It
probably will behave correctly but I think we want to avoid using anything other
than a smart pointer after a std::move, and I don't want to introduce that
constraint into this class -- "don't use 'transactions_' after calling this
method" - It's cleaner to me to just have it be empty w/ a standard method.

+  std::swap(temp_map, transactions_);
+  for (const auto& pair : temp_map) {
+    pair.second->Abort(error);
+  }
 }
 
 }  // namespace content