Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/sync_file_system/drive_metadata_store.h" | 5 #include "chrome/browser/sync_file_system/drive_metadata_store.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 #include <vector> | 8 #include <vector> |
| 9 | 9 |
| 10 #include "base/bind.h" | 10 #include "base/bind.h" |
| (...skipping 92 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 103 | 103 |
| 104 found->second = resource_id; | 104 found->second = resource_id; |
| 105 return true; | 105 return true; |
| 106 } | 106 } |
| 107 | 107 |
| 108 } // namespace | 108 } // namespace |
| 109 | 109 |
| 110 class DriveMetadataDB { | 110 class DriveMetadataDB { |
| 111 public: | 111 public: |
| 112 enum OriginSyncType { | 112 enum OriginSyncType { |
| 113 BATCH_SYNC_ORIGIN, | |
| 114 INCREMENTAL_SYNC_ORIGIN, | 113 INCREMENTAL_SYNC_ORIGIN, |
| 115 DISABLED_ORIGIN | 114 DISABLED_ORIGIN |
| 116 }; | 115 }; |
| 117 | 116 |
| 118 typedef DriveMetadataStore::MetadataMap MetadataMap; | 117 typedef DriveMetadataStore::MetadataMap MetadataMap; |
| 119 | 118 |
| 120 DriveMetadataDB(const base::FilePath& base_dir, | 119 DriveMetadataDB(const base::FilePath& base_dir, |
| 121 base::SequencedTaskRunner* task_runner); | 120 base::SequencedTaskRunner* task_runner); |
| 122 ~DriveMetadataDB(); | 121 ~DriveMetadataDB(); |
| 123 | 122 |
| (...skipping 20 matching lines...) Expand all Loading... | |
| 144 SyncStatusCode UpdateOriginAsBatchSync(const GURL& origin, | 143 SyncStatusCode UpdateOriginAsBatchSync(const GURL& origin, |
| 145 const std::string& resource_id); | 144 const std::string& resource_id); |
| 146 SyncStatusCode UpdateOriginAsIncrementalSync(const GURL& origin, | 145 SyncStatusCode UpdateOriginAsIncrementalSync(const GURL& origin, |
| 147 const std::string& resource_id); | 146 const std::string& resource_id); |
| 148 SyncStatusCode EnableOrigin(const GURL& origin, | 147 SyncStatusCode EnableOrigin(const GURL& origin, |
| 149 const std::string& resource_id); | 148 const std::string& resource_id); |
| 150 SyncStatusCode DisableOrigin(const GURL& origin, | 149 SyncStatusCode DisableOrigin(const GURL& origin, |
| 151 const std::string& resource_id); | 150 const std::string& resource_id); |
| 152 SyncStatusCode RemoveOrigin(const GURL& origin); | 151 SyncStatusCode RemoveOrigin(const GURL& origin); |
| 153 | 152 |
| 154 SyncStatusCode GetOrigins(ResourceIdByOrigin* batch_sync_origins, | 153 SyncStatusCode GetOrigins(ResourceIdByOrigin* incremental_sync_origins, |
| 155 ResourceIdByOrigin* incremental_sync_origins, | |
| 156 ResourceIdByOrigin* disabled_origins); | 154 ResourceIdByOrigin* disabled_origins); |
| 157 | 155 |
| 158 private: | 156 private: |
| 157 friend class DriveMetadataStore; | |
| 158 | |
| 159 bool CalledOnValidThread() const { | 159 bool CalledOnValidThread() const { |
| 160 return task_runner_->RunsTasksOnCurrentThread(); | 160 return task_runner_->RunsTasksOnCurrentThread(); |
| 161 } | 161 } |
| 162 | 162 |
| 163 scoped_refptr<base::SequencedTaskRunner> task_runner_; | 163 scoped_refptr<base::SequencedTaskRunner> task_runner_; |
| 164 | 164 |
| 165 std::string db_path_; | 165 std::string db_path_; |
| 166 scoped_ptr<leveldb::DB> db_; | 166 scoped_ptr<leveldb::DB> db_; |
| 167 | 167 |
| 168 DISALLOW_COPY_AND_ASSIGN(DriveMetadataDB); | 168 DISALLOW_COPY_AND_ASSIGN(DriveMetadataDB); |
| 169 }; | 169 }; |
| 170 | 170 |
| 171 struct DriveMetadataDBContents { | 171 struct DriveMetadataDBContents { |
| 172 int64 largest_changestamp; | 172 int64 largest_changestamp; |
| 173 DriveMetadataStore::MetadataMap metadata_map; | 173 DriveMetadataStore::MetadataMap metadata_map; |
| 174 std::string sync_root_directory_resource_id; | 174 std::string sync_root_directory_resource_id; |
| 175 ResourceIdByOrigin batch_sync_origins; | |
| 176 ResourceIdByOrigin incremental_sync_origins; | 175 ResourceIdByOrigin incremental_sync_origins; |
| 177 ResourceIdByOrigin disabled_origins; | 176 ResourceIdByOrigin disabled_origins; |
| 178 }; | 177 }; |
| 179 | 178 |
| 180 namespace { | 179 namespace { |
| 181 | 180 |
| 182 SyncStatusCode InitializeDBOnFileThread(DriveMetadataDB* db, | 181 SyncStatusCode InitializeDBOnFileThread(DriveMetadataDB* db, |
| 183 DriveMetadataDBContents* contents) { | 182 DriveMetadataDBContents* contents) { |
| 184 DCHECK(db); | 183 DCHECK(db); |
| 185 DCHECK(contents); | 184 DCHECK(contents); |
| 186 | 185 |
| 187 contents->largest_changestamp = 0; | 186 contents->largest_changestamp = 0; |
| 188 contents->metadata_map.clear(); | 187 contents->metadata_map.clear(); |
| 189 contents->batch_sync_origins.clear(); | |
| 190 contents->incremental_sync_origins.clear(); | 188 contents->incremental_sync_origins.clear(); |
| 191 contents->disabled_origins.clear(); | 189 contents->disabled_origins.clear(); |
| 192 | 190 |
| 193 bool created = false; | 191 bool created = false; |
| 194 SyncStatusCode status = db->Initialize(&created); | 192 SyncStatusCode status = db->Initialize(&created); |
| 195 if (status != SYNC_STATUS_OK) | 193 if (status != SYNC_STATUS_OK) |
| 196 return status; | 194 return status; |
| 197 | 195 |
| 198 if (!created) { | 196 if (!created) { |
| 199 status = db->MigrateDatabaseIfNeeded(); | 197 status = db->MigrateDatabaseIfNeeded(); |
| 200 if (status != SYNC_STATUS_OK) { | 198 if (status != SYNC_STATUS_OK) { |
| 201 LOG(WARNING) << "Failed to migrate DriveMetadataStore to latest version."; | 199 LOG(WARNING) << "Failed to migrate DriveMetadataStore to latest version."; |
| 202 return status; | 200 return status; |
| 203 } | 201 } |
| 204 } | 202 } |
| 205 | 203 |
| 206 return db->ReadContents(contents); | 204 return db->ReadContents(contents); |
| 207 } | 205 } |
| 208 | 206 |
| 209 // Returns a key string for the given origin. | 207 // Returns a key string for the given origin. |
| 210 // For example, when |origin| is "http://www.example.com" and |sync_type| is | 208 // For example, when |origin| is "http://www.example.com" and |sync_type| is |
| 211 // BATCH_SYNC_ORIGIN, returns "BSYNC_ORIGIN: http://www.example.com". | 209 // BATCH_SYNC_ORIGIN, returns "BSYNC_ORIGIN: http://www.example.com". |
| 212 std::string CreateKeyForOriginRoot(const GURL& origin, | 210 std::string CreateKeyForOriginRoot(const GURL& origin, |
| 213 DriveMetadataDB::OriginSyncType sync_type) { | 211 DriveMetadataDB::OriginSyncType sync_type) { |
| 214 DCHECK(origin.is_valid()); | 212 DCHECK(origin.is_valid()); |
| 215 switch (sync_type) { | 213 switch (sync_type) { |
| 216 case DriveMetadataDB::BATCH_SYNC_ORIGIN: | |
| 217 return kDriveBatchSyncOriginKeyPrefix + origin.spec(); | |
| 218 case DriveMetadataDB::INCREMENTAL_SYNC_ORIGIN: | 214 case DriveMetadataDB::INCREMENTAL_SYNC_ORIGIN: |
| 219 return kDriveIncrementalSyncOriginKeyPrefix + origin.spec(); | 215 return kDriveIncrementalSyncOriginKeyPrefix + origin.spec(); |
| 220 case DriveMetadataDB::DISABLED_ORIGIN: | 216 case DriveMetadataDB::DISABLED_ORIGIN: |
| 221 return kDriveDisabledOriginKeyPrefix + origin.spec(); | 217 return kDriveDisabledOriginKeyPrefix + origin.spec(); |
| 222 } | 218 } |
| 223 NOTREACHED(); | 219 NOTREACHED(); |
| 224 return std::string(); | 220 return std::string(); |
| 225 } | 221 } |
| 226 | 222 |
| 227 void AddOriginsToVector(std::vector<GURL>* all_origins, | 223 void AddOriginsToVector(std::vector<GURL>* all_origins, |
| (...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 287 | 283 |
| 288 db_status_ = status; | 284 db_status_ = status; |
| 289 if (status != SYNC_STATUS_OK) { | 285 if (status != SYNC_STATUS_OK) { |
| 290 callback.Run(status, false); | 286 callback.Run(status, false); |
| 291 return; | 287 return; |
| 292 } | 288 } |
| 293 | 289 |
| 294 largest_changestamp_ = contents->largest_changestamp; | 290 largest_changestamp_ = contents->largest_changestamp; |
| 295 metadata_map_.swap(contents->metadata_map); | 291 metadata_map_.swap(contents->metadata_map); |
| 296 sync_root_directory_resource_id_ = contents->sync_root_directory_resource_id; | 292 sync_root_directory_resource_id_ = contents->sync_root_directory_resource_id; |
| 297 batch_sync_origins_.swap(contents->batch_sync_origins); | |
| 298 incremental_sync_origins_.swap(contents->incremental_sync_origins); | 293 incremental_sync_origins_.swap(contents->incremental_sync_origins); |
| 299 disabled_origins_.swap(contents->disabled_origins); | 294 disabled_origins_.swap(contents->disabled_origins); |
| 300 // |largest_changestamp_| is set to 0 for a fresh empty database. | 295 // |largest_changestamp_| is set to 0 for a fresh empty database. |
| 301 | 296 |
| 302 origin_by_resource_id_.clear(); | 297 origin_by_resource_id_.clear(); |
| 303 InsertReverseMap(batch_sync_origins_, &origin_by_resource_id_); | |
| 304 InsertReverseMap(incremental_sync_origins_, &origin_by_resource_id_); | 298 InsertReverseMap(incremental_sync_origins_, &origin_by_resource_id_); |
| 305 InsertReverseMap(disabled_origins_, &origin_by_resource_id_); | 299 InsertReverseMap(disabled_origins_, &origin_by_resource_id_); |
| 306 | 300 |
| 307 callback.Run(status, largest_changestamp_ <= 0); | 301 callback.Run(status, largest_changestamp_ <= 0); |
| 308 } | 302 } |
| 309 | 303 |
| 310 void DriveMetadataStore::RestoreSyncRootDirectory( | 304 void DriveMetadataStore::RestoreSyncRootDirectory( |
| 311 const SyncStatusCallback& callback) { | 305 const SyncStatusCallback& callback) { |
| 312 DCHECK(CalledOnValidThread()); | 306 DCHECK(CalledOnValidThread()); |
| 313 std::string* sync_root_directory_resource_id = new std::string; | 307 std::string* sync_root_directory_resource_id = new std::string; |
| (...skipping 20 matching lines...) Expand all Loading... | |
| 334 return; | 328 return; |
| 335 } | 329 } |
| 336 | 330 |
| 337 sync_root_directory_resource_id_.swap(*sync_root_directory_resource_id); | 331 sync_root_directory_resource_id_.swap(*sync_root_directory_resource_id); |
| 338 callback.Run(status); | 332 callback.Run(status); |
| 339 } | 333 } |
| 340 | 334 |
| 341 void DriveMetadataStore::RestoreOrigins( | 335 void DriveMetadataStore::RestoreOrigins( |
| 342 const SyncStatusCallback& callback) { | 336 const SyncStatusCallback& callback) { |
| 343 DCHECK(CalledOnValidThread()); | 337 DCHECK(CalledOnValidThread()); |
| 344 ResourceIdByOrigin* batch_sync_origins = new ResourceIdByOrigin; | |
| 345 ResourceIdByOrigin* incremental_sync_origins = new ResourceIdByOrigin; | 338 ResourceIdByOrigin* incremental_sync_origins = new ResourceIdByOrigin; |
| 346 ResourceIdByOrigin* disabled_origins = new ResourceIdByOrigin; | 339 ResourceIdByOrigin* disabled_origins = new ResourceIdByOrigin; |
| 347 base::PostTaskAndReplyWithResult( | 340 base::PostTaskAndReplyWithResult( |
| 348 file_task_runner_, FROM_HERE, | 341 file_task_runner_, FROM_HERE, |
| 349 base::Bind(&DriveMetadataDB::GetOrigins, | 342 base::Bind(&DriveMetadataDB::GetOrigins, |
| 350 base::Unretained(db_.get()), | 343 base::Unretained(db_.get()), |
| 351 batch_sync_origins, | |
| 352 incremental_sync_origins, | 344 incremental_sync_origins, |
| 353 disabled_origins), | 345 disabled_origins), |
| 354 base::Bind(&DriveMetadataStore::DidRestoreOrigins, | 346 base::Bind(&DriveMetadataStore::DidRestoreOrigins, |
| 355 AsWeakPtr(), callback, | 347 AsWeakPtr(), callback, |
| 356 base::Owned(batch_sync_origins), | |
| 357 base::Owned(incremental_sync_origins), | 348 base::Owned(incremental_sync_origins), |
| 358 base::Owned(disabled_origins))); | 349 base::Owned(disabled_origins))); |
| 359 } | 350 } |
| 360 | 351 |
| 361 void DriveMetadataStore::DidRestoreOrigins( | 352 void DriveMetadataStore::DidRestoreOrigins( |
| 362 const SyncStatusCallback& callback, | 353 const SyncStatusCallback& callback, |
| 363 ResourceIdByOrigin* batch_sync_origins, | |
| 364 ResourceIdByOrigin* incremental_sync_origins, | 354 ResourceIdByOrigin* incremental_sync_origins, |
| 365 ResourceIdByOrigin* disabled_origins, | 355 ResourceIdByOrigin* disabled_origins, |
| 366 SyncStatusCode status) { | 356 SyncStatusCode status) { |
| 367 DCHECK(CalledOnValidThread()); | 357 DCHECK(CalledOnValidThread()); |
| 368 DCHECK(batch_sync_origins); | |
| 369 DCHECK(incremental_sync_origins); | 358 DCHECK(incremental_sync_origins); |
| 370 DCHECK(disabled_origins); | 359 DCHECK(disabled_origins); |
| 371 | 360 |
| 372 db_status_ = status; | 361 db_status_ = status; |
| 373 if (status != SYNC_STATUS_OK) { | 362 if (status != SYNC_STATUS_OK) { |
| 374 callback.Run(status); | 363 callback.Run(status); |
| 375 return; | 364 return; |
| 376 } | 365 } |
| 377 | 366 |
| 378 batch_sync_origins_.swap(*batch_sync_origins); | |
| 379 incremental_sync_origins_.swap(*incremental_sync_origins); | 367 incremental_sync_origins_.swap(*incremental_sync_origins); |
| 380 disabled_origins_.swap(*disabled_origins); | 368 disabled_origins_.swap(*disabled_origins); |
| 381 | 369 |
| 382 origin_by_resource_id_.clear(); | 370 origin_by_resource_id_.clear(); |
| 383 InsertReverseMap(batch_sync_origins_, &origin_by_resource_id_); | |
| 384 InsertReverseMap(incremental_sync_origins_, &origin_by_resource_id_); | 371 InsertReverseMap(incremental_sync_origins_, &origin_by_resource_id_); |
| 385 InsertReverseMap(disabled_origins_, &origin_by_resource_id_); | 372 InsertReverseMap(disabled_origins_, &origin_by_resource_id_); |
| 386 | 373 |
| 387 callback.Run(status); | 374 callback.Run(status); |
| 388 } | 375 } |
| 389 | 376 |
| 377 leveldb::DB* DriveMetadataStore::GetDBInstanceForTesting() { | |
| 378 return db_->db_.get(); | |
| 379 } | |
| 380 | |
| 390 void DriveMetadataStore::SetLargestChangeStamp( | 381 void DriveMetadataStore::SetLargestChangeStamp( |
| 391 int64 largest_changestamp, | 382 int64 largest_changestamp, |
| 392 const SyncStatusCallback& callback) { | 383 const SyncStatusCallback& callback) { |
| 393 DCHECK(CalledOnValidThread()); | 384 DCHECK(CalledOnValidThread()); |
| 394 DCHECK_EQ(SYNC_STATUS_OK, db_status_); | 385 DCHECK_EQ(SYNC_STATUS_OK, db_status_); |
| 395 largest_changestamp_ = largest_changestamp; | 386 largest_changestamp_ = largest_changestamp; |
| 396 base::PostTaskAndReplyWithResult( | 387 base::PostTaskAndReplyWithResult( |
| 397 file_task_runner_, FROM_HERE, | 388 file_task_runner_, FROM_HERE, |
| 398 base::Bind(&DriveMetadataDB::SetLargestChangestamp, | 389 base::Bind(&DriveMetadataDB::SetLargestChangestamp, |
| 399 base::Unretained(db_.get()), largest_changestamp), | 390 base::Unretained(db_.get()), largest_changestamp), |
| (...skipping 67 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 467 if (found == found_origin->second.end()) | 458 if (found == found_origin->second.end()) |
| 468 return SYNC_DATABASE_ERROR_NOT_FOUND; | 459 return SYNC_DATABASE_ERROR_NOT_FOUND; |
| 469 | 460 |
| 470 *metadata = found->second; | 461 *metadata = found->second; |
| 471 return SYNC_STATUS_OK; | 462 return SYNC_STATUS_OK; |
| 472 } | 463 } |
| 473 | 464 |
| 474 void DriveMetadataStore::AddIncrementalSyncOrigin( | 465 void DriveMetadataStore::AddIncrementalSyncOrigin( |
| 475 const GURL& origin, | 466 const GURL& origin, |
| 476 const std::string& resource_id) { | 467 const std::string& resource_id) { |
| 477 AddBatchSyncOrigin(origin, resource_id); | 468 DCHECK(CalledOnValidThread()); |
| 478 MoveBatchSyncOriginToIncremental(origin); | 469 DCHECK(!IsIncrementalSyncOrigin(origin)); |
| 470 DCHECK(!IsOriginDisabled(origin)); | |
| 471 DCHECK_EQ(SYNC_STATUS_OK, db_status_); | |
| 472 | |
| 473 incremental_sync_origins_.insert(std::make_pair(origin, resource_id)); | |
| 474 origin_by_resource_id_.insert(std::make_pair(resource_id, origin)); | |
| 475 | |
| 476 // Store a pair of |origin| and |resource_id| in the DB. | |
| 477 base::PostTaskAndReplyWithResult( | |
| 478 file_task_runner_, FROM_HERE, | |
| 479 base::Bind(&DriveMetadataDB::UpdateOriginAsIncrementalSync, | |
| 480 base::Unretained(db_.get()), origin, resource_id), | |
| 481 base::Bind(&DriveMetadataStore::UpdateDBStatus, AsWeakPtr())); | |
| 479 } | 482 } |
| 480 | 483 |
| 481 void DriveMetadataStore::SetSyncRootDirectory(const std::string& resource_id) { | 484 void DriveMetadataStore::SetSyncRootDirectory(const std::string& resource_id) { |
| 482 DCHECK(CalledOnValidThread()); | 485 DCHECK(CalledOnValidThread()); |
| 483 | 486 |
| 484 sync_root_directory_resource_id_ = resource_id; | 487 sync_root_directory_resource_id_ = resource_id; |
| 485 | 488 |
| 486 // Set the resource ID for the sync root directory in the DB. | 489 // Set the resource ID for the sync root directory in the DB. |
| 487 base::PostTaskAndReplyWithResult( | 490 base::PostTaskAndReplyWithResult( |
| 488 file_task_runner_, FROM_HERE, | 491 file_task_runner_, FROM_HERE, |
| 489 base::Bind(&DriveMetadataDB::SetSyncRootDirectory, | 492 base::Bind(&DriveMetadataDB::SetSyncRootDirectory, |
| 490 base::Unretained(db_.get()), resource_id), | 493 base::Unretained(db_.get()), resource_id), |
| 491 base::Bind(&DriveMetadataStore::UpdateDBStatus, AsWeakPtr())); | 494 base::Bind(&DriveMetadataStore::UpdateDBStatus, AsWeakPtr())); |
| 492 } | 495 } |
| 493 | 496 |
| 494 void DriveMetadataStore::SetOriginRootDirectory( | 497 void DriveMetadataStore::SetOriginRootDirectory( |
| 495 const GURL& origin, | 498 const GURL& origin, |
| 496 const std::string& resource_id) { | 499 const std::string& resource_id) { |
| 497 DCHECK(CalledOnValidThread()); | 500 DCHECK(CalledOnValidThread()); |
| 498 DCHECK(IsKnownOrigin(origin)); | 501 DCHECK(IsKnownOrigin(origin)); |
| 499 | 502 |
| 500 DriveMetadataDB::OriginSyncType sync_type; | 503 DriveMetadataDB::OriginSyncType sync_type; |
| 501 if (UpdateResourceIdMap(&batch_sync_origins_, &origin_by_resource_id_, | 504 if (UpdateResourceIdMap( |
| 502 origin, resource_id)) { | |
| 503 sync_type = DriveMetadataDB::BATCH_SYNC_ORIGIN; | |
| 504 } else if (UpdateResourceIdMap( | |
| 505 &incremental_sync_origins_, &origin_by_resource_id_, | 505 &incremental_sync_origins_, &origin_by_resource_id_, |
| 506 origin, resource_id)) { | 506 origin, resource_id)) { |
| 507 sync_type = DriveMetadataDB::INCREMENTAL_SYNC_ORIGIN; | 507 sync_type = DriveMetadataDB::INCREMENTAL_SYNC_ORIGIN; |
| 508 } else if (UpdateResourceIdMap(&disabled_origins_, &origin_by_resource_id_, | 508 } else if (UpdateResourceIdMap(&disabled_origins_, &origin_by_resource_id_, |
| 509 origin, resource_id)) { | 509 origin, resource_id)) { |
| 510 sync_type = DriveMetadataDB::DISABLED_ORIGIN; | 510 sync_type = DriveMetadataDB::DISABLED_ORIGIN; |
| 511 } else { | 511 } else { |
| 512 return; | 512 return; |
| 513 } | 513 } |
| 514 base::PostTaskAndReplyWithResult( | 514 base::PostTaskAndReplyWithResult( |
| 515 file_task_runner_, FROM_HERE, | 515 file_task_runner_, FROM_HERE, |
| 516 base::Bind(&DriveMetadataDB::SetOriginRootDirectory, | 516 base::Bind(&DriveMetadataDB::SetOriginRootDirectory, |
| 517 base::Unretained(db_.get()), origin, sync_type, resource_id), | 517 base::Unretained(db_.get()), origin, sync_type, resource_id), |
| 518 base::Bind(&DriveMetadataStore::UpdateDBStatus, AsWeakPtr())); | 518 base::Bind(&DriveMetadataStore::UpdateDBStatus, AsWeakPtr())); |
| 519 } | 519 } |
| 520 | 520 |
| 521 bool DriveMetadataStore::IsKnownOrigin(const GURL& origin) const { | 521 bool DriveMetadataStore::IsKnownOrigin(const GURL& origin) const { |
| 522 DCHECK(CalledOnValidThread()); | 522 DCHECK(CalledOnValidThread()); |
| 523 return IsBatchSyncOrigin(origin) || | 523 return IsIncrementalSyncOrigin(origin) || IsOriginDisabled(origin); |
| 524 IsIncrementalSyncOrigin(origin) || | |
| 525 IsOriginDisabled(origin); | |
| 526 } | |
| 527 | |
| 528 bool DriveMetadataStore::IsBatchSyncOrigin(const GURL& origin) const { | |
| 529 DCHECK(CalledOnValidThread()); | |
| 530 return ContainsKey(batch_sync_origins_, origin); | |
| 531 } | 524 } |
| 532 | 525 |
| 533 bool DriveMetadataStore::IsIncrementalSyncOrigin(const GURL& origin) const { | 526 bool DriveMetadataStore::IsIncrementalSyncOrigin(const GURL& origin) const { |
| 534 DCHECK(CalledOnValidThread()); | 527 DCHECK(CalledOnValidThread()); |
| 535 return ContainsKey(incremental_sync_origins_, origin); | 528 return ContainsKey(incremental_sync_origins_, origin); |
| 536 } | 529 } |
| 537 | 530 |
| 538 bool DriveMetadataStore::IsOriginDisabled(const GURL& origin) const { | 531 bool DriveMetadataStore::IsOriginDisabled(const GURL& origin) const { |
| 539 DCHECK(CalledOnValidThread()); | 532 DCHECK(CalledOnValidThread()); |
| 540 return ContainsKey(disabled_origins_, origin); | 533 return ContainsKey(disabled_origins_, origin); |
| (...skipping 26 matching lines...) Expand all Loading... | |
| 567 base::Bind(&DriveMetadataStore::DidUpdateOrigin, AsWeakPtr(), callback)); | 560 base::Bind(&DriveMetadataStore::DidUpdateOrigin, AsWeakPtr(), callback)); |
| 568 } | 561 } |
| 569 | 562 |
| 570 void DriveMetadataStore::DisableOrigin( | 563 void DriveMetadataStore::DisableOrigin( |
| 571 const GURL& origin, | 564 const GURL& origin, |
| 572 const SyncStatusCallback& callback) { | 565 const SyncStatusCallback& callback) { |
| 573 DCHECK(CalledOnValidThread()); | 566 DCHECK(CalledOnValidThread()); |
| 574 | 567 |
| 575 std::string resource_id; | 568 std::string resource_id; |
| 576 std::map<GURL, std::string>::iterator found = | 569 std::map<GURL, std::string>::iterator found = |
| 577 batch_sync_origins_.find(origin); | 570 incremental_sync_origins_.find(origin); |
| 578 if (found != batch_sync_origins_.end()) { | |
| 579 resource_id = found->second; | |
| 580 batch_sync_origins_.erase(found); | |
| 581 } | |
| 582 | |
| 583 found = incremental_sync_origins_.find(origin); | |
| 584 if (found != incremental_sync_origins_.end()) { | 571 if (found != incremental_sync_origins_.end()) { |
| 585 resource_id = found->second; | 572 resource_id = found->second; |
| 586 incremental_sync_origins_.erase(found); | 573 incremental_sync_origins_.erase(found); |
| 587 } | 574 } |
| 588 | 575 |
| 589 if (resource_id.empty()) { | 576 if (resource_id.empty()) { |
| 590 // |origin| has not been registered yet. | 577 // |origin| has not been registered yet. |
| 591 return; | 578 return; |
| 592 } | 579 } |
| 593 | 580 |
| 594 disabled_origins_.insert(std::make_pair(origin, resource_id)); | 581 disabled_origins_.insert(std::make_pair(origin, resource_id)); |
| 595 | 582 |
| 596 // Store a pair of |origin| and |resource_id| in the DB. | 583 // Store a pair of |origin| and |resource_id| in the DB. |
| 597 base::PostTaskAndReplyWithResult( | 584 base::PostTaskAndReplyWithResult( |
| 598 file_task_runner_, FROM_HERE, | 585 file_task_runner_, FROM_HERE, |
| 599 base::Bind(&DriveMetadataDB::DisableOrigin, | 586 base::Bind(&DriveMetadataDB::DisableOrigin, |
| 600 base::Unretained(db_.get()), origin, resource_id), | 587 base::Unretained(db_.get()), origin, resource_id), |
| 601 base::Bind(&DriveMetadataStore::DidUpdateOrigin, AsWeakPtr(), callback)); | 588 base::Bind(&DriveMetadataStore::DidUpdateOrigin, AsWeakPtr(), callback)); |
| 602 } | 589 } |
| 603 | 590 |
| 604 void DriveMetadataStore::RemoveOrigin( | 591 void DriveMetadataStore::RemoveOrigin( |
| 605 const GURL& origin, | 592 const GURL& origin, |
| 606 const SyncStatusCallback& callback) { | 593 const SyncStatusCallback& callback) { |
| 607 DCHECK(CalledOnValidThread()); | 594 DCHECK(CalledOnValidThread()); |
| 608 | 595 |
| 609 metadata_map_.erase(origin); | 596 metadata_map_.erase(origin); |
| 610 | 597 |
| 611 std::string resource_id; | 598 std::string resource_id; |
| 612 if (EraseIfExists(&batch_sync_origins_, origin, &resource_id) || | 599 if (EraseIfExists(&incremental_sync_origins_, origin, &resource_id) || |
| 613 EraseIfExists(&incremental_sync_origins_, origin, &resource_id) || | |
| 614 EraseIfExists(&disabled_origins_, origin, &resource_id)) { | 600 EraseIfExists(&disabled_origins_, origin, &resource_id)) { |
| 615 origin_by_resource_id_.erase(resource_id); | 601 origin_by_resource_id_.erase(resource_id); |
| 616 } | 602 } |
| 617 | 603 |
| 618 base::PostTaskAndReplyWithResult( | 604 base::PostTaskAndReplyWithResult( |
| 619 file_task_runner_, FROM_HERE, | 605 file_task_runner_, FROM_HERE, |
| 620 base::Bind(&DriveMetadataDB::RemoveOrigin, | 606 base::Bind(&DriveMetadataDB::RemoveOrigin, |
| 621 base::Unretained(db_.get()), origin), | 607 base::Unretained(db_.get()), origin), |
| 622 base::Bind(&DriveMetadataStore::DidUpdateOrigin, AsWeakPtr(), callback)); | 608 base::Bind(&DriveMetadataStore::DidUpdateOrigin, AsWeakPtr(), callback)); |
| 623 } | 609 } |
| 624 | 610 |
| 625 void DriveMetadataStore::DidUpdateOrigin( | 611 void DriveMetadataStore::DidUpdateOrigin( |
| 626 const SyncStatusCallback& callback, | 612 const SyncStatusCallback& callback, |
| 627 SyncStatusCode status) { | 613 SyncStatusCode status) { |
| 628 UpdateDBStatus(status); | 614 UpdateDBStatus(status); |
| 629 callback.Run(status); | 615 callback.Run(status); |
| 630 } | 616 } |
| 631 | 617 |
| 632 void DriveMetadataStore::AddBatchSyncOrigin(const GURL& origin, | |
| 633 const std::string& resource_id) { | |
| 634 DCHECK(CalledOnValidThread()); | |
| 635 DCHECK(!IsIncrementalSyncOrigin(origin)); | |
| 636 DCHECK(!IsOriginDisabled(origin)); | |
| 637 DCHECK_EQ(SYNC_STATUS_OK, db_status_); | |
| 638 | |
| 639 if (IsBatchSyncOrigin(origin)) | |
| 640 return; | |
| 641 | |
| 642 batch_sync_origins_.insert(std::make_pair(origin, resource_id)); | |
| 643 origin_by_resource_id_.insert(std::make_pair(resource_id, origin)); | |
| 644 | |
| 645 // Store a pair of |origin| and |resource_id| in the DB. | |
| 646 base::PostTaskAndReplyWithResult( | |
| 647 file_task_runner_, FROM_HERE, | |
| 648 base::Bind(&DriveMetadataDB::UpdateOriginAsBatchSync, | |
| 649 base::Unretained(db_.get()), origin, resource_id), | |
| 650 base::Bind(&DriveMetadataStore::UpdateDBStatus, AsWeakPtr())); | |
| 651 } | |
| 652 | |
| 653 void DriveMetadataStore::MoveBatchSyncOriginToIncremental(const GURL& origin) { | |
| 654 DCHECK(CalledOnValidThread()); | |
| 655 DCHECK(IsBatchSyncOrigin(origin)); | |
| 656 DCHECK(!IsIncrementalSyncOrigin(origin)); | |
| 657 DCHECK(!IsOriginDisabled(origin)); | |
| 658 DCHECK_EQ(SYNC_STATUS_OK, db_status_); | |
| 659 | |
| 660 std::map<GURL, std::string>::iterator found = | |
| 661 batch_sync_origins_.find(origin); | |
| 662 incremental_sync_origins_.insert(std::make_pair(origin, found->second)); | |
| 663 | |
| 664 // Store a pair of |origin| and |resource_id| in the DB. | |
| 665 base::PostTaskAndReplyWithResult( | |
| 666 file_task_runner_, FROM_HERE, | |
| 667 base::Bind(&DriveMetadataDB::UpdateOriginAsIncrementalSync, | |
| 668 base::Unretained(db_.get()), origin, found->second), | |
| 669 base::Bind(&DriveMetadataStore::UpdateDBStatus, AsWeakPtr())); | |
| 670 | |
| 671 batch_sync_origins_.erase(found); | |
| 672 } | |
| 673 | |
| 674 void DriveMetadataStore::UpdateDBStatus(SyncStatusCode status) { | 618 void DriveMetadataStore::UpdateDBStatus(SyncStatusCode status) { |
| 675 DCHECK(CalledOnValidThread()); | 619 DCHECK(CalledOnValidThread()); |
| 676 if (db_status_ != SYNC_STATUS_OK && | 620 if (db_status_ != SYNC_STATUS_OK && |
| 677 db_status_ != SYNC_DATABASE_ERROR_NOT_FOUND) { | 621 db_status_ != SYNC_DATABASE_ERROR_NOT_FOUND) { |
| 678 // TODO(tzik): Handle database corruption. http://crbug.com/153709 | 622 // TODO(tzik): Handle database corruption. http://crbug.com/153709 |
| 679 db_status_ = status; | 623 db_status_ = status; |
| 680 LOG(WARNING) << "DriveMetadataStore turned to wrong state: " << status; | 624 LOG(WARNING) << "DriveMetadataStore turned to wrong state: " << status; |
| 681 return; | 625 return; |
| 682 } | 626 } |
| 683 db_status_ = SYNC_STATUS_OK; | 627 db_status_ = SYNC_STATUS_OK; |
| (...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 741 // initialized) just return empty string, as the origin directories | 685 // initialized) just return empty string, as the origin directories |
| 742 // in the root directory must have become invalid now too. | 686 // in the root directory must have become invalid now too. |
| 743 if (sync_root_directory().empty()) | 687 if (sync_root_directory().empty()) |
| 744 return std::string(); | 688 return std::string(); |
| 745 | 689 |
| 746 ResourceIdByOrigin::const_iterator found = | 690 ResourceIdByOrigin::const_iterator found = |
| 747 incremental_sync_origins_.find(origin); | 691 incremental_sync_origins_.find(origin); |
| 748 if (found != incremental_sync_origins_.end()) | 692 if (found != incremental_sync_origins_.end()) |
| 749 return found->second; | 693 return found->second; |
| 750 | 694 |
| 751 found = batch_sync_origins_.find(origin); | |
| 752 if (found != batch_sync_origins_.end()) | |
| 753 return found->second; | |
| 754 | |
| 755 found = disabled_origins_.find(origin); | 695 found = disabled_origins_.find(origin); |
| 756 if (found != disabled_origins_.end()) | 696 if (found != disabled_origins_.end()) |
| 757 return found->second; | 697 return found->second; |
| 758 | 698 |
| 759 return std::string(); | 699 return std::string(); |
| 760 } | 700 } |
| 761 | 701 |
| 762 void DriveMetadataStore::GetAllOrigins(std::vector<GURL>* origins) { | 702 void DriveMetadataStore::GetAllOrigins(std::vector<GURL>* origins) { |
| 763 DCHECK(CalledOnValidThread()); | 703 DCHECK(CalledOnValidThread()); |
| 764 DCHECK(origins); | 704 DCHECK(origins); |
| 765 origins->clear(); | 705 origins->clear(); |
| 766 origins->reserve(batch_sync_origins_.size() + | 706 origins->reserve(incremental_sync_origins_.size() + |
| 767 incremental_sync_origins_.size() + | |
| 768 disabled_origins_.size()); | 707 disabled_origins_.size()); |
| 769 AddOriginsToVector(origins, batch_sync_origins_); | |
| 770 AddOriginsToVector(origins, incremental_sync_origins_); | 708 AddOriginsToVector(origins, incremental_sync_origins_); |
| 771 AddOriginsToVector(origins, disabled_origins_); | 709 AddOriginsToVector(origins, disabled_origins_); |
| 772 } | 710 } |
| 773 | 711 |
| 774 bool DriveMetadataStore::GetOriginByOriginRootDirectoryId( | 712 bool DriveMetadataStore::GetOriginByOriginRootDirectoryId( |
| 775 const std::string& resource_id, | 713 const std::string& resource_id, |
| 776 GURL* origin) { | 714 GURL* origin) { |
| 777 DCHECK(CalledOnValidThread()); | 715 DCHECK(CalledOnValidThread()); |
| 778 DCHECK(origin); | 716 DCHECK(origin); |
| 779 | 717 |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 820 status = db->Put(leveldb::WriteOptions(), | 758 status = db->Put(leveldb::WriteOptions(), |
| 821 kDatabaseVersionKey, | 759 kDatabaseVersionKey, |
| 822 base::Int64ToString(kCurrentDatabaseVersion)); | 760 base::Int64ToString(kCurrentDatabaseVersion)); |
| 823 if (!status.ok()) { | 761 if (!status.ok()) { |
| 824 delete db; | 762 delete db; |
| 825 return LevelDBStatusToSyncStatusCode(status); | 763 return LevelDBStatusToSyncStatusCode(status); |
| 826 } | 764 } |
| 827 } | 765 } |
| 828 | 766 |
| 829 db_.reset(db); | 767 db_.reset(db); |
| 830 return SYNC_STATUS_OK; | 768 |
| 769 | |
| 770 // Remove legacy batch sync origin entries that are no longer needed. | |
| 771 leveldb::WriteBatch batch; | |
| 772 scoped_ptr<leveldb::Iterator> batch_origin_itr( | |
| 773 db_->NewIterator(leveldb::ReadOptions())); | |
| 774 for (batch_origin_itr->Seek(kDriveBatchSyncOriginKeyPrefix); | |
| 775 batch_origin_itr->Valid(); | |
| 776 batch_origin_itr->Next()) { | |
| 777 std::string key = batch_origin_itr->key().ToString(); | |
| 778 if (!StartsWithASCII(key, kDriveBatchSyncOriginKeyPrefix, false)) | |
|
tzik
2013/05/27 12:14:10
s/false/true/
This should be case sensitive.
calvinlo
2013/05/28 02:18:55
Done.
| |
| 779 break; | |
| 780 | |
| 781 batch.Delete(key); | |
| 782 } | |
| 783 status = db_->Write(leveldb::WriteOptions(), &batch); | |
| 784 return LevelDBStatusToSyncStatusCode(status); | |
| 831 } | 785 } |
| 832 | 786 |
| 833 SyncStatusCode DriveMetadataDB::ReadContents( | 787 SyncStatusCode DriveMetadataDB::ReadContents( |
| 834 DriveMetadataDBContents* contents) { | 788 DriveMetadataDBContents* contents) { |
| 835 DCHECK(CalledOnValidThread()); | 789 DCHECK(CalledOnValidThread()); |
| 836 DCHECK(db_.get()); | 790 DCHECK(db_.get()); |
| 837 DCHECK(contents); | 791 DCHECK(contents); |
| 838 | 792 |
| 839 contents->largest_changestamp = 0; | 793 contents->largest_changestamp = 0; |
| 840 contents->metadata_map.clear(); | 794 contents->metadata_map.clear(); |
| 841 contents->batch_sync_origins.clear(); | |
| 842 contents->incremental_sync_origins.clear(); | 795 contents->incremental_sync_origins.clear(); |
| 843 | 796 |
| 844 scoped_ptr<leveldb::Iterator> itr(db_->NewIterator(leveldb::ReadOptions())); | 797 scoped_ptr<leveldb::Iterator> itr(db_->NewIterator(leveldb::ReadOptions())); |
| 845 for (itr->SeekToFirst(); itr->Valid(); itr->Next()) { | 798 for (itr->SeekToFirst(); itr->Valid(); itr->Next()) { |
| 846 std::string key = itr->key().ToString(); | 799 std::string key = itr->key().ToString(); |
| 847 if (key == kChangeStampKey) { | 800 if (key == kChangeStampKey) { |
| 848 bool success = base::StringToInt64(itr->value().ToString(), | 801 bool success = base::StringToInt64(itr->value().ToString(), |
| 849 &contents->largest_changestamp); | 802 &contents->largest_changestamp); |
| 850 DCHECK(success); | 803 DCHECK(success); |
| 851 } else if (StartsWithASCII(key, kDriveMetadataKeyPrefix, true)) { | 804 } else if (StartsWithASCII(key, kDriveMetadataKeyPrefix, true)) { |
| 852 GURL origin; | 805 GURL origin; |
| 853 base::FilePath path; | 806 base::FilePath path; |
| 854 MetadataKeyToOriginAndPath(key, &origin, &path); | 807 MetadataKeyToOriginAndPath(key, &origin, &path); |
| 855 | 808 |
| 856 DriveMetadata metadata; | 809 DriveMetadata metadata; |
| 857 bool success = metadata.ParseFromString(itr->value().ToString()); | 810 bool success = metadata.ParseFromString(itr->value().ToString()); |
| 858 DCHECK(success); | 811 DCHECK(success); |
| 859 | 812 |
| 860 success = contents->metadata_map[origin].insert( | 813 success = contents->metadata_map[origin].insert( |
| 861 std::make_pair(path, metadata)).second; | 814 std::make_pair(path, metadata)).second; |
| 862 DCHECK(success); | 815 DCHECK(success); |
| 863 } | 816 } |
| 864 } | 817 } |
| 865 | 818 |
| 866 SyncStatusCode status = GetOrigins(&contents->batch_sync_origins, | 819 SyncStatusCode status = GetOrigins(&contents->incremental_sync_origins, |
| 867 &contents->incremental_sync_origins, | |
| 868 &contents->disabled_origins); | 820 &contents->disabled_origins); |
| 869 if (status != SYNC_STATUS_OK && | 821 if (status != SYNC_STATUS_OK && |
| 870 status != SYNC_DATABASE_ERROR_NOT_FOUND) | 822 status != SYNC_DATABASE_ERROR_NOT_FOUND) |
| 871 return status; | 823 return status; |
| 872 | 824 |
| 873 status = GetSyncRootDirectory(&contents->sync_root_directory_resource_id); | 825 status = GetSyncRootDirectory(&contents->sync_root_directory_resource_id); |
| 874 if (status != SYNC_STATUS_OK && | 826 if (status != SYNC_STATUS_OK && |
| 875 status != SYNC_DATABASE_ERROR_NOT_FOUND) | 827 status != SYNC_DATABASE_ERROR_NOT_FOUND) |
| 876 return status; | 828 return status; |
| 877 | 829 |
| (...skipping 153 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 1031 SyncStatusCode DriveMetadataDB::DeleteEntry(const FileSystemURL& url) { | 983 SyncStatusCode DriveMetadataDB::DeleteEntry(const FileSystemURL& url) { |
| 1032 DCHECK(CalledOnValidThread()); | 984 DCHECK(CalledOnValidThread()); |
| 1033 DCHECK(db_.get()); | 985 DCHECK(db_.get()); |
| 1034 | 986 |
| 1035 std::string metadata_key = FileSystemURLToMetadataKey(url); | 987 std::string metadata_key = FileSystemURLToMetadataKey(url); |
| 1036 leveldb::Status status = db_->Delete( | 988 leveldb::Status status = db_->Delete( |
| 1037 leveldb::WriteOptions(), metadata_key); | 989 leveldb::WriteOptions(), metadata_key); |
| 1038 return LevelDBStatusToSyncStatusCode(status); | 990 return LevelDBStatusToSyncStatusCode(status); |
| 1039 } | 991 } |
| 1040 | 992 |
| 1041 SyncStatusCode DriveMetadataDB::UpdateOriginAsBatchSync( | |
| 1042 const GURL& origin, const std::string& resource_id) { | |
| 1043 DCHECK(CalledOnValidThread()); | |
| 1044 DCHECK(db_.get()); | |
| 1045 | |
| 1046 leveldb::Status status = db_->Put( | |
| 1047 leveldb::WriteOptions(), | |
| 1048 CreateKeyForOriginRoot(origin, BATCH_SYNC_ORIGIN), | |
| 1049 resource_id); | |
| 1050 return LevelDBStatusToSyncStatusCode(status); | |
| 1051 } | |
| 1052 | |
| 1053 SyncStatusCode DriveMetadataDB::UpdateOriginAsIncrementalSync( | 993 SyncStatusCode DriveMetadataDB::UpdateOriginAsIncrementalSync( |
| 1054 const GURL& origin, const std::string& resource_id) { | 994 const GURL& origin, const std::string& resource_id) { |
| 1055 DCHECK(CalledOnValidThread()); | 995 DCHECK(CalledOnValidThread()); |
| 1056 DCHECK(db_.get()); | 996 DCHECK(db_.get()); |
| 1057 | 997 |
| 1058 leveldb::WriteBatch batch; | 998 leveldb::WriteBatch batch; |
| 1059 batch.Delete(CreateKeyForOriginRoot(origin, BATCH_SYNC_ORIGIN)); | |
| 1060 batch.Delete(CreateKeyForOriginRoot(origin, DISABLED_ORIGIN)); | 999 batch.Delete(CreateKeyForOriginRoot(origin, DISABLED_ORIGIN)); |
| 1061 batch.Put(CreateKeyForOriginRoot(origin, INCREMENTAL_SYNC_ORIGIN), | 1000 batch.Put(CreateKeyForOriginRoot(origin, INCREMENTAL_SYNC_ORIGIN), |
| 1062 resource_id); | 1001 resource_id); |
| 1063 leveldb::Status status = db_->Write(leveldb::WriteOptions(), &batch); | 1002 leveldb::Status status = db_->Write(leveldb::WriteOptions(), &batch); |
| 1064 | 1003 |
| 1065 return LevelDBStatusToSyncStatusCode(status); | 1004 return LevelDBStatusToSyncStatusCode(status); |
| 1066 } | 1005 } |
| 1067 | 1006 |
| 1068 SyncStatusCode DriveMetadataDB::EnableOrigin( | 1007 SyncStatusCode DriveMetadataDB::EnableOrigin( |
| 1069 const GURL& origin, const std::string& resource_id) { | 1008 const GURL& origin, const std::string& resource_id) { |
| 1070 DCHECK(CalledOnValidThread()); | 1009 DCHECK(CalledOnValidThread()); |
| 1071 DCHECK(db_.get()); | 1010 DCHECK(db_.get()); |
| 1072 | 1011 |
| 1012 // No DB entry as enabled origins go back to | |
| 1013 // DriveFileSyncService.pending_batch_sync_origins only. | |
| 1073 leveldb::WriteBatch batch; | 1014 leveldb::WriteBatch batch; |
| 1074 batch.Delete(CreateKeyForOriginRoot(origin, INCREMENTAL_SYNC_ORIGIN)); | 1015 batch.Delete(CreateKeyForOriginRoot(origin, INCREMENTAL_SYNC_ORIGIN)); |
| 1075 batch.Delete(CreateKeyForOriginRoot(origin, DISABLED_ORIGIN)); | 1016 batch.Delete(CreateKeyForOriginRoot(origin, DISABLED_ORIGIN)); |
| 1076 batch.Put(CreateKeyForOriginRoot(origin, BATCH_SYNC_ORIGIN), resource_id); | |
| 1077 leveldb::Status status = db_->Write(leveldb::WriteOptions(), &batch); | 1017 leveldb::Status status = db_->Write(leveldb::WriteOptions(), &batch); |
| 1078 | 1018 |
| 1079 return LevelDBStatusToSyncStatusCode(status); | 1019 return LevelDBStatusToSyncStatusCode(status); |
| 1080 } | 1020 } |
| 1081 | 1021 |
| 1082 SyncStatusCode DriveMetadataDB::DisableOrigin( | 1022 SyncStatusCode DriveMetadataDB::DisableOrigin( |
| 1083 const GURL& origin_to_disable, const std::string& resource_id) { | 1023 const GURL& origin_to_disable, const std::string& resource_id) { |
| 1084 DCHECK(CalledOnValidThread()); | 1024 DCHECK(CalledOnValidThread()); |
| 1085 DCHECK(db_.get()); | 1025 DCHECK(db_.get()); |
| 1086 | 1026 |
| 1087 leveldb::WriteBatch batch; | 1027 leveldb::WriteBatch batch; |
| 1088 batch.Delete(CreateKeyForOriginRoot(origin_to_disable, BATCH_SYNC_ORIGIN)); | |
| 1089 batch.Delete(CreateKeyForOriginRoot(origin_to_disable, | 1028 batch.Delete(CreateKeyForOriginRoot(origin_to_disable, |
| 1090 INCREMENTAL_SYNC_ORIGIN)); | 1029 INCREMENTAL_SYNC_ORIGIN)); |
| 1091 batch.Put(CreateKeyForOriginRoot(origin_to_disable, DISABLED_ORIGIN), | 1030 batch.Put(CreateKeyForOriginRoot(origin_to_disable, DISABLED_ORIGIN), |
| 1092 resource_id); | 1031 resource_id); |
| 1093 | 1032 |
| 1094 // Remove entries specified by |origin_to_disable|. | 1033 // Remove entries specified by |origin_to_disable|. |
| 1095 scoped_ptr<leveldb::Iterator> itr(db_->NewIterator(leveldb::ReadOptions())); | 1034 scoped_ptr<leveldb::Iterator> itr(db_->NewIterator(leveldb::ReadOptions())); |
| 1096 std::string metadata_key = kDriveMetadataKeyPrefix + origin_to_disable.spec(); | 1035 std::string metadata_key = kDriveMetadataKeyPrefix + origin_to_disable.spec(); |
| 1097 for (itr->Seek(metadata_key); itr->Valid(); itr->Next()) { | 1036 for (itr->Seek(metadata_key); itr->Valid(); itr->Next()) { |
| 1098 std::string key = itr->key().ToString(); | 1037 std::string key = itr->key().ToString(); |
| 1099 if (!StartsWithASCII(key, kDriveMetadataKeyPrefix, true)) | 1038 if (!StartsWithASCII(key, kDriveMetadataKeyPrefix, true)) |
| 1100 break; | 1039 break; |
| 1101 GURL origin; | 1040 GURL origin; |
| 1102 base::FilePath path; | 1041 base::FilePath path; |
| 1103 MetadataKeyToOriginAndPath(key, &origin, &path); | 1042 MetadataKeyToOriginAndPath(key, &origin, &path); |
| 1104 if (origin != origin_to_disable) | 1043 if (origin != origin_to_disable) |
| 1105 break; | 1044 break; |
| 1106 batch.Delete(key); | 1045 batch.Delete(key); |
| 1107 } | 1046 } |
| 1108 leveldb::Status status = db_->Write(leveldb::WriteOptions(), &batch); | 1047 leveldb::Status status = db_->Write(leveldb::WriteOptions(), &batch); |
| 1109 | 1048 |
| 1110 return LevelDBStatusToSyncStatusCode(status); | 1049 return LevelDBStatusToSyncStatusCode(status); |
| 1111 } | 1050 } |
| 1112 | 1051 |
| 1113 SyncStatusCode DriveMetadataDB::RemoveOrigin(const GURL& origin_to_remove) { | 1052 SyncStatusCode DriveMetadataDB::RemoveOrigin(const GURL& origin_to_remove) { |
| 1114 DCHECK(CalledOnValidThread()); | 1053 DCHECK(CalledOnValidThread()); |
| 1115 | 1054 |
| 1116 leveldb::WriteBatch batch; | 1055 leveldb::WriteBatch batch; |
| 1117 batch.Delete(CreateKeyForOriginRoot(origin_to_remove, BATCH_SYNC_ORIGIN)); | |
| 1118 batch.Delete(CreateKeyForOriginRoot(origin_to_remove, | 1056 batch.Delete(CreateKeyForOriginRoot(origin_to_remove, |
| 1119 INCREMENTAL_SYNC_ORIGIN)); | 1057 INCREMENTAL_SYNC_ORIGIN)); |
| 1120 batch.Delete(CreateKeyForOriginRoot(origin_to_remove, DISABLED_ORIGIN)); | 1058 batch.Delete(CreateKeyForOriginRoot(origin_to_remove, DISABLED_ORIGIN)); |
| 1121 | 1059 |
| 1122 scoped_ptr<leveldb::Iterator> itr(db_->NewIterator(leveldb::ReadOptions())); | 1060 scoped_ptr<leveldb::Iterator> itr(db_->NewIterator(leveldb::ReadOptions())); |
| 1123 std::string metadata_key = kDriveMetadataKeyPrefix + origin_to_remove.spec(); | 1061 std::string metadata_key = kDriveMetadataKeyPrefix + origin_to_remove.spec(); |
| 1124 for (itr->Seek(metadata_key); itr->Valid(); itr->Next()) { | 1062 for (itr->Seek(metadata_key); itr->Valid(); itr->Next()) { |
| 1125 std::string key = itr->key().ToString(); | 1063 std::string key = itr->key().ToString(); |
| 1126 if (!StartsWithASCII(key, kDriveMetadataKeyPrefix, true)) | 1064 if (!StartsWithASCII(key, kDriveMetadataKeyPrefix, true)) |
| 1127 break; | 1065 break; |
| 1128 GURL origin; | 1066 GURL origin; |
| 1129 base::FilePath path; | 1067 base::FilePath path; |
| 1130 MetadataKeyToOriginAndPath(key, &origin, &path); | 1068 MetadataKeyToOriginAndPath(key, &origin, &path); |
| 1131 if (origin != origin_to_remove) | 1069 if (origin != origin_to_remove) |
| 1132 break; | 1070 break; |
| 1133 batch.Delete(key); | 1071 batch.Delete(key); |
| 1134 } | 1072 } |
| 1135 | 1073 |
| 1136 leveldb::Status status = db_->Write(leveldb::WriteOptions(), &batch); | 1074 leveldb::Status status = db_->Write(leveldb::WriteOptions(), &batch); |
| 1137 return LevelDBStatusToSyncStatusCode(status); | 1075 return LevelDBStatusToSyncStatusCode(status); |
| 1138 } | 1076 } |
| 1139 | 1077 |
| 1140 SyncStatusCode DriveMetadataDB::GetOrigins( | 1078 SyncStatusCode DriveMetadataDB::GetOrigins( |
| 1141 ResourceIdByOrigin* batch_sync_origins, | |
| 1142 ResourceIdByOrigin* incremental_sync_origins, | 1079 ResourceIdByOrigin* incremental_sync_origins, |
| 1143 ResourceIdByOrigin* disabled_origins) { | 1080 ResourceIdByOrigin* disabled_origins) { |
| 1144 DCHECK(CalledOnValidThread()); | 1081 DCHECK(CalledOnValidThread()); |
| 1145 DCHECK(db_.get()); | 1082 DCHECK(db_.get()); |
| 1146 | 1083 |
| 1147 scoped_ptr<leveldb::Iterator> itr(db_->NewIterator(leveldb::ReadOptions())); | 1084 scoped_ptr<leveldb::Iterator> itr(db_->NewIterator(leveldb::ReadOptions())); |
| 1148 | 1085 |
| 1149 // Get batch sync origins from the DB. | |
| 1150 for (itr->Seek(kDriveBatchSyncOriginKeyPrefix); | |
| 1151 itr->Valid(); itr->Next()) { | |
| 1152 std::string key = itr->key().ToString(); | |
| 1153 if (!StartsWithASCII(key, kDriveBatchSyncOriginKeyPrefix, true)) | |
| 1154 break; | |
| 1155 GURL origin(std::string( | |
| 1156 key.begin() + arraysize(kDriveBatchSyncOriginKeyPrefix) - 1, | |
| 1157 key.end())); | |
| 1158 DCHECK(origin.is_valid()); | |
| 1159 bool result = batch_sync_origins->insert( | |
| 1160 std::make_pair(origin, itr->value().ToString())).second; | |
| 1161 DCHECK(result); | |
| 1162 } | |
| 1163 | |
| 1164 // Get incremental sync origins from the DB. | 1086 // Get incremental sync origins from the DB. |
| 1165 for (itr->Seek(kDriveIncrementalSyncOriginKeyPrefix); | 1087 for (itr->Seek(kDriveIncrementalSyncOriginKeyPrefix); |
| 1166 itr->Valid(); itr->Next()) { | 1088 itr->Valid(); itr->Next()) { |
| 1167 std::string key = itr->key().ToString(); | 1089 std::string key = itr->key().ToString(); |
| 1168 if (!StartsWithASCII(key, kDriveIncrementalSyncOriginKeyPrefix, true)) | 1090 if (!StartsWithASCII(key, kDriveIncrementalSyncOriginKeyPrefix, true)) |
| 1169 break; | 1091 break; |
| 1170 GURL origin(std::string( | 1092 GURL origin(std::string( |
| 1171 key.begin() + arraysize(kDriveIncrementalSyncOriginKeyPrefix) - 1, | 1093 key.begin() + arraysize(kDriveIncrementalSyncOriginKeyPrefix) - 1, |
| 1172 key.end())); | 1094 key.end())); |
| 1173 DCHECK(origin.is_valid()); | 1095 DCHECK(origin.is_valid()); |
| (...skipping 14 matching lines...) Expand all Loading... | |
| 1188 DCHECK(origin.is_valid()); | 1110 DCHECK(origin.is_valid()); |
| 1189 bool result = disabled_origins->insert( | 1111 bool result = disabled_origins->insert( |
| 1190 std::make_pair(origin, itr->value().ToString())).second; | 1112 std::make_pair(origin, itr->value().ToString())).second; |
| 1191 DCHECK(result); | 1113 DCHECK(result); |
| 1192 } | 1114 } |
| 1193 | 1115 |
| 1194 return SYNC_STATUS_OK; | 1116 return SYNC_STATUS_OK; |
| 1195 } | 1117 } |
| 1196 | 1118 |
| 1197 } // namespace sync_file_system | 1119 } // namespace sync_file_system |
| OLD | NEW |