IDBObserver: Lifetime Management: Adding Observer

third_party/WebKit/Source/modules/indexeddb/IDBObserver.cpp

 // Copyright 2016 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 "modules/indexeddb/IDBObserver.h"
 
 #include "bindings/core/v8/ExceptionState.h"
 #include "bindings/modules/v8/ToV8ForModules.h"
 #include "bindings/modules/v8/V8BindingForModules.h"
+#include "core/dom/ExceptionCode.h"
+#include "modules/indexeddb/IDBDatabase.h"
 #include "modules/indexeddb/IDBObserverCallback.h"
 #include "modules/indexeddb/IDBObserverInit.h"
+#include "modules/indexeddb/IDBTransaction.h"
+#include "modules/indexeddb/WebIDBObserverImpl.h"
 
 namespace blink {
 
 IDBObserver* IDBObserver::create(IDBObserverCallback& callback, const IDBObserverInit& options)
 {
     return new IDBObserver(callback, options);
 }
 
 IDBObserver::IDBObserver(IDBObserverCallback& callback, const IDBObserverInit& options)
     : m_callback(&callback)
     , m_transaction(options.transaction())
     , m_values(options.values())
     , m_noRecords(options.noRecords())
 {
 }
 
 void IDBObserver::observe(IDBDatabase* database, IDBTransaction* transaction, ExceptionState& exceptionState)
 {
-    // TODO(palakj): Finish implementation.
+    if (transaction->isFinished() || transaction->isFinishing()) {
+        exceptionState.throwDOMException(TransactionInactiveError, IDBDatabase::transactionFinishedErrorMessage);
+        return;
+    }
+    if (!transaction->isActive()) {
+        exceptionState.throwDOMException(TransactionInactiveError, IDBDatabase::transactionInactiveErrorMessage);
+        return;
+    }
+    if (!database->backend()) {
+        exceptionState.throwDOMException(InvalidStateError, IDBDatabase::databaseClosedErrorMessage);
+        return;
+    }
+    int32_t observerId = database->backend()->addObserver(WebIDBObserverImpl::create(this).release(), transaction->id());
+    m_observerIds.insert(observerId);
+}
+
+void IDBObserver::unobserve(IDBDatabase* database, ExceptionState& exceptionState)
+{
+    if (!database->backend()) {
+        exceptionState.throwDOMException(InvalidStateError, IDBDatabase::databaseClosedErrorMessage);
+        return;
+    }
+    std::set<int32_t>::iterator it = m_observerIds.begin();
+    std::vector<int32_t> observerIdsToRemove;
+    while (it != m_observerIds.end()) {
+        if (database->backend()->containsObserverId(*it)) {
+            observerIdsToRemove.push_back(*it);
+            it = m_observerIds.erase(it);
+        } else {
+            it++;
+        }
+    }
+    if (!observerIdsToRemove.empty())
+        database->backend()->removeObservers(observerIdsToRemove);
+}
+
+void IDBObserver::removeObserver(int32_t id)
+{
+    m_observerIds.erase(id);
 }
 
 DEFINE_TRACE(IDBObserver)
 {
     visitor->trace(m_callback);
 }
 
 } // namespace blink