Reading List create protobuf store

ios/chrome/browser/reading_list/reading_list_store.cc

+// Copyright 2016 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 "ios/chrome/browser/reading_list/reading_list_store.h"
+
+#include "base/bind.h"
+#include "base/files/file_path.h"
+#include "base/logging.h"
+#include "base/memory/ptr_util.h"
+#include "ios/chrome/browser/reading_list/proto/reading_list.pb.h"
+#include "ios/chrome/browser/reading_list/reading_list_model_impl.h"
+#include "ios/web/public/web_thread.h"
+
+ReadingListStore::ReadingListStore(std::unique_ptr<ReadingListDB> database,
+                                   const base::FilePath& database_dir)
+    : database_(std::move(database)),
+      database_loaded_(false),
+      pending_transaction_(0),
+      weak_ptr_factory_(this) {
+  database_->Init("ReadingList", database_dir,
+                  base::Bind(&ReadingListStore::OnDatabaseInit,
+                             weak_ptr_factory_.GetWeakPtr()));
+}
+
+ReadingListStore::~ReadingListStore() {
+  DCHECK(pending_transaction_ == 0);
+}
+
+void ReadingListStore::OnDatabaseInit(bool success) {
+  DCHECK_CURRENTLY_ON(web::WebThread::UI);
+  if (!success) {
+    database_.reset();
+  }
+}
+
+void ReadingListStore::SetReadingListModel(ReadingListModelImpl* model) {
+  DCHECK_CURRENTLY_ON(web::WebThread::UI);
+  model_ = model;
+}
+
+void ReadingListStore::LoadPersistentLists() {
+  DCHECK_CURRENTLY_ON(web::WebThread::UI);
+  DCHECK(model_);
+  database_->LoadEntries(base::Bind(&ReadingListStore::OnDatabaseLoad,
+                                    weak_ptr_factory_.GetWeakPtr()));
+}
+
+void ReadingListStore::BeginTransaction() {
+  DCHECK_CURRENTLY_ON(web::WebThread::UI);
+  pending_transaction_++;
+  if (pending_transaction_ == 1) {
+    pending_keys_to_save_ = base::MakeUnique<ReadingListDB::KeyEntryVector>();
+    pending_keys_to_remove_ = base::MakeUnique<std::vector<std::string>>();
+  }
+}
+
+void ReadingListStore::CommitTransaction() {
+  DCHECK_CURRENTLY_ON(web::WebThread::UI);
+  pending_transaction_--;
+  if (pending_transaction_ == 0) {
+    database_->UpdateEntries(std::move(pending_keys_to_save_),
+                             std::move(pending_keys_to_remove_),
+                             base::Bind(&ReadingListStore::OnDatabaseSave,
+                                        weak_ptr_factory_.GetWeakPtr()));
+    pending_keys_to_save_ = nullptr;
+    pending_keys_to_remove_ = nullptr;
+  }
+}
+
+void ReadingListStore::SaveEntry(const ReadingListEntry& entry, bool read) {
+  DCHECK_CURRENTLY_ON(web::WebThread::UI);
+  BeginTransaction();
+
+  std::unique_ptr<reading_list::ReadingListLocal> pb_entry =
+      entry.AsReadingListLocal(read);
+  // Unref the URL before making asynchronous call.
+  std::string local_key = entry.URL().spec();
+  pending_keys_to_save_->push_back(std::make_pair(local_key, *pb_entry));
+
+  CommitTransaction();
+}
+
+void ReadingListStore::RemoveEntry(const ReadingListEntry& entry) {
+  DCHECK_CURRENTLY_ON(web::WebThread::UI);
+  BeginTransaction();
+  pending_keys_to_remove_->push_back(entry.URL().spec());
+  CommitTransaction();
+}
+
+void ReadingListStore::OnDatabaseLoad(bool success,
+                                      std::unique_ptr<EntryVector> entries) {
+  DCHECK_CURRENTLY_ON(web::WebThread::UI);
+  if (!success) {
+    database_.reset();
+    return;
+  }
+  database_loaded_ = true;
+  auto read = base::MakeUnique<ReadingListEntries>();
+  auto unread = base::MakeUnique<ReadingListEntries>();
+
+  for (const reading_list::ReadingListLocal& pb_entry : *entries) {
+    std::unique_ptr<ReadingListEntry> entry(
+        ReadingListEntry::FromReadingListLocal(pb_entry));
+    if (!entry) {
+      continue;
+    }
+    if (pb_entry.entry().status() == sync_pb::ReadingListSpecifics::READ) {
+      read->push_back(std::move(*entry));
+    } else {
+      unread->push_back(std::move(*entry));
+    }
+  }
+  std::sort(read->begin(), read->end(),
+            ReadingListEntry::CompareEntryUpdateTime);
+  std::sort(unread->begin(), unread->end(),
+            ReadingListEntry::CompareEntryUpdateTime);
+
+  model_->ModelLoaded(std::move(unread), std::move(read));
+}
+
+void ReadingListStore::OnDatabaseSave(bool success) {
+  DCHECK_CURRENTLY_ON(web::WebThread::UI);
+  if (!success) {
+    database_.reset();
+    database_loaded_ = false;
+  }
+}