1 files changed, 145 insertions, 0 deletions
diff --git a/src/mongo/db/free_mon/free_mon_op_observer.cpp b/src/mongo/db/free_mon/free_mon_op_observer.cpp
new file mode 100644
index 00000000000..b44e74f4765
--- /dev/null
+++ b/src/mongo/db/free_mon/free_mon_op_observer.cpp
@@ -0,0 +1,145 @@
+/**
+*    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 <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/free_mon/free_mon_op_observer.h"
+
+#include "mongo/db/free_mon/free_mon_controller.h"
+#include "mongo/db/free_mon/free_mon_storage.h"
+
+namespace mongo {
+namespace {
+
+bool isStandaloneOrPrimary(OperationContext* opCtx) {
+    auto replCoord = repl::ReplicationCoordinator::get(opCtx);
+    const bool isReplSet =
+        replCoord->getReplicationMode() == repl::ReplicationCoordinator::modeReplSet;
+    return !isReplSet || (repl::ReplicationCoordinator::get(opCtx)->getMemberState() ==
+                          repl::MemberState::RS_PRIMARY);
+}
+
+}  // namespace
+
+FreeMonOpObserver::FreeMonOpObserver() = default;
+
+FreeMonOpObserver::~FreeMonOpObserver() = default;
+
+repl::OpTime FreeMonOpObserver::onDropCollection(OperationContext* opCtx,
+                                                 const NamespaceString& collectionName,
+                                                 OptionalCollectionUUID uuid) {
+    if (collectionName == NamespaceString::kServerConfigurationNamespace) {
+        auto controller = FreeMonController::get(opCtx->getServiceContext());
+
+        if (controller != nullptr) {
+            controller->notifyOnDelete();
+        }
+    }
+
+    return {};
+}
+
+void FreeMonOpObserver::onInserts(OperationContext* opCtx,
+                                  const NamespaceString& nss,
+                                  OptionalCollectionUUID uuid,
+                                  std::vector<InsertStatement>::const_iterator begin,
+                                  std::vector<InsertStatement>::const_iterator end,
+                                  bool fromMigrate) {
+    if (isStandaloneOrPrimary(opCtx)) {
+        return;
+    }
+
+    for (auto it = begin; it != end; ++it) {
+        const auto& insertedDoc = it->doc;
+
+        if (nss == NamespaceString::kServerConfigurationNamespace) {
+            if (auto idElem = insertedDoc["_id"]) {
+                if (idElem.str() == FreeMonStorage::kFreeMonDocIdKey) {
+                    auto controller = FreeMonController::get(opCtx->getServiceContext());
+
+                    if (controller != nullptr) {
+                        controller->notifyOnUpsert(insertedDoc);
+                    }
+                }
+            }
+        }
+    }
+}
+
+void FreeMonOpObserver::onUpdate(OperationContext* opCtx, const OplogUpdateEntryArgs& args) {
+    if (isStandaloneOrPrimary(opCtx)) {
+        return;
+    }
+
+    if (args.nss == NamespaceString::kServerConfigurationNamespace) {
+        if (args.updatedDoc["_id"].str() == FreeMonStorage::kFreeMonDocIdKey) {
+            auto controller = FreeMonController::get(opCtx->getServiceContext());
+
+            if (controller != nullptr) {
+                controller->notifyOnUpsert(args.updatedDoc);
+            }
+        }
+    }
+}
+
+void FreeMonOpObserver::onDelete(OperationContext* opCtx,
+                                 const NamespaceString& nss,
+                                 OptionalCollectionUUID uuid,
+                                 StmtId stmtId,
+                                 bool fromMigrate,
+                                 const boost::optional<BSONObj>& deletedDoc) {
+    if (isStandaloneOrPrimary(opCtx)) {
+        return;
+    }
+
+    if (nss == NamespaceString::kServerConfigurationNamespace) {
+        if (deletedDoc.get()["_id"].str() == FreeMonStorage::kFreeMonDocIdKey) {
+            auto controller = FreeMonController::get(opCtx->getServiceContext());
+
+            if (controller != nullptr) {
+                controller->notifyOnDelete();
+            }
+        }
+    }
+}
+
+void FreeMonOpObserver::onReplicationRollback(OperationContext* opCtx,
+                                              const RollbackObserverInfo& rbInfo) {
+    // Invalidate any in-memory auth data if necessary.
+    const auto& rollbackNamespaces = rbInfo.rollbackNamespaces;
+    if (rollbackNamespaces.count(NamespaceString::kServerConfigurationNamespace) == 1) {
+        auto controller = FreeMonController::get(opCtx->getServiceContext());
+
+        if (controller != nullptr) {
+            controller->notifyOnRollback();
+        }
+    }
+}
+
+
+}  // namespace mongo