/**
* Copyright (C) 2018 MongoDB, Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License, version 3,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see .
*
* As a special exception, the copyright holders give permission to link the
* code of portions of this program with the OpenSSL library under certain
* conditions as described in each individual source file and distribute
* linked combinations including the program with the OpenSSL library. You
* must comply with the GNU Affero General Public License in all respects for
* all of the code used other than as permitted herein. If you modify file(s)
* with this exception, you may extend this exception to your version of the
* file(s), but you are not obligated to do so. If you do not wish to do so,
* delete this exception statement from your version. If you delete this
* exception statement from all source files in the program, then also delete
* it in the license file.
*/
#include "mongo/platform/basic.h"
#include "mongo/db/server_transactions_metrics.h"
#include "mongo/db/commands/server_status.h"
#include "mongo/db/jsobj.h"
#include "mongo/db/operation_context.h"
#include "mongo/db/retryable_writes_stats.h"
#include "mongo/db/service_context.h"
#include "mongo/db/transactions_stats_gen.h"
namespace mongo {
namespace {
const auto ServerTransactionsMetricsDecoration =
ServiceContext::declareDecoration();
} // namespace
ServerTransactionsMetrics* ServerTransactionsMetrics::get(ServiceContext* service) {
return &ServerTransactionsMetricsDecoration(service);
}
ServerTransactionsMetrics* ServerTransactionsMetrics::get(OperationContext* opCtx) {
return get(opCtx->getServiceContext());
}
unsigned long long ServerTransactionsMetrics::getCurrentActive() const {
return _currentActive.load();
}
void ServerTransactionsMetrics::decrementCurrentActive() {
_currentActive.fetchAndSubtract(1);
}
void ServerTransactionsMetrics::incrementCurrentActive() {
_currentActive.fetchAndAdd(1);
}
unsigned long long ServerTransactionsMetrics::getCurrentInactive() const {
return _currentInactive.load();
}
void ServerTransactionsMetrics::decrementCurrentInactive() {
_currentInactive.fetchAndSubtract(1);
}
void ServerTransactionsMetrics::incrementCurrentInactive() {
_currentInactive.fetchAndAdd(1);
}
unsigned long long ServerTransactionsMetrics::getCurrentOpen() const {
return _currentOpen.load();
}
void ServerTransactionsMetrics::decrementCurrentOpen() {
_currentOpen.fetchAndSubtract(1);
}
void ServerTransactionsMetrics::incrementCurrentOpen() {
_currentOpen.fetchAndAdd(1);
}
unsigned long long ServerTransactionsMetrics::getTotalStarted() const {
return _totalStarted.load();
}
void ServerTransactionsMetrics::incrementTotalStarted() {
_totalStarted.fetchAndAdd(1);
}
unsigned long long ServerTransactionsMetrics::getTotalAborted() const {
return _totalAborted.load();
}
void ServerTransactionsMetrics::incrementTotalAborted() {
_totalAborted.fetchAndAdd(1);
}
unsigned long long ServerTransactionsMetrics::getTotalCommitted() const {
return _totalCommitted.load();
}
void ServerTransactionsMetrics::incrementTotalCommitted() {
_totalCommitted.fetchAndAdd(1);
}
boost::optional ServerTransactionsMetrics::getOldestActiveTS() const {
if (_oldestActiveOplogEntryTS.empty()) {
return boost::none;
}
return *(_oldestActiveOplogEntryTS.begin());
}
void ServerTransactionsMetrics::addActiveTS(Timestamp oldestOplogEntryTS) {
auto ret = _oldestActiveOplogEntryTS.insert(oldestOplogEntryTS);
// If ret.second is false, the timestamp we tried to insert already existed.
invariant(ret.second == true,
str::stream() << "This oplog entry timestamp already exists."
<< "TS: "
<< oldestOplogEntryTS.toString());
}
void ServerTransactionsMetrics::removeActiveTS(Timestamp oldestOplogEntryTS) {
auto it = _oldestActiveOplogEntryTS.find(oldestOplogEntryTS);
invariant(it != _oldestActiveOplogEntryTS.end(),
str::stream() << "This oplog entry timestamp does not exist "
<< "or has already been removed."
<< "TS: "
<< oldestOplogEntryTS.toString());
_oldestActiveOplogEntryTS.erase(it);
}
unsigned int ServerTransactionsMetrics::getTotalActiveTS() const {
return _oldestActiveOplogEntryTS.size();
}
void ServerTransactionsMetrics::updateStats(TransactionsStats* stats) {
stats->setCurrentActive(_currentActive.load());
stats->setCurrentInactive(_currentInactive.load());
stats->setCurrentOpen(_currentOpen.load());
stats->setTotalAborted(_totalAborted.load());
stats->setTotalCommitted(_totalCommitted.load());
stats->setTotalStarted(_totalStarted.load());
}
class TransactionsSSS : public ServerStatusSection {
public:
TransactionsSSS() : ServerStatusSection("transactions") {}
virtual ~TransactionsSSS() {}
virtual bool includeByDefault() const {
return true;
}
virtual BSONObj generateSection(OperationContext* opCtx,
const BSONElement& configElement) const {
TransactionsStats stats;
// Retryable writes and multi-document transactions metrics are both included in the same
// serverStatus section because both utilize similar internal machinery for tracking their
// lifecycle within a session. Both are assigned transaction numbers, and so both are often
// referred to as “transactions”.
RetryableWritesStats::get(opCtx)->updateStats(&stats);
ServerTransactionsMetrics::get(opCtx)->updateStats(&stats);
return stats.toBSON();
}
} transactionsSSS;
} // namespace mongo