[Sync] Add context proto and directory support

sync/syncable/directory_backing_store.cc

 // Copyright 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 "sync/syncable/directory_backing_store.h"
 
 #include "build/build_config.h"
 
 #include <limits>
 
@@ skipping 17 matching lines @@
 using std::string;
 
 namespace syncer {
 namespace syncable {
 
 // This just has to be big enough to hold an UPDATE or INSERT statement that
 // modifies all the columns in the entry table.
 static const string::size_type kUpdateStatementBufferSize = 2048;
 
 // Increment this version whenever updating DB tables.
-const int32 kCurrentDBVersion = 87;
+const int32 kCurrentDBVersion = 88;
 
 // Iterate over the fields of |entry| and bind each to |statement| for
 // updating.  Returns the number of args bound.
 void BindFields(const EntryKernel& entry,
                 sql::Statement* statement) {
   int index = 0;
   int i = 0;
   for (i = BEGIN_FIELDS; i < INT64_FIELDS_END; ++i) {
     statement->BindInt64(index++, entry.ref(static_cast<Int64Field>(i)));
   }
@@ skipping 202 matching lines @@
     s1.BindInt64(1, info.next_id);
     s1.BindBlob(2, info.bag_of_chips.data(), info.bag_of_chips.size());
 
     if (!s1.Run())
       return false;
     DCHECK_EQ(db_->GetLastChangeCount(), 1);
 
     sql::Statement s2(db_->GetCachedStatement(
             SQL_FROM_HERE,
             "INSERT OR REPLACE "
-            "INTO models (model_id, progress_marker, transaction_version) "
-            "VALUES (?, ?, ?)"));
+            "INTO models (model_id, "
+                         "progress_marker, "
+                         "transaction_version, "
+                         "context) "
+            "VALUES (?, ?, ?, ?)"));
 
     ModelTypeSet protocol_types = ProtocolTypes();
     for (ModelTypeSet::Iterator iter = protocol_types.First(); iter.Good();
          iter.Inc()) {
       ModelType type = iter.Get();
       // We persist not ModelType but rather a protobuf-derived ID.
       string model_id = ModelTypeEnumToModelId(type);
       string progress_marker;
       info.download_progress[type].SerializeToString(&progress_marker);
       s2.BindBlob(0, model_id.data(), model_id.length());
       s2.BindBlob(1, progress_marker.data(), progress_marker.length());
       s2.BindInt64(2, info.transaction_version[type]);
+      string context;
+      info.datatype_context[type].SerializeToString(&context);
+      s2.BindBlob(3, context.data(), context.length());
       if (!s2.Run())
         return false;
       DCHECK_EQ(db_->GetLastChangeCount(), 1);
       s2.Reset(true);
     }
   }
 
   return transaction.Commit();
 }
 
@@ skipping 120 matching lines @@
     if (MigrateVersion85To86())
       version_on_disk = 86;
   }
 
   // Version 87 migration adds a collection of attachment ids per sync entry.
   if (version_on_disk == 86) {
     if (MigrateVersion86To87())
       version_on_disk = 87;
   }
 
