Changes to fit better with the needs of the download manager.

components/offline_pages/background/request_queue_store_sql.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 "components/offline_pages/background/request_queue_store_sql.h"
 
 #include "base/bind.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/location.h"
@@ skipping 48 matching lines @@
 }
 
 bool DeleteRequestById(sql::Connection* db, int64_t request_id) {
   const char kSql[] =
       "DELETE FROM " REQUEST_QUEUE_TABLE_NAME " WHERE request_id=?";
   sql::Statement statement(db->GetCachedStatement(SQL_FROM_HERE, kSql));
   statement.BindInt64(0, request_id);
   return statement.Run();
 }
 
-bool DeleteRequestByClientId(sql::Connection* db, const ClientId& client_id) {
-  const char kSql[] = "DELETE FROM " REQUEST_QUEUE_TABLE_NAME
-                      " WHERE client_namespace=? AND client_id=?";
-  sql::Statement statement(db->GetCachedStatement(SQL_FROM_HERE, kSql));
-  statement.BindString(0, client_id.name_space);
-  statement.BindString(1, client_id.id);
-  DVLOG(2) << kSql << " client_namespace " << client_id.name_space
-           << " client_id " << client_id.id;
-  return statement.Run();
-}
-
-bool DeleteRequestsByIds(sql::Connection* db,
-                         const std::vector<int64_t>& request_ids,
-                         int* count) {
-  DCHECK(count);
+bool DeleteRequestsByIds(
+    sql::Connection* db,
+    const std::vector<int64_t>& request_ids,
+    std::vector<RequestQueue::UpdateRequestResult>& results) {
   // If you create a transaction but don't Commit() it is automatically
   // rolled back by its destructor when it falls out of scope.
   sql::Transaction transaction(db);
-  if (!transaction.Begin())
+  if (!transaction.Begin()) {
+    for (size_t i = 0; i < request_ids.size(); ++i)
+      results.push_back(RequestQueue::UpdateRequestResult::STORE_FAILURE);
     return false;
-
-  *count = 0;
-  for (auto request_id : request_ids) {
-    if (!DeleteRequestById(db, request_id))
-      return false;
-    *count += db->GetLastChangeCount();
   }
 
-  if (!transaction.Commit())
+  for (auto request_id : request_ids) {
+    if (DeleteRequestById(db, request_id))
+      results.push_back(RequestQueue::UpdateRequestResult::SUCCESS);
+    else
+      results.push_back(
+          RequestQueue::UpdateRequestResult::REQUEST_DOES_NOT_EXIST);
+  }
+
+  if (!transaction.Commit()) {
+    results.clear();
+    for (size_t i = 0; i < request_ids.size(); ++i)
+      results.push_back(RequestQueue::UpdateRequestResult::STORE_FAILURE);
     return false;
+  }
 
   return true;
 }
 
-bool DeleteRequestsByClientIds(sql::Connection* db,
-                               const std::vector<ClientId>& client_ids,
-                               int* count) {
-  DCHECK(count);
-  // If you create a transaction but don't Commit() it is automatically
-  // rolled back by its destructor when it falls out of scope.
-  sql::Transaction transaction(db);
-  if (!transaction.Begin())
-    return false;
-
-  *count = 0;
-  for (const auto& client_id : client_ids) {
-    if (!DeleteRequestByClientId(db, client_id))
-      return false;
-    *count += db->GetLastChangeCount();
-  }
-
-  if (!transaction.Commit())
-    return false;
-
-  return true;
-}
-
 // Create a save page request from a SQL result.  Expects complete rows with
 // all columns present.  Columns are in order they are defined in select query
 // in |RequestQueueStore::RequestSync| method.
 SavePageRequest MakeSavePageRequest(const sql::Statement& statement) {
   const int64_t id = statement.ColumnInt64(0);
   const base::Time creation_time =
       base::Time::FromInternalValue(statement.ColumnInt64(1));
   const base::Time activation_time =
       base::Time::FromInternalValue(statement.ColumnInt64(2));
   const base::Time last_attempt_time =
@@ skipping 124 matching lines @@
   RequestQueueStore::UpdateStatus status = InsertOrReplace(db, request);
   runner->PostTask(FROM_HERE, base::Bind(callback, status));
 }
 
 // static
 void RequestQueueStoreSQL::RemoveRequestsSync(
     sql::Connection* db,
     scoped_refptr<base::SingleThreadTaskRunner> runner,
     const std::vector<int64_t>& request_ids,
     const RemoveCallback& callback) {
+  std::vector<RequestQueue::UpdateRequestResult> results;
   // TODO(fgorski): add UMA metrics here.
-  int count = 0;
-  if (DeleteRequestsByIds(db, request_ids, &count))
-    runner->PostTask(FROM_HERE, base::Bind(callback, true, count));
-  else
-    runner->PostTask(FROM_HERE, base::Bind(callback, false, 0));
+  DeleteRequestsByIds(db, request_ids, results);
+  runner->PostTask(FROM_HERE, base::Bind(callback, request_ids, results));
 }
 
 // static
-void RequestQueueStoreSQL::RemoveRequestsByClientIdSync(
-    sql::Connection* db,
-    scoped_refptr<base::SingleThreadTaskRunner> runner,
-    const std::vector<ClientId>& client_ids,
-    const RemoveCallback& callback) {
-  // TODO(fgorski): add UMA metrics here.
-  int count = 0;
-  if (DeleteRequestsByClientIds(db, client_ids, &count))
-    runner->PostTask(FROM_HERE, base::Bind(callback, true, count));
-  else
-    runner->PostTask(FROM_HERE, base::Bind(callback, false, 0));
-}
-
-// static
 void RequestQueueStoreSQL::ResetSync(
     sql::Connection* db,
     const base::FilePath& db_file_path,
     scoped_refptr<base::SingleThreadTaskRunner> runner,
     const ResetCallback& callback) {
   // This method deletes the content of the whole store and reinitializes it.
   bool success = db->Raze();
   db->Close();
   if (success)
     success = InitDatabase(db, db_file_path);
@@ skipping 25 matching lines @@
         FROM_HERE, base::Bind(callback, UpdateStatus::FAILED));
     return;
   }
 
   background_task_runner_->PostTask(
       FROM_HERE,
       base::Bind(&RequestQueueStoreSQL::AddOrUpdateRequestSync, db_.get(),
                  base::ThreadTaskRunnerHandle::Get(), request, callback));
 }
 
