YugabyteDB (2.13.0.0-b42, bfc6a6643e7399ac8a0e81d06a3ee6d6571b33ab)

//

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

//

//

#include "yb/common/transaction.h"

#include "yb/common/common.pb.h"

#include "yb/util/result.h"

#include "yb/util/tsan_util.h"

using namespace std::literals;

DEFINE_int64(transaction_rpc_timeout_ms, 5000 * yb::kTimeMultiplier,

             "Timeout used by transaction related RPCs in milliseconds.");

namespace yb {

YB_STRONGLY_TYPED_UUID_IMPL(TransactionId);

const char* kGlobalTransactionsTableName = "transactions";

const std::string kMetricsSnapshotsTableName = "metrics";

const std::string kTransactionTablePrefix = "transactions_";

TransactionStatusResult::TransactionStatusResult(TransactionStatus status_, HybridTime status_time_)

    : TransactionStatusResult(status_, status_time_, AbortedSubTransactionSet()) {}

Unexecuted instantiation: _ZN2yb23TransactionStatusResultC2ENS_17TransactionStatusENS_10HybridTimeE

_ZN2yb23TransactionStatusResultC1ENS_17TransactionStatusENS_10HybridTimeE

Line

Count

Source

36

1.25M

    : TransactionStatusResult(status_, status_time_, AbortedSubTransactionSet()) {}

TransactionStatusResult::TransactionStatusResult(

    TransactionStatus status_, HybridTime status_time_,

    AbortedSubTransactionSet aborted_subtxn_set_)

    : status(status_), status_time(status_time_), aborted_subtxn_set(aborted_subtxn_set_) {

  DCHECK(status == TransactionStatus::ABORTED || status_time.is_valid())

      << "Status: " << status << ", status_time: " << status_time;

}

Result<TransactionMetadata> TransactionMetadata::FromPB(const TransactionMetadataPB& source) {

  TransactionMetadata result;

  auto id = FullyDecodeTransactionId(source.transaction_id());

  RETURN_NOT_OK(id);

  result.transaction_id = *id;

  if (source.has_isolation()) {

    result.isolation = source.isolation();

    result.status_tablet = source.status_tablet();

    result.priority = source.priority();

    result.start_time = HybridTime(source.start_hybrid_time());

  }

  if (source.has_locality()) {

    result.locality = source.locality();

  } else {

    result.locality = TransactionLocality::GLOBAL;

  }

  return result;

}

void TransactionMetadata::ToPB(TransactionMetadataPB* dest) const {

  if (isolation != IsolationLevel::NON_TRANSACTIONAL) {

    ForceToPB(dest);

  } else {

    TransactionIdToPB(dest);

  }

}

void TransactionMetadata::TransactionIdToPB(TransactionMetadataPB* dest) const {

  dest->set_transaction_id(transaction_id.data(), transaction_id.size());

}

void TransactionMetadata::ForceToPB(TransactionMetadataPB* dest) const {

  TransactionIdToPB(dest);

  dest->set_isolation(isolation);

  dest->set_status_tablet(status_tablet);

  dest->set_priority(priority);

  dest->set_start_hybrid_time(start_time.ToUint64());

  dest->set_locality(locality);

}

bool operator==(const TransactionMetadata& lhs, const TransactionMetadata& rhs) {

  return lhs.transaction_id == rhs.transaction_id &&

         lhs.isolation == rhs.isolation &&

         lhs.status_tablet == rhs.status_tablet &&

         lhs.priority == rhs.priority &&

         lhs.start_time == rhs.start_time &&

         lhs.locality == rhs.locality;

}

std::ostream& operator<<(std::ostream& out, const TransactionMetadata& metadata) {

  return out << metadata.ToString();

}

void SubTransactionMetadata::ToPB(SubTransactionMetadataPB* dest) const {

  dest->set_subtransaction_id(subtransaction_id);

  aborted.ToPB(dest->mutable_aborted()->mutable_set());

}

Result<SubTransactionMetadata> SubTransactionMetadata::FromPB(

    const SubTransactionMetadataPB& source) {

  return SubTransactionMetadata {

    .subtransaction_id = source.has_subtransaction_id()

        ? source.subtransaction_id()

        : kMinSubTransactionId,

    .aborted = VERIFY_RESULT(AbortedSubTransactionSet::FromPB(source.aborted().set())),

  };

}

bool SubTransactionMetadata::IsDefaultState() const {

  DCHECK(subtransaction_id >= kMinSubTransactionId);

  return subtransaction_id == kMinSubTransactionId && aborted.IsEmpty();

}

std::ostream& operator<<(std::ostream& out, const SubTransactionMetadata& metadata) {

  return out << metadata.ToString();

}

MonoDelta TransactionRpcTimeout() {

  return FLAGS_transaction_rpc_timeout_ms * 1ms * kTimeMultiplier;

}

// TODO(dtxn) correct deadline should be calculated and propagated.

CoarseTimePoint TransactionRpcDeadline() {

  return CoarseMonoClock::Now() + TransactionRpcTimeout();

}

TransactionOperationContext::TransactionOperationContext()

    : transaction_id(TransactionId::Nil()), txn_status_manager(nullptr) {}

TransactionOperationContext::TransactionOperationContext(

    const TransactionId& transaction_id_, TransactionStatusManager* txn_status_manager_)

    : transaction_id(transaction_id_),

      txn_status_manager(DCHECK_NOTNULL(txn_status_manager_)) {}

TransactionOperationContext::TransactionOperationContext(

    const TransactionId& transaction_id_,

    SubTransactionMetadata&& subtransaction_,

    TransactionStatusManager* txn_status_manager_)

    : transaction_id(transaction_id_),

      subtransaction(std::move(subtransaction_)),

      txn_status_manager(DCHECK_NOTNULL(txn_status_manager_)) {}

bool TransactionOperationContext::transactional() const {

  return !transaction_id.IsNil();

}

} // namespace yb