+  // Version 88 migration adds datatype contexts to the models table.
+  if (version_on_disk == 87) {
+    if (MigrateVersion87To88())
+      version_on_disk = 88;
+  }
+
   // If one of the migrations requested it, drop columns that aren't current.
   // It's only safe to do this after migrating all the way to the current
   // version.
   if (version_on_disk == kCurrentDBVersion && needs_column_refresh_) {
     if (!RefreshColumns())
       version_on_disk = 0;
   }
 
   // A final, alternative catch-all migration to simply re-sync everything.
   if (version_on_disk != kCurrentDBVersion) {
@@ skipping 134 matching lines @@
 
     // Verify there was only one row returned.
     DCHECK(!s.Step());
     DCHECK(s.Succeeded());
   }
 
   {
     sql::Statement s(
         db_->GetUniqueStatement(
             "SELECT model_id, progress_marker, "
-            "transaction_version FROM models"));
+            "transaction_version, context FROM models"));
 
     while (s.Step()) {
       ModelType type = ModelIdToModelTypeEnum(s.ColumnBlob(0),
                                               s.ColumnByteLength(0));
       if (type != UNSPECIFIED && type != TOP_LEVEL_FOLDER) {
         info->kernel_info.download_progress[type].ParseFromArray(
             s.ColumnBlob(1), s.ColumnByteLength(1));
         info->kernel_info.transaction_version[type] = s.ColumnInt64(2);
+        info->kernel_info.datatype_context[type].ParseFromArray(
+            s.ColumnBlob(3), s.ColumnByteLength(3));
       }
     }
     if (!s.Succeeded())
       return false;
   }
   {
     sql::Statement s(
         db_->GetUniqueStatement(
             "SELECT MAX(metahandle) FROM metas"));
     if (!s.Step())
@@ skipping 539 matching lines @@
   if (!db_->Execute(query.c_str()))
     return false;
   SetVersion(84);
   return true;
 }
 
 bool DirectoryBackingStore::MigrateVersion84To85() {
   // Version 85 removes the initial_sync_ended flag.
   if (!db_->Execute("ALTER TABLE models RENAME TO temp_models"))
     return false;
-  if (!CreateModelsTable())
+  if (!CreateV81ModelsTable())
     return false;
   if (!db_->Execute("INSERT INTO models SELECT "
                     "model_id, progress_marker, transaction_version "
                     "FROM temp_models")) {
     return false;
   }
   SafeDropTable("temp_models");
 
   SetVersion(85);
   return true;
@@ skipping 129 matching lines @@
   if (!db_->Execute(
           "ALTER TABLE metas ADD COLUMN "
           "attachment_metadata BLOB")) {
     return false;
   }
   SetVersion(87);
   needs_column_refresh_ = true;
   return true;
 }
 
+bool DirectoryBackingStore::MigrateVersion87To88() {
+  // Version 88 adds the datatype context to the models table.
+  if (!db_->Execute("ALTER TABLE models ADD COLUMN context blob"))
+    return false;
+
+  SetVersion(88);
+  return true;
+}
+
 bool DirectoryBackingStore::CreateTables() {
   DVLOG(1) << "First run, creating tables";
   // Create two little tables share_version and share_info
   if (!db_->Execute(
           "CREATE TABLE share_version ("
           "id VARCHAR(128) primary key, data INT)")) {
     return false;
   }
 
   {
@@ skipping 96 matching lines @@
   return db_->Execute(
       "CREATE TABLE models ("
       "model_id BLOB primary key, "
       "progress_marker BLOB, "
       // Gets set if the syncer ever gets updates from the
       // server and the server returns 0.  Lets us detect the
       // end of the initial sync.
       "initial_sync_ended BOOLEAN default 0)");
 }
 
+bool DirectoryBackingStore::CreateV81ModelsTable() {
+  // This is an old schema for the Models table, used from versions 81 to 87.
+  return db_->Execute(
+      "CREATE TABLE models ("
+      "model_id BLOB primary key, "
+      "progress_marker BLOB, "
+      // Gets set if the syncer ever gets updates from the
+      // server and the server returns 0.  Lets us detect the
+      // end of the initial sync.
+      "transaction_version BIGINT default 0)");
+}
+
 bool DirectoryBackingStore::CreateModelsTable() {
-  // This is the current schema for the Models table, from version 81
+  // This is the current schema for the Models table, from version 88
   // onward.  If you change the schema, you'll probably want to double-check
   // the use of this function in the v84-v85 migration.
   return db_->Execute(
       "CREATE TABLE models ("
       "model_id BLOB primary key, "
       "progress_marker BLOB, "
       // Gets set if the syncer ever gets updates from the
       // server and the server returns 0.  Lets us detect the
       // end of the initial sync.
-      "transaction_version BIGINT default 0)");
+      "transaction_version BIGINT default 0,"
+      "context BLOB)");
 }
 
 bool DirectoryBackingStore::CreateShareInfoTable(bool is_temporary) {
   const char* name = is_temporary ? "temp_share_info" : "share_info";
   string query = "CREATE TABLE ";
   query.append(name);
   // This is the current schema for the ShareInfo table, from version 76
   // onward.
   query.append(" ("
       "id TEXT primary key, "
@@ skipping 88 matching lines @@
   }
   query.append(" ) ");
   values.append(" )");
   query.append(values);
   save_statement->Assign(db_->GetUniqueStatement(
       base::StringPrintf(query.c_str(), "metas").c_str()));
 }
 
 }  // namespace syncable
 }  // namespace syncer