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