Add best-effort/allow rollback flags on WorkItem.

chrome/installer/util/work_item.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.h"
 
 #include <windows.h>
 
+#include "base/logging.h"
 #include "chrome/installer/util/callback_work_item.h"
 #include "chrome/installer/util/conditional_work_item_list.h"
 #include "chrome/installer/util/copy_tree_work_item.h"
 #include "chrome/installer/util/create_dir_work_item.h"
 #include "chrome/installer/util/create_reg_key_work_item.h"
 #include "chrome/installer/util/delete_reg_key_work_item.h"
 #include "chrome/installer/util/delete_reg_value_work_item.h"
 #include "chrome/installer/util/delete_tree_work_item.h"
 #include "chrome/installer/util/move_tree_work_item.h"
 #include "chrome/installer/util/self_reg_work_item.h"
 #include "chrome/installer/util/set_reg_value_work_item.h"
 #include "chrome/installer/util/work_item_list.h"
 
-WorkItem::WorkItem() : ignore_failure_(false) {
-}
-
-WorkItem::~WorkItem() {
-}
+WorkItem::WorkItem() = default;
+WorkItem::~WorkItem() = default;
 
 CallbackWorkItem* WorkItem::CreateCallbackWorkItem(
     base::Callback<bool(const CallbackWorkItem&)> callback) {
   return new CallbackWorkItem(callback);
 }
 
 CopyTreeWorkItem* WorkItem::CreateCopyTreeWorkItem(
     const base::FilePath& source_path,
     const base::FilePath& dest_path,
     const base::FilePath& temp_dir,
@@ skipping 118 matching lines @@
 }
 
 // static
 WorkItemList* WorkItem::CreateNoRollbackWorkItemList() {
   return new NoRollbackWorkItemList();
 }
 
 WorkItemList* WorkItem::CreateConditionalWorkItemList(Condition* condition) {
   return new ConditionalWorkItemList(condition);
 }
+
+bool WorkItem::Do() {
+  DCHECK_EQ(BEFORE_DO, state_);
+  const bool success = DoImpl();
+  state_ = AFTER_DO;
+  return best_effort() ? true : success;
+}
+
+void WorkItem::Rollback() {
+  DCHECK_EQ(AFTER_DO, state_);
+  if (rollback_enabled())
+    RollbackImpl();
+  state_ = AFTER_ROLLBACK;
+}
+
+void WorkItem::set_best_effort(bool best_effort) {
+  DCHECK_EQ(BEFORE_DO, state());
+  best_effort_ = best_effort;
+}
+
+void WorkItem::set_rollback_enabled(bool rollback_enabled) {
+  DCHECK_EQ(BEFORE_DO, state());
+  rollback_enabled_ = rollback_enabled;
+}