From ce76085e2464c38119b307851d4c63687d9a581f Mon Sep 17 00:00:00 2001
From: Benety Goh <benety@mongodb.com>
Date: Fri, 28 Apr 2017 12:18:56 -0400
Subject: SERVER-28988 add ReplicationProcess

This class is intended to hold replication state in a secondary node related to
the copying and applying operations from the sync source.

This initial implementation holds a cached copy of the rollback ID only.
---
 src/mongo/db/repl/replication_process.cpp | 122 ++++++++++++++++++++++++++++++
 1 file changed, 122 insertions(+)
 create mode 100644 src/mongo/db/repl/replication_process.cpp

(limited to 'src/mongo/db/repl/replication_process.cpp')

diff --git a/src/mongo/db/repl/replication_process.cpp b/src/mongo/db/repl/replication_process.cpp
new file mode 100644
index 00000000000..a50b83ddcfc
--- /dev/null
+++ b/src/mongo/db/repl/replication_process.cpp
@@ -0,0 +1,122 @@
+/**
+ *    Copyright (C) 2017 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 <http://www.gnu.org/licenses/>.
+ *
+ *    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/client.h"
+#include "mongo/db/operation_context.h"
+#include "mongo/db/repl/replication_process.h"
+#include "mongo/db/repl/storage_interface.h"
+#include "mongo/db/service_context.h"
+#include "mongo/util/assert_util.h"
+
+namespace mongo {
+namespace repl {
+
+
+namespace {
+
+const auto getReplicationProcess =
+    ServiceContext::declareDecoration<std::unique_ptr<ReplicationProcess>>();
+
+const int kUninitializedRollbackId = -1;
+
+}  // namespace
+
+ReplicationProcess* ReplicationProcess::get(ServiceContext* service) {
+    return getReplicationProcess(service).get();
+}
+
+ReplicationProcess* ReplicationProcess::get(ServiceContext& service) {
+    return getReplicationProcess(service).get();
+}
+
+ReplicationProcess* ReplicationProcess::get(OperationContext* opCtx) {
+    return get(opCtx->getClient()->getServiceContext());
+}
+
+
+void ReplicationProcess::set(ServiceContext* service, std::unique_ptr<ReplicationProcess> storage) {
+    auto& storageInterface = getReplicationProcess(service);
+    storageInterface = std::move(storage);
+}
+
+ReplicationProcess::ReplicationProcess(StorageInterface* storageInterface)
+    : _storageInterface(storageInterface), _rbid(kUninitializedRollbackId) {}
+
+StatusWith<int> ReplicationProcess::getRollbackID(OperationContext* opCtx) {
+    stdx::lock_guard<stdx::mutex> lock(_mutex);
+
+    if (kUninitializedRollbackId != _rbid) {
+        return _rbid;
+    }
+
+    // The _rbid, which caches the rollback ID persisted in the local.system.rollback.id collection,
+    // may be uninitialized for a couple of reasons:
+    // 1) This is the first time we are retrieving the rollback ID; or
+    // 2) The rollback ID was incremented previously using this class which has the side-effect of
+    //    invalidating the cached value.
+    auto rbidResult = _storageInterface->getRollbackID(opCtx);
+    if (!rbidResult.isOK()) {
+        return rbidResult;
+    }
+    _rbid = rbidResult.getValue();
+
+    invariant(kUninitializedRollbackId != _rbid);
+    return _rbid;
+}
+
+Status ReplicationProcess::initializeRollbackID(OperationContext* opCtx) {
+    stdx::lock_guard<stdx::mutex> lock(_mutex);
+
+    invariant(kUninitializedRollbackId == _rbid);
+
+    // Do not make any assumptions about the starting value of the rollback ID in the
+    // local.system.rollback.id collection other than it cannot be "kUninitializedRollbackId".
+    // Leave _rbid uninitialized until the next getRollbackID() to retrieve the actual value
+    // from storage.
+
+    return _storageInterface->initializeRollbackID(opCtx);
+}
+
+Status ReplicationProcess::incrementRollbackID(OperationContext* opCtx) {
+    stdx::lock_guard<stdx::mutex> lock(_mutex);
+
+    auto status = _storageInterface->incrementRollbackID(opCtx);
+
+    // If the rollback ID was incremented successfully, reset _rbid so that we will read from
+    // storage next time getRollbackID() is called.
+    if (status.isOK()) {
+        _rbid = kUninitializedRollbackId;
+    }
+
+    return status;
+}
+
+}  // namespace repl
+}  // namespace mongo
-- 
cgit v1.2.1