Replacing Indexed DB Chromium IPC with Mojo

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

 /*
  * Copyright (C) 2012 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  * 1.  Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
  * 2.  Redistributions in binary form must reproduce the above copyright
@@ skipping 12 matching lines @@
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "modules/indexeddb/IDBOpenDBRequest.h"
 
 #include "bindings/core/v8/Nullable.h"
 #include "core/dom/DOMException.h"
 #include "core/dom/ExceptionCode.h"
 #include "core/dom/ExecutionContext.h"
 #include "modules/indexeddb/IDBDatabase.h"
-#include "modules/indexeddb/IDBDatabaseCallbacks.h"
+#include "modules/indexeddb/IDBDatabaseObserverImpl.h"
+#include "modules/indexeddb/IDBDatabaseProxy.h"
+#include "modules/indexeddb/IDBMojoUtil.h"
 #include "modules/indexeddb/IDBTracing.h"
 #include "modules/indexeddb/IDBVersionChangeEvent.h"
 #include <memory>
 
-using blink::WebIDBDatabase;
+using blink::IDBDatabaseProxy;
 
 namespace blink {
 
-IDBOpenDBRequest* IDBOpenDBRequest::create(ScriptState* scriptState, IDBDatabaseCallbacks* callbacks, int64_t transactionId, int64_t version)
+IDBOpenDBRequest* IDBOpenDBRequest::create(ScriptState* scriptState, int64_t transactionId, int64_t version, indexed_db::mojom::blink::OpenRequestObserverRequest request)
 {
-    IDBOpenDBRequest* request = new IDBOpenDBRequest(scriptState, callbacks, transactionId, version);
-    request->suspendIfNeeded();
-    return request;
+    IDBOpenDBRequest* idbRequest = new IDBOpenDBRequest(scriptState, transactionId, version, std::move(request));
+    idbRequest->suspendIfNeeded();
+    return idbRequest;
 }
 
-IDBOpenDBRequest::IDBOpenDBRequest(ScriptState* scriptState, IDBDatabaseCallbacks* callbacks, int64_t transactionId, int64_t version)
+IDBOpenDBRequest::IDBOpenDBRequest(ScriptState* scriptState, int64_t transactionId, int64_t version, indexed_db::mojom::blink::OpenRequestObserverRequest request)
     : IDBRequest(scriptState, IDBAny::createNull(), nullptr)
-    , m_databaseCallbacks(callbacks)
     , m_transactionId(transactionId)
     , m_version(version)
+    , m_binding(this, std::move(request))
 {
     ASSERT(!resultAsAny());
 }
 
 IDBOpenDBRequest::~IDBOpenDBRequest()
 {
 }
 
 DEFINE_TRACE(IDBOpenDBRequest)
 {
-    visitor->trace(m_databaseCallbacks);
     IDBRequest::trace(visitor);
 }
 
 const AtomicString& IDBOpenDBRequest::interfaceName() const
 {
     return EventTargetNames::IDBOpenDBRequest;
 }
 
 void IDBOpenDBRequest::onBlocked(int64_t oldVersion)
 {
     IDB_TRACE("IDBOpenDBRequest::onBlocked()");
     if (!shouldEnqueueEvent())
         return;
     Nullable<unsigned long long> newVersionNullable = (m_version == IDBDatabaseMetadata::DefaultVersion) ? Nullable<unsigned long long>() : Nullable<unsigned long long>(m_version);
     enqueueEvent(IDBVersionChangeEvent::create(EventTypeNames::blocked, oldVersion, newVersionNullable));
 }
 
-void IDBOpenDBRequest::onUpgradeNeeded(int64_t oldVersion, std::unique_ptr<WebIDBDatabase> backend, const IDBDatabaseMetadata& metadata, WebIDBDataLoss dataLoss, String dataLossMessage)
+IDBDatabase* IDBOpenDBRequest::getDatabase() const
+{
+    if (resultAsAny())
+        return resultAsAny()->idbDatabase();
+    return nullptr;
+}
+
+void IDBOpenDBRequest::onUpgradeNeeded(int64_t oldVersion, std::unique_ptr<IDBDatabaseProxy> backend, const IDBDatabaseMetadata& metadata, WebIDBDataLoss dataLoss, String dataLossMessage)
 {
     IDB_TRACE("IDBOpenDBRequest::onUpgradeNeeded()");
     if (m_contextStopped || !getExecutionContext()) {
-        std::unique_ptr<WebIDBDatabase> db = std::move(backend);
-        db->abort(m_transactionId);
-        db->close();
+        std::unique_ptr<IDBDatabaseProxy> db = std::move(backend);
+        db->Abort(m_transactionId);
+        db->Close();
         return;
     }
     if (!shouldEnqueueEvent())
         return;
 
-    ASSERT(m_databaseCallbacks);
-
-    IDBDatabase* idbDatabase = IDBDatabase::create(getExecutionContext(), std::move(backend), m_databaseCallbacks.release());
+    IDBDatabase* idbDatabase = IDBDatabase::create(getExecutionContext(), std::move(backend));
     idbDatabase->setMetadata(metadata);
 
     if (oldVersion == IDBDatabaseMetadata::NoVersion) {
         // This database hasn't had a version before.
         oldVersion = IDBDatabaseMetadata::DefaultVersion;
     }
     IDBDatabaseMetadata oldMetadata(metadata);
     oldMetadata.version = oldVersion;
 
     m_transaction = IDBTransaction::create(getScriptState(), m_transactionId, idbDatabase, this, oldMetadata);
     setResult(IDBAny::create(idbDatabase));
 
     if (m_version == IDBDatabaseMetadata::NoVersion)
         m_version = 1;
     enqueueEvent(IDBVersionChangeEvent::create(EventTypeNames::upgradeneeded, oldVersion, m_version, dataLoss, dataLossMessage));
 }
 
-void IDBOpenDBRequest::onSuccess(std::unique_ptr<WebIDBDatabase> backend, const IDBDatabaseMetadata& metadata)
+void IDBOpenDBRequest::onSuccess(std::unique_ptr<IDBDatabaseProxy> backend, const IDBDatabaseMetadata& metadata)
 {
     IDB_TRACE("IDBOpenDBRequest::onSuccess()");
     if (m_contextStopped || !getExecutionContext()) {
-        std::unique_ptr<WebIDBDatabase> db = std::move(backend);
+        std::unique_ptr<IDBDatabaseProxy> db = std::move(backend);
         if (db)
-            db->close();
+            db->Close();
         return;
     }
     if (!shouldEnqueueEvent())
         return;
 
     IDBDatabase* idbDatabase = nullptr;
     if (resultAsAny()) {
         // Previous onUpgradeNeeded call delivered the backend.
         ASSERT(!backend.get());
         idbDatabase = resultAsAny()->idbDatabase();
         ASSERT(idbDatabase);
-        ASSERT(!m_databaseCallbacks);
     } else {
         ASSERT(backend.get());
-        ASSERT(m_databaseCallbacks);
-        idbDatabase = IDBDatabase::create(getExecutionContext(), std::move(backend), m_databaseCallbacks.release());
+        idbDatabase = IDBDatabase::create(getExecutionContext(), std::move(backend));
         setResult(IDBAny::create(idbDatabase));
     }
     idbDatabase->setMetadata(metadata);
     enqueueEvent(Event::create(EventTypeNames::success));
 }
 
 void IDBOpenDBRequest::onSuccess(int64_t oldVersion)
 {
     IDB_TRACE("IDBOpenDBRequest::onSuccess()");
     if (!shouldEnqueueEvent())
@@ skipping 23 matching lines @@
     if (event->type() == EventTypeNames::success && resultAsAny()->getType() == IDBAny::IDBDatabaseType && resultAsAny()->idbDatabase()->isClosePending()) {
         dequeueEvent(event);
         setResult(nullptr);
         onError(DOMException::create(AbortError, "The connection was closed."));
         return DispatchEventResult::CanceledBeforeDispatch;
     }
 
     return IDBRequest::dispatchEventInternal(event);
 }
 
+void IDBOpenDBRequest::onOpenResult(indexed_db::mojom::blink::OpenResultPtr result)
+{
+    if (result->is_error()) {
+        // TODO(cmumford): Finish this.
+        NOTREACHED();
+    } else if (result->is_data()) {
+        // Fix this.
+        bool updateWasCalled = true;
+        // TODO(cmumford): Get the metadata.
+        IDBDatabaseMetadata metadata;
+        convertMetadata(&metadata, result->get_data()->metadata);
+        if (updateWasCalled) {
+            onSuccess(nullptr, metadata);
+        } else {
+            std::unique_ptr<IDBDatabaseProxy> proxy(IDBDatabaseProxy::create(std::move(result->get_data()->database)));
+            onSuccess(std::move(proxy), metadata);
+        }
+    } else {
+        // TODO(cmumford): Finish this.
+        NOTREACHED();
+    }
+}
+
+void IDBOpenDBRequest::OnBlocked(int64_t oldVersion)
+{
+    // TODO(cmumford): Combine this and onBlocked().
+    onBlocked(oldVersion);
+}
+
+void IDBOpenDBRequest::OnUpgradeNeeded(int64_t oldVersion, indexed_db::mojom::blink::DatabasePtr database, indexed_db::mojom::blink::DatabaseMetadataPtr metadata, indexed_db::mojom::blink::DataLossInfoPtr dataLossInfo)
+{
+    // TODO(cmumford): Combine this and onUpgradeNeeded().
+    std::unique_ptr<IDBDatabaseProxy> openRequestProxy(IDBDatabaseProxy::create(std::move(database)));
+    onUpgradeNeeded(oldVersion, std::move(openRequestProxy), IDBDatabaseMetadata(std::move(metadata)), dataLossInfo->status, dataLossInfo->message);
+}
+
 } // namespace blink