Add best-effort/allow rollback flags on WorkItem.

chrome/installer/util/work_item_list.cc

 // Copyright (c) 2011 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 "chrome/installer/util/work_item_list.h"
 
 #include "base/files/file_path.h"
 #include "base/logging.h"
 #include "chrome/installer/util/callback_work_item.h"
 #include "chrome/installer/util/copy_tree_work_item.h"
@@ skipping 10 matching lines @@
 WorkItemList::~WorkItemList() {
   for (WorkItemIterator itr = list_.begin(); itr != list_.end(); ++itr) {
     delete (*itr);
   }
   for (WorkItemIterator itr = executed_list_.begin();
        itr != executed_list_.end(); ++itr) {
     delete (*itr);
   }
 }
 
-WorkItemList::WorkItemList()
-    : status_(ADD_ITEM) {
-}
+WorkItemList::WorkItemList() = default;
 
-bool WorkItemList::Do() {
-  if (status_ != ADD_ITEM)
-    return false;
-
+bool WorkItemList::DoImpl() {
   bool result = true;
   while (!list_.empty()) {
     WorkItem* work_item = list_.front();
     list_.pop_front();
     executed_list_.push_front(work_item);
+    work_item->set_best_effort(best_effort());

[grt (UTC plus 2), 2016/05/06 18:39:13]

hmm. one flaw with this is that you can't add a best-effort item to a regular
work item list. maybe this isn't the right thing after all. :-(

[fdoray, 2016/05/10 18:35:32]

Now I only make the work items best-effort if the list is best-effort.

+    work_item->set_rollback_enabled(rollback_enabled());
     if (!work_item->Do()) {
       LOG(ERROR) << "item execution failed " << work_item->log_message();
       result = false;
       break;
     }
   }
 
   if (result)
     VLOG(1) << "list execution succeeded";
 
-  status_ = LIST_EXECUTED;
   return result;
 }
 
-void WorkItemList::Rollback() {
-  if (status_ != LIST_EXECUTED)
-    return;
-
+void WorkItemList::RollbackImpl() {
   for (WorkItemIterator itr = executed_list_.begin();
        itr != executed_list_.end(); ++itr) {
     (*itr)->Rollback();
   }
-
-  status_ = LIST_ROLLED_BACK;
-  return;
 }
 
 void WorkItemList::AddWorkItem(WorkItem* work_item) {
-  DCHECK(status_ == ADD_ITEM);
+  DCHECK_EQ(WorkItem::BEFORE_DO, state());

[grt (UTC plus 2), 2016/05/06 18:39:13]

nit: no need for WorkItem:: here

[fdoray, 2016/05/10 18:35:31]

Done.

   list_.push_back(work_item);
 }
 
 WorkItem* WorkItemList::AddCallbackWorkItem(
     base::Callback<bool(const CallbackWorkItem&)> callback) {
   WorkItem* item = WorkItem::CreateCallbackWorkItem(callback);
   AddWorkItem(item);
   return item;
 }
 
@@ skipping 146 matching lines @@
   WorkItem* item = WorkItem::CreateSelfRegWorkItem(dll_path, do_register,
                                                    user_level_registration);
   AddWorkItem(item);
   return item;
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 NoRollbackWorkItemList::~NoRollbackWorkItemList() {
 }
 
-bool NoRollbackWorkItemList::Do() {
-  if (status_ != ADD_ITEM)
-    return false;
-
+bool NoRollbackWorkItemList::DoImpl() {
   bool result = true;
   while (!list_.empty()) {
     WorkItem* work_item = list_.front();
     list_.pop_front();
     executed_list_.push_front(work_item);
-    work_item->set_ignore_failure(true);
+    work_item->set_rollback_enabled(false);
     if (!work_item->Do()) {
       LOG(ERROR) << "NoRollbackWorkItemList: item execution failed "
                  << work_item->log_message();
       result = false;
     }
   }
 
   if (result)
     VLOG(1) << "NoRollbackWorkItemList: list execution succeeded";
 
-  status_ = LIST_EXECUTED;
   return result;
 }
 
-void NoRollbackWorkItemList::Rollback() {
+void NoRollbackWorkItemList::RollbackImpl() {
   // Ignore rollback.
 }