ServiceWorker: Implement DeleteAllDataForOrigin() in ServiceWorkerDatabase

content/browser/service_worker/service_worker_database.cc

 // Copyright 2014 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/service_worker/service_worker_database.h"
 
 #include <string>
 
 #include "base/file_util.h"
 #include "base/location.h"
@@ skipping 481 matching lines @@
 bool ServiceWorkerDatabase::WritePurgeableResourceIds(
     const std::set<int64>& ids) {
   return WriteResourceIds(kPurgeableResIdKeyPrefix, ids);
 }
 
 bool ServiceWorkerDatabase::ClearPurgeableResourceIds(
     const std::set<int64>& ids) {
   return DeleteResourceIds(kPurgeableResIdKeyPrefix, ids);
 }
 
+bool ServiceWorkerDatabase::DeleteAllDataForOrigin(const GURL& origin) {
+  DCHECK(sequence_checker_.CalledOnValidSequencedThread());
+  if (!LazyOpen(true) || is_disabled_ || !origin.is_valid())
+    return false;
+
+  leveldb::WriteBatch batch;
+
+  // Delete from the unique origin list.
+  batch.Delete(CreateUniqueOriginKey(origin));
+
+  std::vector<RegistrationData> registrations;
+  if (!GetRegistrationsForOrigin(origin, &registrations))
+    return false;
+
+  // Delete registrations and resource records.
+  for (std::vector<RegistrationData>::const_iterator itr =
+           registrations.begin(); itr != registrations.end(); ++itr) {
+    batch.Delete(CreateRegistrationKey(itr->registration_id, origin));
+    if (!DeleteResourceRecords(itr->version_id, &batch))
+      return false;
+  }
+
+  return WriteBatch(&batch);
+}
+
+bool ServiceWorkerDatabase::DeleteAllData() {
+  DCHECK(sequence_checker_.CalledOnValidSequencedThread());
+  if (!LazyOpen(true) || is_disabled_)
+    return false;
+
+  leveldb::WriteBatch batch;
+
+  // Delete until the iterator reaches the end of records.

[nhiroki, 2014/05/02 16:28:04]

Is this expected behavior? I wonder if you want to delete only registration
related data, that is, not want to delete "next xxx ids" and
"uncommitted/purgeable lists".

[jsbell, 2014/05/02 17:32:56]

If we're going to delete everything, why not just delete the database and
recreate it (or let it be recreated on next use)?

If we do keep this approach, you may want to consider forcing a compaction after
the batch commits to really free up the space on disk - depends on what the
intent of this function really is. But note
https://code.google.com/p/leveldb/issues/detail?id=221

[michaeln, 2014/05/05 21:46:36]

Right, we probably don't need this method at all. Remove it for now?

[nhiroki, 2014/05/06 15:11:32]

Okay, removed.

+  scoped_ptr<leveldb::Iterator> itr(db_->NewIterator(leveldb::ReadOptions()));
+  for (itr->SeekToFirst(); itr->Valid(); itr->Next()) {
+    if (!itr->status().ok()) {
+      HandleError(FROM_HERE, itr->status());
+      return false;
+    }
+    batch.Delete(itr->key());
+  }
+
+  if (!WriteBatch(&batch))
+    return false;
+
+  // Initialize on-memory data.
+  next_avail_registration_id_ = 0;
+  next_avail_resource_id_ = 0;
+  next_avail_version_id_ = 0;
+  is_initialized_ = false;
+
+  return false;
+}
+
 bool ServiceWorkerDatabase::LazyOpen(bool create_if_needed) {
   DCHECK(sequence_checker_.CalledOnValidSequencedThread());
   if (IsOpen())
     return true;
 
   // Do not try to open a database if we tried and failed once.
   if (is_disabled_)
     return false;
 
   // When |path_| is empty, open a database in-memory.
@@ skipping 295 matching lines @@
   // TODO(nhiroki): Add an UMA histogram.
   DLOG(ERROR) << "Failed at: " << from_here.ToString()
               << " with error: " << status.ToString();
   is_disabled_ = true;
   if (status.IsCorruption())
     was_corruption_detected_ = true;
   db_.reset();
 }
 
 }  // namespace content