From 14d626d96181fd746cd9abd22843197eed26fe71 Mon Sep 17 00:00:00 2001
From: Muhammed Galib Uludag <m231@tnctr.com>
Date: Sun, 27 Jun 2021 16:16:39 +0300
Subject: [PATCH] added mongodb sink

---
 include/spdlog/sinks/mongo_sink.h | 91 +++++++++++++++++++++++++++++++
 1 file changed, 91 insertions(+)
 create mode 100644 include/spdlog/sinks/mongo_sink.h

diff --git a/include/spdlog/sinks/mongo_sink.h b/include/spdlog/sinks/mongo_sink.h
new file mode 100644
index 00000000..254b8e18
--- /dev/null
+++ b/include/spdlog/sinks/mongo_sink.h
@@ -0,0 +1,91 @@
+/*****************************************************************************
+ * mongo_sink.h
+ *
+ * Created: 6/27/2021 2021 by mguludag
+ *
+ * Copyright 2021 mguludag. All rights reserved.
+ *
+ * This file may be distributed under the terms of GNU Public License version
+ * 3 (GPL v3) as defined by the Free Software Foundation (FSF). A copy of the
+ * license should have been included with this file, or the project in which
+ * this file belongs to. You may also find the details of GPL v3 at:
+ * http://www.gnu.org/licenses/gpl-3.0.txt
+ *
+ * If you have any questions regarding the use of this file, feel free to
+ * contact the author of this file, or the owner of the project in which
+ * this file belongs to.
+ *****************************************************************************/
+#pragma once
+#include "spdlog/common.h"
+#include "spdlog/details/log_msg.h"
+#include "spdlog/pattern_formatter.h"
+#include "spdlog/sinks/sink.h"
+
+#include <bsoncxx/builder/stream/document.hpp>
+#include <bsoncxx/types.hpp>
+#include <bsoncxx/view_or_value.hpp>
+
+#include <mongocxx/client.hpp>
+#include <mongocxx/instance.hpp>
+#include <mongocxx/uri.hpp>
+
+#include <mutex>
+
+namespace spdlog {
+namespace sinks {
+class mongo_sink : public sink {
+public:
+  mongo_sink(const std::string &db_name, const std::string &collection_name,
+             const std::string &uri = "mongodb://localhost:27017") {
+    try {
+      client_ = std::make_unique<mongocxx::client>(mongocxx::uri{uri});
+      db_name_ = db_name;
+      coll_name_ = collection_name;
+      set_pattern("%v");
+    } catch (const std::exception &e) {
+      std::cerr << e.what() << '\n';
+      throw spdlog_ex("Error opening database");
+    }
+  }
+
+  void set_pattern(const std::string &pattern) override {
+    formatter_ = std::unique_ptr<spdlog::pattern_formatter>(
+        new spdlog::pattern_formatter(pattern, pattern_time_type::local, ""));
+  }
+
+  void set_formatter(std::unique_ptr<spdlog::formatter> sink_formatter) {}
+
+  ~mongo_sink() { flush(); }
+
+  void flush() override {}
+
+  void log(const details::log_msg &msg) override {
+    using bsoncxx::builder::stream::document;
+    using bsoncxx::builder::stream::finalize;
+
+    if (client_ != nullptr) {
+      memory_buf_t formatted;
+      formatter_->format(msg, formatted);
+      auto doc = document{}
+                 << "timestamp" << bsoncxx::types::b_date(msg.time) << "level"
+                 << level::to_string_view(msg.level).data() << "message"
+                 << std::string(formatted.begin(), formatted.end())
+                 << "logger_name"
+                 << std::string(msg.logger_name.begin(), msg.logger_name.end())
+                 << "thread_id" << static_cast<int>(msg.thread_id) << finalize;
+      std::lock_guard<std::mutex> guard(mtx_);
+      client_->database(db_name_).collection(coll_name_).insert_one(doc.view());
+    }
+  }
+
+private:
+  static mongocxx::instance instance_;
+  std::mutex mtx_;
+  std::string db_name_;
+  std::string coll_name_;
+  std::unique_ptr<mongocxx::client> client_ = nullptr;
+  std::unique_ptr<spdlog::pattern_formatter> formatter_;
+};
+mongocxx::instance mongo_sink::instance_{};
+} // namespace sinks
+} // namespace spdlog