chrome/browser/tab/tab_state_storage_updater.cc

// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/tab/tab_state_storage_updater.h"

#include <memory>
#include <utility>

#include "base/logging.h"
#include "chrome/browser/tab/storage_update_unit.h"
#include "chrome/browser/tab/tab_state_storage_database.h"

namespace tabs {

using Transaction = TabStateStorageDatabase::Transaction;

TabStateStorageUpdater::TabStateStorageUpdater(TabStateStorageDatabase* db)
    : db_(db) {}

TabStateStorageUpdater::~TabStateStorageUpdater() = default;

void TabStateStorageUpdater::Add(std::unique_ptr<StorageUpdateUnit> unit) {
  updates_.push_back(std::move(unit));
}

bool TabStateStorageUpdater::Execute() {
  std::unique_ptr<Transaction> transaction = db_->CreateTransaction();
  if (!transaction->Begin()) {
    DLOG(ERROR) << "Could not start transaction.";
    return false;
  }

  for (auto& op : updates_) {
    if (!op->Execute(db_, transaction.get())) {
      transaction->Rollback();
      return false;
    }
  }

  if (!transaction->Commit()) {
    DLOG(ERROR) << "Could not commit transaction.";
    return false;
  }

  return true;
}

}  // namespace tabs