YugabyteDB (2.13.0.0-b42, bfc6a6643e7399ac8a0e81d06a3ee6d6571b33ab)

//  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.

//  This source code is licensed under the BSD-style license found in the

//  LICENSE file in the root directory of this source tree. An additional grant

//  of patent rights can be found in the PATENTS file in the same directory.

//

// The following only applies to changes made to this file as part of YugaByte development.

//

// Portions Copyright (c) YugaByte, Inc.

//

// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except

// in compliance with the License.  You may obtain a copy of the License at

//

// http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing, software distributed under the License

// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express

// or implied.  See the License for the specific language governing permissions and limitations

// under the License.

//

#ifndef ROCKSDB_LITE

#include "yb/rocksdb/utilities/transactions/optimistic_transaction_db_impl.h"

#include <string>

#include <vector>

#include "yb/rocksdb/db.h"

#include "yb/rocksdb/options.h"

#include "yb/rocksdb/utilities/optimistic_transaction_db.h"

#include "yb/rocksdb/utilities/transactions/optimistic_transaction_impl.h"

namespace rocksdb {

Transaction* OptimisticTransactionDBImpl::BeginTransaction(

    const WriteOptions& write_options,

    const OptimisticTransactionOptions& txn_options, Transaction* old_txn) {

  if (old_txn != nullptr) {

    ReinitializeTransaction(old_txn, write_options, txn_options);

    return old_txn;

  } else {

    return new OptimisticTransactionImpl(this, write_options, txn_options);

  }

}

Status OptimisticTransactionDB::Open(const Options& options,

                                     const std::string& dbname,

                                     OptimisticTransactionDB** dbptr) {

  DBOptions db_options(options);

  ColumnFamilyOptions cf_options(options);

  std::vector<ColumnFamilyDescriptor> column_families;

  column_families.push_back(

      ColumnFamilyDescriptor(kDefaultColumnFamilyName, cf_options));

  std::vector<ColumnFamilyHandle*> handles;

  Status s = Open(db_options, dbname, column_families, &handles, dbptr);

  if (s.ok()) {

    assert(handles.size() == 1);

    // i can delete the handle since DBImpl is always holding a reference to

    // default column family

    delete handles[0];

  }

  return s;

}

Status OptimisticTransactionDB::Open(

    const DBOptions& db_options, const std::string& dbname,

    const std::vector<ColumnFamilyDescriptor>& column_families,

    std::vector<ColumnFamilyHandle*>* handles,

    OptimisticTransactionDB** dbptr) {

  Status s;

  DB* db;

  std::vector<ColumnFamilyDescriptor> column_families_copy = column_families;

  // Enable MemTable History if not already enabled

  for (auto& column_family : column_families_copy) {

    ColumnFamilyOptions* options = &column_family.options;

    if (options->max_write_buffer_number_to_maintain == 0) {

      // Setting to -1 will set the History size to max_write_buffer_number.

      options->max_write_buffer_number_to_maintain = -1;

    }

  }

  s = DB::Open(db_options, dbname, column_families_copy, handles, &db);

  if (s.ok()) {

    *dbptr = new OptimisticTransactionDBImpl(db);

  }

  return s;

}

void OptimisticTransactionDBImpl::ReinitializeTransaction(

    Transaction* txn, const WriteOptions& write_options,

    const OptimisticTransactionOptions& txn_options) {

  assert(dynamic_cast<OptimisticTransactionImpl*>(txn) != nullptr);

  auto txn_impl = reinterpret_cast<OptimisticTransactionImpl*>(txn);

  txn_impl->Reinitialize(this, write_options, txn_options);

}

}  //  namespace rocksdb

#endif  // ROCKSDB_LITE