ServiceWorker: Migrate the script cache backend from BlockFile to Simple

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 "base/files/file_util.h"
 #include "base/lazy_instance.h"
 #include "base/location.h"
 #include "base/logging.h"
@@ skipping 55 matching lines @@
 //     (ex. "RES:123456\x00654321")
 //   value: <ServiceWorkerResourceRecord serialized as a string>
 //
 //   key: "URES:" + <int64 'uncommitted_resource_id'>
 //   value: <empty>
 //
 // Version 2
 //
 //   key: "REGID_TO_ORIGIN:" + <int64 'registration_id'>
 //   value: <GURL 'origin'>
+//
+//   key: "INITDATA_DISKCACHE_MIGRATED"
+//   value: <empty>
+//     - This entry represents that the diskcache was migrated from BlockFile
+//       backend to Simple backend (http://crbug.com/487482).
+//
+//   key: "INITDATA_OLD_DISKCACHE_EXISTS"
+//   value: <empty>
+//     - This entry represents that the diskcahes was migrate, but the old
+//       BlockFile diskcache still exists (http://crbug.com/487482).
 namespace content {
 
 namespace {
 
 const char kDatabaseVersionKey[] = "INITDATA_DB_VERSION";
 const char kNextRegIdKey[] = "INITDATA_NEXT_REGISTRATION_ID";
 const char kNextResIdKey[] = "INITDATA_NEXT_RESOURCE_ID";
 const char kNextVerIdKey[] = "INITDATA_NEXT_VERSION_ID";
 const char kUniqueOriginKey[] = "INITDATA_UNIQUE_ORIGIN:";
+const char kDiskCacheMigratedKey[] = "INITDATA_DISKCACHE_MIGRATED";
+const char kOldDiskCacheExistsKey[] = "INITDATA_OLD_DISKCACHE_EXISTS";
 
 const char kRegKeyPrefix[] = "REG:";
 const char kRegUserDataKeyPrefix[] = "REG_USER_DATA:";
 const char kRegHasUserDataKeyPrefix[] = "REG_HAS_USER_DATA:";
 const char kRegIdToOriginKeyPrefix[] = "REGID_TO_ORIGIN:";
 const char kResKeyPrefix[] = "RES:";
 const char kKeySeparator = '\x00';
+const char kEmptyValue[] = "";
 
 const char kUncommittedResIdKeyPrefix[] = "URES:";
 const char kPurgeableResIdKeyPrefix[] = "PRES:";
 
 const int64 kCurrentSchemaVersion = 2;
 
 class ServiceWorkerEnv : public leveldb_env::ChromiumEnv {
  public:
   ServiceWorkerEnv()
       : ChromiumEnv("LevelDBEnv.ServiceWorker", false /* make_backup */) {}
@@ skipping 310 matching lines @@
   if (status != STATUS_OK)
     return status;
 
   *next_avail_registration_id = next_avail_registration_id_;
   *next_avail_version_id = next_avail_version_id_;
   *next_avail_resource_id = next_avail_resource_id_;
   return STATUS_OK;
 }
 
 ServiceWorkerDatabase::Status
+ServiceWorkerDatabase::ReadDiskCacheMigrationState(
+    DiskCacheMigrationState* state) {
+  DCHECK(sequence_checker_.CalledOnValidSequencedThread());
+
+  Status status = LazyOpen(false);
+  if (IsNewOrNonexistentDatabase(status)) {
+    *state = DISKCACHE_NOT_USED;
+    return STATUS_OK;
+  }
+  if (status != STATUS_OK)
+    return status;
+
+  std::string value;
+  status = LevelDBStatusToStatus(
+      db_->Get(leveldb::ReadOptions(), kDiskCacheMigratedKey, &value));
+  if (status != STATUS_ERROR_NOT_FOUND) {
+    if (status == STATUS_OK)
+      *state = DISKCACHE_MIGRATED;
+    HandleReadResult(FROM_HERE, status);
+    return status;
+  }
+  HandleReadResult(FROM_HERE, STATUS_OK);
+
+  status = LevelDBStatusToStatus(
+      db_->Get(leveldb::ReadOptions(), kOldDiskCacheExistsKey, &value));
+  if (status != STATUS_ERROR_NOT_FOUND) {
+    if (status == STATUS_OK)
+      *state = DISKCACHE_NEEDS_TO_DELETE_OLD;
+    HandleReadResult(FROM_HERE, status);
+    return status;
+  }
+
+  *state = DISKCACHE_NEEDS_TO_MIGRATE;
+  HandleReadResult(FROM_HERE, STATUS_OK);
+  return STATUS_OK;
+}
+
+ServiceWorkerDatabase::Status
+ServiceWorkerDatabase::WriteDiskCacheMigrationState(
+    DiskCacheMigrationState state) {
+  DCHECK(sequence_checker_.CalledOnValidSequencedThread());
+  DCHECK(state == DISKCACHE_MIGRATED || state == DISKCACHE_NEEDS_TO_DELETE_OLD)
+      << state;
+
+  Status status = LazyOpen(true);
+  if (status != STATUS_OK)
+    return status;
+
+  leveldb::WriteBatch batch;
+  if (state == DISKCACHE_MIGRATED) {
+    batch.Put(kDiskCacheMigratedKey, kEmptyValue);
+    batch.Delete(kOldDiskCacheExistsKey);
+  } else if (state == DISKCACHE_NEEDS_TO_DELETE_OLD) {
+    batch.Put(kOldDiskCacheExistsKey, kEmptyValue);
+  }
+  return WriteBatch(&batch);
+}
+
+ServiceWorkerDatabase::Status
 ServiceWorkerDatabase::GetOriginsWithRegistrations(std::set<GURL>* origins) {
   DCHECK(sequence_checker_.CalledOnValidSequencedThread());
   DCHECK(origins->empty());
 
   Status status = LazyOpen(false);
   if (IsNewOrNonexistentDatabase(status))
     return STATUS_OK;
   if (status != STATUS_OK)
     return status;
 
@@ skipping 1065 matching lines @@
 
 void ServiceWorkerDatabase::HandleWriteResult(
     const tracked_objects::Location& from_here,
     Status status) {
   if (status != STATUS_OK)
     Disable(from_here, status);
   ServiceWorkerMetrics::CountWriteDatabaseResult(status);
 }
 
 }  // namespace content