Index: webkit/dom_storage/session_storage_database.cc |
diff --git a/webkit/dom_storage/session_storage_database.cc b/webkit/dom_storage/session_storage_database.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..e66d477442b9067d7bfb87b76296417023f49a48 |
--- /dev/null |
+++ b/webkit/dom_storage/session_storage_database.cc |
@@ -0,0 +1,658 @@ |
+// 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/session_storage_database.h" |
+ |
+#include "base/file_util.h" |
+#include "base/logging.h" |
+#include "base/stringprintf.h" |
+#include "base/string_number_conversions.h" |
+#include "base/utf_string_conversions.h" |
+#include "googleurl/src/gurl.h" |
+#include "third_party/leveldatabase/src/include/leveldb/db.h" |
+#include "third_party/leveldatabase/src/include/leveldb/iterator.h" |
+#include "third_party/leveldatabase/src/include/leveldb/status.h" |
+#include "third_party/leveldatabase/src/include/leveldb/options.h" |
+#include "third_party/leveldatabase/src/include/leveldb/write_batch.h" |
+ |
+// Layout of the database: |
+// | key | value | |
+// ----------------------------------------------------------------------- |
+// | map-1 | 2 (refcount, start of map-1-* keys)| |
+// | map-1-a | b (a = b in map 1) | |
+// | ... | | |
+// | namespace- | dummy (start of namespace-* keys) | |
+// | namespace-1 (1 = namespace id) | dummy (start of namespace-1-* keys)| |
+// | namespace-1-origin1 | 1 (mapid) | |
+// | namespace-1-origin2 | 2 | |
+// | namespace-2 | dummy | |
+// | namespace-2-origin1 | 1 (shallow copy) | |
+// | namespace-2-origin2 | 2 (shallow copy) | |
+// | namespace-3 | dummy | |
+// | namespace-3-origin1 | 3 (deep copy) | |
+// | namespace-3-origin2 | 2 (shallow copy) | |
+// | next-namespace-id | 4 | |
+// | next-map-id | 4 | |
+ |
+namespace dom_storage { |
+ |
+SessionStorageDatabase::SessionStorageDatabase(const FilePath& file_path) |
+ : file_path_(file_path), |
+ db_error_(false), |
+ is_inconsistent_(false), |
+ namespace_offset_(0) { } |
+ |
+SessionStorageDatabase::~SessionStorageDatabase() { |
+} |
+ |
+void SessionStorageDatabase::ReadAreaValues(int64 namespace_id, |
+ const GURL& origin, |
+ ValuesMap* result) { |
+ // We don't create a database if it doesn't exist. In that case, there is |
+ // nothing to be added to the result. |
+ if (!LazyOpen(false)) |
+ return; |
+ std::string map_id; |
+ bool exists; |
+ if (!GetMapForArea(namespace_id, origin, &exists, &map_id)) |
+ return; |
+ if (exists) |
+ ReadMap(map_id, result, false); |
+} |
+ |
+bool SessionStorageDatabase::CommitAreaChanges(int64 namespace_id, |
+ const GURL& origin, |
+ bool clear_all_first, |
+ const ValuesMap& changes) { |
+ // Even if |changes| is empty, we need to write the appropriate placeholders |
+ // in the database, so that it can be later shallow-copied succssfully. |
+ if (!LazyOpen(true)) |
+ return false; |
+ |
+ leveldb::WriteBatch batch; |
+ // Ensure that the keys "namespace-" "namespace-N" (see the schema above) |
+ // exist. |
+ const bool kOkIfExists = true; |
+ if (!CreateNamespace(namespace_id, kOkIfExists, &batch)) |
+ return false; |
+ |
+ std::string map_id; |
+ bool exists; |
+ if (!GetMapForArea(namespace_id, origin, &exists, &map_id)) |
+ return false; |
+ if (exists) { |
+ // We shouldn't write data into a shallow copy. If this is a shallow copy, |
+ // it's a caller error (not an inconsistency in the database). |
+ int64 ref_count; |
+ if (!GetMapRefCount(map_id, &ref_count)) |
+ return false; |
+ if (!CallerErrorCheck(ref_count == 1)) |
+ return false; |
+ |
+ if (clear_all_first) { |
+ if (!ClearMap(map_id, &batch)) |
+ return false; |
+ } |
+ } else { |
+ // Map doesn't exist, create it now if needed. |
+ if (!changes.empty()) { |
+ if (!CreateMapForArea(namespace_id, origin, &map_id, &batch)) |
+ return false; |
+ } |
+ } |
+ |
+ WriteValuesToMap(map_id, changes, &batch); |
+ |
+ leveldb::Status s = db_->Write(leveldb::WriteOptions(), &batch); |
+ return DatabaseErrorCheck(s.ok()); |
+} |
+ |
+bool SessionStorageDatabase::CloneNamespace(int64 namespace_id, |
+ int64 new_namespace_id) { |
+ // Go through all origins in the namespace |namespace_id|, create placeholders |
+ // for them in |new_namespace_id|, and associate them with the existing maps. |
+ |
+ // Example, data before shallow copy: |
+ // | map-1 | 1 (refcount) | |
+ // | map-1-a | b | |
+ // | namespace-1 (1 = namespace id) | dummy | |
+ // | namespace-1-origin1 | 1 (mapid) | |
+ |
+ // Example, data after shallow copy: |
+ // | map-1 | 2 (inc. refcount) | |
+ // | map-1-a | b | |
+ // | namespace-1 (1 = namespace id) | dummy | |
+ // | namespace-1-origin1 | 1 (mapid) | |
+ // | namespace-2 | dummy | |
+ // | namespace-2-origin1 | 1 (mapid) << references the same map |
+ |
+ if (!LazyOpen(true)) |
+ return false; |
+ |
+ leveldb::WriteBatch batch; |
+ const bool kOkIfExists = false; |
+ if (!CreateNamespace(new_namespace_id, kOkIfExists, &batch)) |
+ return false; |
+ |
+ std::map<std::string, std::string> areas; |
+ if (!GetAreasInNamespace(namespace_id, &areas)) |
+ return false; |
+ |
+ for (std::map<std::string, std::string>::const_iterator it = areas.begin(); |
+ it != areas.end(); ++it) { |
+ const std::string& origin = it->first; |
+ const std::string& map_id = it->second; |
+ if (!IncreaseMapRefCount(map_id, &batch)) |
+ return false; |
+ AddAreaToNamespace(new_namespace_id, origin, map_id, &batch); |
+ } |
+ leveldb::Status s = db_->Write(leveldb::WriteOptions(), &batch); |
+ return DatabaseErrorCheck(s.ok()); |
+} |
+ |
+bool SessionStorageDatabase::DeepCopyArea(int64 namespace_id, |
+ const GURL& origin) { |
+ // Example, data before deep copy: |
+ // | namespace-1 (1 = namespace id) | dummy | |
+ // | namespace-1-origin1 | 1 (mapid) | |
+ // | namespace-2 | dummy | |
+ // | namespace-2-origin1 | 1 (mapid) << references the same map |
+ // | map-1 | 2 (refcount) | |
+ // | map-1-a | b | |
+ |
+ // Example, data after deep copy copy: |
+ // | namespace-1 (1 = namespace id) | dummy | |
+ // | namespace-1-origin1 | 1 (mapid) | |
+ // | namespace-2 | dummy | |
+ // | namespace-2-origin1 | 2 (mapid) << references the new map |
+ // | map-1 | 1 (dec. refcount) | |
+ // | map-1-a | b | |
+ // | map-2 | 1 (refcount) | |
+ // | map-2-a | b | |
+ |
+ if (!LazyOpen(true)) |
+ return false; |
+ |
+ std::string old_map_id; |
+ bool exists; |
+ if (!GetMapForArea(namespace_id, origin, &exists, &old_map_id)) |
+ return false; |
+ |
+ // If the area doesn't exist, or if it's not a shallow copy, it's a caller |
+ // error. |
+ if (!CallerErrorCheck(exists)) |
+ return false; |
+ int64 ref_count; |
+ if (!GetMapRefCount(old_map_id, &ref_count)) |
+ return false; |
+ if (!CallerErrorCheck(ref_count > 1)) |
+ return false; |
+ |
+ leveldb::WriteBatch batch; |
+ std::string new_map_id; |
+ if (!CreateMapForArea(namespace_id, origin, &new_map_id, &batch)) |
+ return false; |
+ |
+ // Copy the values in the map. |
+ ValuesMap values; |
+ if (!ReadMap(old_map_id, &values, false)) |
+ return false; |
+ WriteValuesToMap(new_map_id, values, &batch); |
+ |
+ if (!DecreaseMapRefCount(old_map_id, 1, &batch)) |
+ return false; |
+ |
+ leveldb::Status s = db_->Write(leveldb::WriteOptions(), &batch); |
+ return DatabaseErrorCheck(s.ok()); |
+} |
+ |
+bool SessionStorageDatabase::DeleteArea(int64 namespace_id, |
+ const GURL& origin) { |
+ if (!LazyOpen(false)) { |
+ // No need to create the database if it doesn't exist. |
+ return true; |
+ } |
+ leveldb::WriteBatch batch; |
+ if (!DeleteArea(namespace_id, origin.spec(), &batch)) |
+ return false; |
+ leveldb::Status s = db_->Write(leveldb::WriteOptions(), &batch); |
+ return DatabaseErrorCheck(s.ok()); |
+} |
+ |
+bool SessionStorageDatabase::DeleteNamespace(int64 namespace_id) { |
+ if (!LazyOpen(false)) { |
+ // No need to create the database if it doesn't exist. |
+ return true; |
+ } |
+ // Itereate through the areas in the namespace. |
+ leveldb::WriteBatch batch; |
+ std::map<std::string, std::string> areas; |
+ if (!GetAreasInNamespace(namespace_id, &areas)) |
+ return false; |
+ for (std::map<std::string, std::string>::const_iterator it = areas.begin(); |
+ it != areas.end(); ++it) { |
+ const std::string& origin = it->first; |
+ if (!DeleteArea(namespace_id, origin, &batch)) |
+ return false; |
+ } |
+ batch.Delete(NamespaceStartKey(namespace_id, namespace_offset_)); |
+ leveldb::Status s = db_->Write(leveldb::WriteOptions(), &batch); |
+ return DatabaseErrorCheck(s.ok()); |
+} |
+ |
+bool SessionStorageDatabase::LazyOpen(bool create_if_needed) { |
+ base::AutoLock auto_lock(db_lock_); |
+ if (db_error_ || is_inconsistent_) { |
+ // Don't try to open a database that we know has failed already. |
+ return false; |
+ } |
+ if (IsOpen()) |
+ return true; |
+ |
+ if (!create_if_needed && |
+ (!file_util::PathExists(file_path_) || |
+ file_util::IsDirectoryEmpty(file_path_))) { |
+ // If the directory doesn't exist already and we haven't been asked to |
+ // create a file on disk, then we don't bother opening the database. This |
+ // means we wait until we absolutely need to put something onto disk before |
+ // we do so. |
+ return false; |
+ } |
+ |
+ leveldb::DB* db; |
+ leveldb::Status s = TryToOpen(file_path_, &db); |
+ if (!s.ok()) { |
+ LOG(WARNING) << "Failed to open leveldb in " << file_path_.value() |
+ << ", error: " << s.ToString(); |
+ DCHECK(db == NULL); |
+ |
+ // Clear the directory and try again. |
+ file_util::Delete(file_path_, true); |
+ s = TryToOpen(file_path_, &db); |
+ if (!s.ok()) { |
+ LOG(WARNING) << "Failed to open leveldb in " << file_path_.value() |
+ << ", error: " << s.ToString(); |
+ DCHECK(db == NULL); |
+ db_error_ = true; |
+ return false; |
+ } |
+ } |
+ db_.reset(db); |
+ |
+ return GetNextNamespaceId(&namespace_offset_); |
+} |
+ |
+leveldb::Status SessionStorageDatabase::TryToOpen(const FilePath& file_path, |
+ leveldb::DB** db) { |
+ leveldb::Options options; |
+ // The directory exists but a valid leveldb database might not exist inside it |
+ // (e.g., a subset of the needed files might be missing). Handle this |
+ // situation gracefully by creating the database now. |
+ options.create_if_missing = true; |
+#if defined(OS_WIN) |
+ return leveldb::DB::Open(options, WideToUTF8(file_path.value()), db); |
+#elif defined(OS_POSIX) |
+ return leveldb::DB::Open(options, file_path.value(), db); |
+#endif |
+} |
+ |
+bool SessionStorageDatabase::IsOpen() const { |
+ return db_.get() != NULL; |
+} |
+ |
+bool SessionStorageDatabase::CallerErrorCheck(bool ok) const { |
+ DCHECK(ok); |
+ return ok; |
+} |
+ |
+bool SessionStorageDatabase::ConsistencyCheck(bool ok) { |
+ if (ok) |
+ return true; |
+ base::AutoLock auto_lock(db_lock_); |
+ DCHECK(false); |
+ is_inconsistent_ = true; |
+ // We cannot recover the database during this run, e.g., the upper layer can |
+ // have a different understanding of the database state (shallow and deep |
+ // copies). |
+ // TODO(marja): Error handling. |
+ return false; |
+} |
+ |
+bool SessionStorageDatabase::DatabaseErrorCheck(bool ok) { |
+ if (ok) |
+ return true; |
+ base::AutoLock auto_lock(db_lock_); |
+ db_error_ = true; |
+ // TODO(marja): Error handling. |
+ return false; |
+} |
+ |
+bool SessionStorageDatabase::CreateNamespace(int64 namespace_id, |
+ bool ok_if_exists, |
+ leveldb::WriteBatch* batch) { |
+ std::string namespace_prefix = NamespacePrefix(); |
+ std::string dummy; |
+ leveldb::Status s = db_->Get(leveldb::ReadOptions(), namespace_prefix, |
+ &dummy); |
+ if (!DatabaseErrorCheck(s.ok() || s.IsNotFound())) |
+ return false; |
+ if (s.IsNotFound()) |
+ batch->Put(namespace_prefix, ""); |
+ |
+ std::string namespace_start_key = |
+ NamespaceStartKey(namespace_id, namespace_offset_); |
+ s = db_->Get(leveldb::ReadOptions(), namespace_start_key, &dummy); |
+ if (!DatabaseErrorCheck(s.ok() || s.IsNotFound())) |
+ return false; |
+ if (s.IsNotFound()) { |
+ batch->Put(namespace_start_key, ""); |
+ return UpdateNextNamespaceId(namespace_id, batch); |
+ } |
+ return CallerErrorCheck(ok_if_exists); |
+} |
+ |
+bool SessionStorageDatabase::GetNextNamespaceId(int64* next_namespace_id) { |
+ std::string next_namespace_id_string; |
+ leveldb::Status s = db_->Get(leveldb::ReadOptions(), NextNamespaceIdKey(), |
+ &next_namespace_id_string); |
+ if (!DatabaseErrorCheck(s.ok() || s.IsNotFound())) |
+ return false; |
+ if (s.IsNotFound()) { |
+ *next_namespace_id = 0; |
+ return true; |
+ } |
+ bool conversion_ok = |
+ base::StringToInt64(next_namespace_id_string, next_namespace_id); |
+ return ConsistencyCheck(conversion_ok); |
+} |
+ |
+bool SessionStorageDatabase::UpdateNextNamespaceId(int64 namespace_id, |
+ leveldb::WriteBatch* batch) { |
+ int64 next_namespace_id; |
+ if (!GetNextNamespaceId(&next_namespace_id)) |
+ return false; |
+ if (next_namespace_id < namespace_id + namespace_offset_ + 1) { |
+ next_namespace_id = namespace_id + namespace_offset_ + 1; |
+ batch->Put(NextNamespaceIdKey(), base::Int64ToString(next_namespace_id)); |
+ } |
+ return true; |
+} |
+ |
+bool SessionStorageDatabase::GetAreasInNamespace( |
+ int64 namespace_id, |
+ std::map<std::string, std::string>* areas) { |
+ return GetAreasInNamespace(NamespaceIdStr(namespace_id, namespace_offset_), |
+ areas); |
+} |
+ |
+bool SessionStorageDatabase::GetAreasInNamespace( |
+ const std::string& namespace_id_str, |
+ std::map<std::string, std::string>* areas) { |
+ std::string namespace_start_key = NamespaceStartKey(namespace_id_str); |
+ scoped_ptr<leveldb::Iterator> it(db_->NewIterator(leveldb::ReadOptions())); |
+ it->Seek(namespace_start_key); |
+ if (it->status().IsNotFound()) { |
+ // The namespace_start_key is not found when the namespace doesn't contain |
+ // any areas. We don't need to do anything. |
+ return true; |
+ } |
+ if (!DatabaseErrorCheck(it->status().ok())) |
+ return false; |
+ |
+ // Skip the dummy entry "namespace-<namespaceid>" and iterate the origins. |
+ for (it->Next(); it->Valid(); it->Next()) { |
+ std::string key = it->key().ToString(); |
+ if (key.find(namespace_start_key) != 0) { |
+ // Iterated past the origins for this namespace. |
+ break; |
+ } |
+ size_t second_dash = key.find('-', namespace_start_key.length()); |
+ if (!ConsistencyCheck(second_dash != std::string::npos)) |
+ return false; |
+ std::string origin = key.substr(second_dash + 1); |
+ std::string map_id = it->value().ToString(); |
+ (*areas)[origin] = map_id; |
+ } |
+ return true; |
+} |
+ |
+void SessionStorageDatabase::AddAreaToNamespace(int64 namespace_id, |
+ const std::string& origin, |
+ const std::string& map_id, |
+ leveldb::WriteBatch* batch) { |
+ std::string namespace_key = NamespaceKey( |
+ NamespaceIdStr(namespace_id, namespace_offset_), origin); |
+ batch->Put(namespace_key, map_id); |
+} |
+ |
+bool SessionStorageDatabase::DeleteArea(int64 namespace_id, |
+ const std::string& origin, |
+ leveldb::WriteBatch* batch) { |
+ return DeleteArea(NamespaceIdStr(namespace_id, namespace_offset_), |
+ origin, batch); |
+} |
+ |
+bool SessionStorageDatabase::DeleteArea(const std::string& namespace_id_str, |
+ const std::string& origin, |
+ leveldb::WriteBatch* batch) { |
+ std::string map_id; |
+ bool exists; |
+ if (!GetMapForArea(namespace_id_str, origin, &exists, &map_id)) |
+ return false; |
+ if (!exists) |
+ return true; // Nothing to delete. |
+ if (!DecreaseMapRefCount(map_id, 1, batch)) |
+ return false; |
+ std::string namespace_key = NamespaceKey(namespace_id_str, origin); |
+ batch->Delete(namespace_key); |
+ return true; |
+} |
+ |
+bool SessionStorageDatabase::GetMapForArea(int64 namespace_id, |
+ const GURL& origin, |
+ bool* exists, |
+ std::string* map_id) { |
+ return GetMapForArea( |
+ base::Int64ToString(namespace_id + namespace_offset_), |
+ origin.spec(), exists, map_id); |
+} |
+ |
+bool SessionStorageDatabase::GetMapForArea(const std::string& namespace_id_str, |
+ const std::string& origin, |
+ bool* exists, std::string* map_id) { |
+ std::string namespace_key = NamespaceKey(namespace_id_str, origin); |
+ leveldb::Status s = db_->Get(leveldb::ReadOptions(), namespace_key, map_id); |
+ if (s.IsNotFound()) { |
+ *exists = false; |
+ return true; |
+ } |
+ *exists = true; |
+ return DatabaseErrorCheck(s.ok()); |
+} |
+ |
+bool SessionStorageDatabase::CreateMapForArea(int64 namespace_id, |
+ const GURL& origin, |
+ std::string* map_id, |
+ leveldb::WriteBatch* batch) { |
+ std::string next_map_id_key = NextMapIdKey(); |
+ leveldb::Status s = db_->Get(leveldb::ReadOptions(), next_map_id_key, map_id); |
+ if (!DatabaseErrorCheck(s.ok() || s.IsNotFound())) |
+ return false; |
+ int64 next_map_id = 0; |
+ if (s.IsNotFound()) { |
+ *map_id = "0"; |
+ } else { |
+ bool conversion_ok = base::StringToInt64(*map_id, &next_map_id); |
+ if (!ConsistencyCheck(conversion_ok)) |
+ return false; |
+ } |
+ batch->Put(next_map_id_key, base::Int64ToString(++next_map_id)); |
+ std::string namespace_key = |
+ NamespaceKey(namespace_id, namespace_offset_, origin); |
+ batch->Put(namespace_key, *map_id); |
+ batch->Put(MapRefCountKey(*map_id), "1"); |
+ return true; |
+} |
+ |
+bool SessionStorageDatabase::ReadMap(const std::string& map_id, |
+ ValuesMap* result, |
+ bool only_keys) { |
+ scoped_ptr<leveldb::Iterator> it(db_->NewIterator(leveldb::ReadOptions())); |
+ std::string map_start_key = MapRefCountKey(map_id); |
+ it->Seek(map_start_key); |
+ // The map needs to exist, otherwise we have a stale map_id in the database. |
+ if (!ConsistencyCheck(!it->status().IsNotFound())) |
+ return false; |
+ if (!DatabaseErrorCheck(it->status().ok())) |
+ return false; |
+ const int kPrefixLength = std::string(MapPrefix()).length(); |
+ // Skip the dummy entry "map-<mapid>". |
+ for (it->Next(); it->Valid(); it->Next()) { |
+ // Key is of the form "map-<mapid>-<key>". |
+ std::string key = it->key().ToString(); |
+ size_t second_dash = key.find('-', kPrefixLength); |
+ if (second_dash == std::string::npos || |
+ key.substr(kPrefixLength, second_dash - kPrefixLength) != map_id) { |
+ // Iterated beyond the keys in this map. |
+ break; |
+ } |
+ string16 key16 = UTF8ToUTF16(key.substr(second_dash + 1)); |
+ if (only_keys) { |
+ (*result)[key16] = NullableString16(true); |
+ } else { |
+ // Convert the raw data stored in std::string (it->value()) to raw data |
+ // stored in string16. |
+ size_t len = it->value().size() / sizeof(char16); |
+ const char16* data_ptr = |
+ reinterpret_cast<const char16*>(it->value().data()); |
+ (*result)[key16] = NullableString16(string16(data_ptr, len), false); |
+ } |
+ } |
+ return true; |
+} |
+ |
+void SessionStorageDatabase::WriteValuesToMap(const std::string& map_id, |
+ const ValuesMap& values, |
+ leveldb::WriteBatch* batch) { |
+ for (ValuesMap::const_iterator it = values.begin(); it != values.end(); |
+ ++it) { |
+ NullableString16 value = it->second; |
+ std::string key = MapKey(map_id, UTF16ToUTF8(it->first)); |
+ if (value.is_null()) { |
+ batch->Delete(key); |
+ } else { |
+ // Convert the raw data stored in string16 to raw data stored in |
+ // std::string. |
+ const char* data = reinterpret_cast<const char*>(value.string().data()); |
+ size_t size = value.string().size() * 2; |
+ batch->Put(key, leveldb::Slice(data, size)); |
+ } |
+ } |
+} |
+ |
+bool SessionStorageDatabase::GetMapRefCount(const std::string& map_id, |
+ int64* ref_count) { |
+ std::string ref_count_string; |
+ leveldb::Status s = db_->Get(leveldb::ReadOptions(), |
+ MapRefCountKey(map_id), &ref_count_string); |
+ if (!ConsistencyCheck(s.ok())) |
+ return false; |
+ bool conversion_ok = base::StringToInt64(ref_count_string, ref_count); |
+ return ConsistencyCheck(conversion_ok); |
+} |
+ |
+bool SessionStorageDatabase::IncreaseMapRefCount(const std::string& map_id, |
+ leveldb::WriteBatch* batch) { |
+ // Increase the ref count for the map. |
+ int64 old_ref_count; |
+ if (!GetMapRefCount(map_id, &old_ref_count)) |
+ return false; |
+ batch->Put(MapRefCountKey(map_id), base::Int64ToString(++old_ref_count)); |
+ return true; |
+} |
+ |
+bool SessionStorageDatabase::DecreaseMapRefCount(const std::string& map_id, |
+ int decrease, |
+ leveldb::WriteBatch* batch) { |
+ // Decrease the ref count for the map. |
+ int64 ref_count; |
+ if (!GetMapRefCount(map_id, &ref_count)) |
+ return false; |
+ if (!ConsistencyCheck(decrease <= ref_count)) |
+ return false; |
+ ref_count -= decrease; |
+ if (ref_count > 0) { |
+ batch->Put(MapRefCountKey(map_id), base::Int64ToString(ref_count)); |
+ } else { |
+ // Clear all keys in the map. |
+ if (!ClearMap(map_id, batch)) |
+ return false; |
+ batch->Delete(MapRefCountKey(map_id)); |
+ } |
+ return true; |
+} |
+ |
+bool SessionStorageDatabase::ClearMap(const std::string& map_id, |
+ leveldb::WriteBatch* batch) { |
+ ValuesMap values; |
+ if (!ReadMap(map_id, &values, true)) |
+ return false; |
+ for (ValuesMap::const_iterator it = values.begin(); it != values.end(); ++it) |
+ batch->Delete(MapKey(map_id, UTF16ToUTF8(it->first))); |
+ return true; |
+} |
+ |
+std::string SessionStorageDatabase::NamespaceStartKey( |
+ const std::string& namespace_id_str) { |
+ return base::StringPrintf("namespace-%s", namespace_id_str.c_str()); |
+} |
+ |
+std::string SessionStorageDatabase::NamespaceStartKey(int64 namespace_id, |
+ int64 namespace_offset) { |
+ return NamespaceStartKey(NamespaceIdStr(namespace_id, namespace_offset)); |
+} |
+ |
+std::string SessionStorageDatabase::NamespaceKey( |
+ const std::string& namespace_id_str, const std::string& origin) { |
+ return base::StringPrintf("namespace-%s-%s", namespace_id_str.c_str(), |
+ origin.c_str()); |
+} |
+ |
+std::string SessionStorageDatabase::NamespaceKey( |
+ int64 namespace_id, int64 namespace_offset, const GURL& origin) { |
+ return NamespaceKey(NamespaceIdStr(namespace_id, namespace_offset), |
+ origin.spec()); |
+} |
+ |
+std::string SessionStorageDatabase::NamespaceIdStr(int64 namespace_id, |
+ int64 namespace_offset) { |
+ return base::Int64ToString(namespace_id + namespace_offset); |
+} |
+ |
+const char* SessionStorageDatabase::NamespacePrefix() { |
+ return "namespace-"; |
+} |
+ |
+std::string SessionStorageDatabase::MapRefCountKey(const std::string& map_id) { |
+ return base::StringPrintf("map-%s", map_id.c_str()); |
+} |
+ |
+std::string SessionStorageDatabase::MapKey(const std::string& map_id, |
+ const std::string& key) { |
+ return base::StringPrintf("map-%s-%s", map_id.c_str(), key.c_str()); |
+} |
+ |
+const char* SessionStorageDatabase::MapPrefix() { |
+ return "map-"; |
+} |
+ |
+const char* SessionStorageDatabase::NextNamespaceIdKey() { |
+ return "next-namespace-id"; |
+} |
+ |
+const char* SessionStorageDatabase::NextMapIdKey() { |
+ return "next-map-id"; |
+} |
+ |
+} // namespace dom_storage |