-// TODO(petewil): This is unused, since request_coordinator doesn't keep
-// request_ids, and neither do clients.  Plan is to remove this API in a future
-// changelist.  If we do discover a need to keep it, name it
 // RemoveRequestsByRequestId to be more parallell with RemoveRequestsByClientId.
 void RequestQueueStoreSQL::RemoveRequests(
     const std::vector<int64_t>& request_ids,
     const RemoveCallback& callback) {
   DCHECK(db_.get());
   if (!db_.get()) {
+    std::vector<RequestQueue::UpdateRequestResult> statuses;
     // Nothing to do, but post a callback instead of calling directly
     // to preserve the async style behavior to prevent bugs.
     base::ThreadTaskRunnerHandle::Get()->PostTask(
-        FROM_HERE, base::Bind(callback, false, 0));
+        FROM_HERE, base::Bind(callback, request_ids, statuses));
     return;
   }
 
   background_task_runner_->PostTask(
       FROM_HERE,
       base::Bind(&RequestQueueStoreSQL::RemoveRequestsSync, db_.get(),
                  base::ThreadTaskRunnerHandle::Get(), request_ids, callback));
 }
 
-void RequestQueueStoreSQL::RemoveRequestsByClientId(
-    const std::vector<ClientId>& client_ids,
-    const RemoveCallback& callback) {
-  DCHECK(db_.get());
-  if (!db_.get()) {
-    // Nothing to do, but post a callback instead of calling directly
-    // to preserve the async style behavior to prevent bugs.
-    base::ThreadTaskRunnerHandle::Get()->PostTask(
-        FROM_HERE, base::Bind(callback, false, 0));
-    return;
-  }
-
-  background_task_runner_->PostTask(
-      FROM_HERE,
-      base::Bind(&RequestQueueStoreSQL::RemoveRequestsByClientIdSync, db_.get(),
-                 base::ThreadTaskRunnerHandle::Get(), client_ids, callback));
-}
-
 void RequestQueueStoreSQL::Reset(const ResetCallback& callback) {
   DCHECK(db_.get());
   if (!db_.get()) {
     // Nothing to do, but post a callback instead of calling directly
     // to preserve the async style behavior to prevent bugs.
     base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
                                                   base::Bind(callback, false));
     return;
   }
 
@@ skipping 25 matching lines @@
 }
 
 void RequestQueueStoreSQL::OnResetDone(const ResetCallback& callback,
                                        bool success) {
   // Complete connection initialization post reset.
   OnOpenConnectionDone(success);
   callback.Run(success);
 }
 
 }  // namespace offline_pages