Persist sessionStorage on disk.

webkit/dom_storage/dom_storage_namespace.cc

 // Copyright (c) 2012 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 "webkit/dom_storage/dom_storage_namespace.h"
 
 #include "base/basictypes.h"
 #include "base/logging.h"
 #include "webkit/dom_storage/dom_storage_area.h"
 #include "webkit/dom_storage/dom_storage_task_runner.h"
 #include "webkit/dom_storage/dom_storage_types.h"
+#include "webkit/dom_storage/session_storage_database.h"
 
 namespace dom_storage {
 
 DomStorageNamespace::DomStorageNamespace(
     const FilePath& directory,
     DomStorageTaskRunner* task_runner)
     : namespace_id_(kLocalStorageNamespaceId),
       directory_(directory),
       task_runner_(task_runner) {
 }
 
 DomStorageNamespace::DomStorageNamespace(
     int64 namespace_id,
     DomStorageTaskRunner* task_runner)
     : namespace_id_(namespace_id),
       task_runner_(task_runner) {
   DCHECK_NE(kLocalStorageNamespaceId, namespace_id);
 }
 
+DomStorageNamespace::DomStorageNamespace(
+    int64 namespace_id,
+    SessionStorageDatabase* session_storage_database,
+    DomStorageTaskRunner* task_runner)
+    : namespace_id_(namespace_id),
+      task_runner_(task_runner),
+      session_storage_database_(session_storage_database) {
+  DCHECK_NE(kLocalStorageNamespaceId, namespace_id);
+}
+
 DomStorageNamespace::~DomStorageNamespace() {
+  if (session_storage_database_.get())
+    session_storage_database_->DeleteNamespace(namespace_id_);
 }
 
 DomStorageArea* DomStorageNamespace::OpenStorageArea(const GURL& origin) {
   if (AreaHolder* holder = GetAreaHolder(origin)) {
     ++(holder->open_count_);
     return holder->area_;
   }
-  DomStorageArea* area = new DomStorageArea(namespace_id_, origin,
-                                            directory_, task_runner_);
+  DomStorageArea* area;
+  if (!directory_.empty()) {
+    // Local storage backed on disk.
+    area = new DomStorageArea(namespace_id_, origin, directory_, task_runner_);
+  } else if (session_storage_database_.get()) {
+    // Session storage backed on disk.
+    area = new DomStorageArea(namespace_id_, origin, session_storage_database_,
+                              task_runner_);
+  } else {
+    // In-memory.
+    area = new DomStorageArea(namespace_id_, origin, task_runner_);
+  }
   areas_[origin] = AreaHolder(area, 1);
   return area;
 }
 
 void DomStorageNamespace::CloseStorageArea(DomStorageArea* area) {
   AreaHolder* holder = GetAreaHolder(area->origin());
   DCHECK(holder);
   DCHECK_EQ(holder->area_.get(), area);
   --(holder->open_count_);
   // TODO(michaeln): Clean up areas that aren't needed in memory anymore.
   // The in-process-webkit based impl didn't do this either, but would be nice.
 }
 
 DomStorageNamespace* DomStorageNamespace::Clone(int64 clone_namespace_id) {
   DCHECK_NE(kLocalStorageNamespaceId, namespace_id_);
   DCHECK_NE(kLocalStorageNamespaceId, clone_namespace_id);
-  DomStorageNamespace* clone =
-      new DomStorageNamespace(clone_namespace_id, task_runner_);
+  DomStorageNamespace* clone = new DomStorageNamespace(
+      clone_namespace_id, session_storage_database_, task_runner_);
   AreaMap::const_iterator it = areas_.begin();
   for (; it != areas_.end(); ++it) {
     DomStorageArea* area = it->second.area_->ShallowCopy(clone_namespace_id);
     clone->areas_[it->first] = AreaHolder(area, 0);
   }
   return clone;
 }
 
 void DomStorageNamespace::DeleteOrigin(const GURL& origin) {
   AreaHolder* holder = GetAreaHolder(origin);
   if (holder) {
     holder->area_->DeleteOrigin();
     return;
   }
   if (!directory_.empty()) {
     scoped_refptr<DomStorageArea> area =
         new DomStorageArea(namespace_id_, origin, directory_, task_runner_);
     area->DeleteOrigin();
+  } else if (session_storage_database_.get()) {
+    scoped_refptr<DomStorageArea> area =
+        new DomStorageArea(namespace_id_, origin,
+                           session_storage_database_.get(), task_runner_);
+    area->DeleteOrigin();
   }
 }
 
 void DomStorageNamespace::PurgeMemory() {
   if (directory_.empty())
     return;  // We can't purge w/o backing on disk.
   AreaMap::iterator it = areas_.begin();
   while (it != areas_.end()) {
     // Leave it alone if changes are pending
     if (it->second.area_->HasUncommittedChanges()) {
@@ skipping 38 matching lines @@
 
 DomStorageNamespace::AreaHolder::AreaHolder(
     DomStorageArea* area, int count)
     : area_(area), open_count_(count) {
 }
 
 DomStorageNamespace::AreaHolder::~AreaHolder() {
 }
 
 }  // namespace dom_storage