YugabyteDB (2.13.1.0-b60, 21121d69985fbf76aa6958d8f04a9bfa936293b5)

//  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.

//

#pragma once

#ifndef ROCKSDB_LITE

#include <mutex>

#include <string>

#include <unordered_map>

#include "yb/rocksdb/db.h"

#include "yb/rocksdb/options.h"

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

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

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

namespace rocksdb {

class TransactionDBImpl : public TransactionDB {

 public:

  explicit TransactionDBImpl(DB* db,

                             const TransactionDBOptions& txn_db_options);

  ~TransactionDBImpl() {}

  Transaction* BeginTransaction(const WriteOptions& write_options,

                                const TransactionOptions& txn_options,

                                Transaction* old_txn) override;

  using StackableDB::Put;

  virtual Status Put(const WriteOptions& options,

                     ColumnFamilyHandle* column_family, const Slice& key,

                     const Slice& val) override;

  using StackableDB::Delete;

  virtual Status Delete(const WriteOptions& wopts,

                        ColumnFamilyHandle* column_family,

                        const Slice& key) override;

  using StackableDB::Merge;

  virtual Status Merge(const WriteOptions& options,

                       ColumnFamilyHandle* column_family, const Slice& key,

                       const Slice& value) override;

  using StackableDB::Write;

  virtual Status Write(const WriteOptions& opts, WriteBatch* updates) override;

  using StackableDB::CreateColumnFamily;

  virtual Status CreateColumnFamily(const ColumnFamilyOptions& options,

                                    const std::string& column_family_name,

                                    ColumnFamilyHandle** handle) override;

  using StackableDB::DropColumnFamily;

  virtual Status DropColumnFamily(ColumnFamilyHandle* column_family) override;

  Status TryLock(TransactionImpl* txn, uint32_t cfh_id, const std::string& key);

  void UnLock(TransactionImpl* txn, const TransactionKeyMap* keys);

  void UnLock(TransactionImpl* txn, uint32_t cfh_id, const std::string& key);

  void AddColumnFamily(const ColumnFamilyHandle* handle);

  static TransactionDBOptions ValidateTxnDBOptions(

      const TransactionDBOptions& txn_db_options);

  const TransactionDBOptions& GetTxnDBOptions() const {

    return txn_db_options_;

  }

  void InsertExpirableTransaction(TransactionID tx_id, TransactionImpl* tx);

  void RemoveExpirableTransaction(TransactionID tx_id);

  // If transaction is no longer available, locks can be stolen

  // If transaction is available, try stealing locks directly from transaction

  // It is the caller's responsibility to ensure that the referred transaction

  // is expirable (GetExpirationTime() > 0) and that it is expired.

  bool TryStealingExpiredTransactionLocks(TransactionID tx_id);

 private:

  void ReinitializeTransaction(

      Transaction* txn, const WriteOptions& write_options,

      const TransactionOptions& txn_options = TransactionOptions());

  const TransactionDBOptions txn_db_options_;

  TransactionLockMgr lock_mgr_;

  // Must be held when adding/dropping column families.

  InstrumentedMutex column_family_mutex_;

  Transaction* BeginInternalTransaction(const WriteOptions& options);

  Status WriteHelper(WriteBatch* updates, TransactionImpl* txn_impl);

  // Used to ensure that no locks are stolen from an expirable transaction

  // that has started a commit. Only transactions with an expiration time

  // should be in this map.

  std::mutex map_mutex_;

  std::unordered_map<TransactionID, TransactionImpl*>

      expirable_transactions_map_;

};

}  //  namespace rocksdb

#endif  // ROCKSDB_LITE