feat mongo: transactions core

commit_hash:fbd68b0242bae6fb5e536b1ecbe1f89f76dbf83b

Author: segoon

.mapping.json

@@ -3037,6 +3037,7 @@
   "mongo/include/userver/storages/mongo/options.hpp":"taxi/uservices/userver/mongo/include/userver/storages/mongo/options.hpp",
   "mongo/include/userver/storages/mongo/pool.hpp":"taxi/uservices/userver/mongo/include/userver/storages/mongo/pool.hpp",
   "mongo/include/userver/storages/mongo/pool_config.hpp":"taxi/uservices/userver/mongo/include/userver/storages/mongo/pool_config.hpp",
+  "mongo/include/userver/storages/mongo/transaction.hpp":"taxi/uservices/userver/mongo/include/userver/storages/mongo/transaction.hpp",
   "mongo/include/userver/storages/mongo/utest/mongo_fixture.hpp":"taxi/uservices/userver/mongo/include/userver/storages/mongo/utest/mongo_fixture.hpp",
   "mongo/include/userver/storages/mongo/utest/mongo_local.hpp":"taxi/uservices/userver/mongo/include/userver/storages/mongo/utest/mongo_local.hpp",
   "mongo/include/userver/storages/mongo/write_result.hpp":"taxi/uservices/userver/mongo/include/userver/storages/mongo/write_result.hpp",
@@ -3079,10 +3080,14 @@
   "mongo/src/storages/mongo/cdriver/collection_impl.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/collection_impl.hpp",
   "mongo/src/storages/mongo/cdriver/cursor_impl.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/cursor_impl.cpp",
   "mongo/src/storages/mongo/cdriver/cursor_impl.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/cursor_impl.hpp",
+  "mongo/src/storages/mongo/cdriver/find_and_modify.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/find_and_modify.cpp",
+  "mongo/src/storages/mongo/cdriver/find_and_modify.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/find_and_modify.hpp",
   "mongo/src/storages/mongo/cdriver/logger.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/logger.cpp",
   "mongo/src/storages/mongo/cdriver/logger.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/logger.hpp",
   "mongo/src/storages/mongo/cdriver/pool_impl.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/pool_impl.cpp",
   "mongo/src/storages/mongo/cdriver/pool_impl.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/pool_impl.hpp",
+  "mongo/src/storages/mongo/cdriver/transaction_collection_impl.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/transaction_collection_impl.cpp",
+  "mongo/src/storages/mongo/cdriver/transaction_collection_impl.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/transaction_collection_impl.hpp",
   "mongo/src/storages/mongo/cdriver/wrappers.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/wrappers.cpp",
   "mongo/src/storages/mongo/cdriver/wrappers.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/cdriver/wrappers.hpp",
   "mongo/src/storages/mongo/collection.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/collection.cpp",
@@ -3131,6 +3136,10 @@
   "mongo/src/storages/mongo/stats_serialize.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/stats_serialize.hpp",
   "mongo/src/storages/mongo/tcp_connect_precheck.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/tcp_connect_precheck.cpp",
   "mongo/src/storages/mongo/tcp_connect_precheck.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/tcp_connect_precheck.hpp",
+  "mongo/src/storages/mongo/transaction.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/transaction.cpp",
+  "mongo/src/storages/mongo/transaction_impl.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/transaction_impl.cpp",
+  "mongo/src/storages/mongo/transaction_impl.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/transaction_impl.hpp",
+  "mongo/src/storages/mongo/transaction_mongotest.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/transaction_mongotest.cpp",
   "mongo/src/storages/mongo/utest/mongo_local.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/utest/mongo_local.cpp",
   "mongo/src/storages/mongo/util_mongotest.cpp":"taxi/uservices/userver/mongo/src/storages/mongo/util_mongotest.cpp",
   "mongo/src/storages/mongo/util_mongotest.hpp":"taxi/uservices/userver/mongo/src/storages/mongo/util_mongotest.hpp",

mongo/include/userver/formats/bson/document.hpp

@@ -30,6 +30,8 @@ class Document : public Value {
     /// @endcond
 };
 
+void PrintTo(const Document& value, std::ostream* os);
+
 }  // namespace formats::bson
 
 USERVER_NAMESPACE_END

mongo/include/userver/storages/mongo/bulk.hpp

@@ -11,6 +11,7 @@ USERVER_NAMESPACE_BEGIN
 
 namespace storages::mongo::impl::cdriver {
 class CDriverCollectionImpl;
+class CDriverTransactionCollectionImpl;
 }  // namespace storages::mongo::impl::cdriver
 
 namespace storages::mongo::operations {
@@ -71,6 +72,7 @@ class Bulk {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 56;

mongo/include/userver/storages/mongo/exception.hpp

@@ -47,6 +47,22 @@ class PoolOverloadException : public MongoException {
     using MongoException::MongoException;
 };
 
+/// Transaction-related exception
+class TransactionException : public MongoException {
+public:
+    enum class Type {
+        kTransientError,  ///< Transient error that may be retried
+        kPermanentError,  ///< Permanent error, cannot retry
+    };
+
+    TransactionException(Type type, std::string_view what);
+
+    Type GetType() const noexcept { return type_; }
+
+private:
+    Type type_;
+};
+
 /// Network (connectivity) error
 class NetworkException : public MongoException {
 public:

mongo/include/userver/storages/mongo/operations.hpp

@@ -12,6 +12,7 @@ USERVER_NAMESPACE_BEGIN
 
 namespace storages::mongo::impl::cdriver {
 class CDriverCollectionImpl;
+class CDriverTransactionCollectionImpl;
 }  // namespace storages::mongo::impl::cdriver
 
 /// Collection operations
@@ -38,6 +39,7 @@ class Count {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 96;
@@ -66,6 +68,7 @@ class CountApprox {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 72;
@@ -100,6 +103,7 @@ class Find {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 96;
@@ -125,6 +129,7 @@ class InsertOne {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 80;
@@ -154,6 +159,7 @@ class InsertMany {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 88;
@@ -180,6 +186,7 @@ class ReplaceOne {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 96;
@@ -213,6 +220,7 @@ class Update {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 96;
@@ -243,6 +251,7 @@ class Delete {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 80;
@@ -274,8 +283,10 @@ class FindAndModify {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
+    FindAndModify(Impl&&);
     static constexpr size_t kSize = 80;
     static constexpr size_t kAlignment = 8;
     // MAC_COMPAT: std::string size differs
@@ -301,8 +312,10 @@ class FindAndRemove {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
+    FindAndRemove(Impl&&);
     static constexpr size_t kSize = 72;
     static constexpr size_t kAlignment = 8;
     // MAC_COMPAT: std::string size differs
@@ -331,6 +344,7 @@ class Aggregate {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 120;
@@ -360,6 +374,7 @@ class Distinct final {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 104;
@@ -383,6 +398,7 @@ class Drop {
 
 private:
     friend class storages::mongo::impl::cdriver::CDriverCollectionImpl;
+    friend class storages::mongo::impl::cdriver::CDriverTransactionCollectionImpl;
 
     class Impl;
     static constexpr size_t kSize = 56;

mongo/include/userver/storages/mongo/pool.hpp

@@ -11,6 +11,7 @@
 #include <userver/dynamic_config/fwd.hpp>
 #include <userver/storages/mongo/collection.hpp>
 #include <userver/storages/mongo/pool_config.hpp>
+#include <userver/storages/mongo/transaction.hpp>
 #include <userver/utils/statistics/fwd.hpp>
 #include <userver/utils/zstring_view.hpp>
 
@@ -55,6 +56,11 @@ class Pool {
     /// @throws storages::mongo::MongoException if failed to connect to the mongo server.
     void Ping();
 
+    /// @brief Begin a new transaction
+    /// @return Transaction handle for executing operations within transaction context
+    /// @throws MongoException if transaction cannot be started
+    Transaction BeginTransaction() const;
+
     /// @cond
     // For internal use only
     Pool(

mongo/include/userver/storages/mongo/transaction.hpp

@@ -0,0 +1,93 @@
+#pragma once
+
+/// @file userver/storages/mongo/transaction.hpp
+/// @brief @copybrief storages::mongo::Transaction
+
+#include <memory>
+#include <string>
+
+#include <userver/storages/mongo/collection.hpp>
+#include <userver/storages/mongo/options.hpp>
+
+USERVER_NAMESPACE_BEGIN
+
+namespace storages::mongo {
+
+namespace impl {
+class TransactionImpl;
+}  // namespace impl
+
+/// @brief MongoDB transaction handle that provides ACID guarantees.
+///
+/// Transactions group multiple operations within logical units of work
+/// and provide ACID guarantees. All operations in a transaction either
+/// succeed completely or fail completely.
+///
+/// ## Usage example:
+///
+/// @snippet mongo/src/storages/mongo/transaction_mongotest.cpp [transaction]
+///
+/// @note Transactions require MongoDB replica sets or sharded clusters.
+/// @note Operations within a transaction are automatically retried according
+///       to MongoDB transaction retry semantics.
+class Transaction {
+public:
+    /// @brief Creates invalid transaction
+    Transaction();
+
+    /// @cond
+    // For internal use only.
+    explicit Transaction(std::unique_ptr<impl::TransactionImpl>);
+    /// @endcond
+
+    /// @brief Move constructor
+    Transaction(Transaction&&) noexcept;
+
+    /// @brief Move assignment operator
+    Transaction& operator=(Transaction&&) noexcept;
+
+    /// @brief Destructor. Aborts transaction if not committed.
+    ~Transaction();
+
+    /// @brief Get a collection handle within the transaction context
+    /// @param name collection name
+    /// @return Collection handle that operates within this transaction
+    Collection GetCollection(std::string name);
+
+    /// @brief Commit the transaction
+    ///
+    /// Makes all changes within the transaction permanent and visible
+    /// to other operations. If the transaction fails to commit,
+    /// an exception will be thrown.
+    ///
+    /// @throws MongoException on commit failure
+    void Commit();
+
+    /// @brief Abort the transaction
+    ///
+    /// Discards all changes made within the transaction.
+    /// This method is idempotent and safe to call multiple times.
+    void Abort() noexcept;
+
+    /// @brief Check if transaction is active
+    /// @return true if transaction is active (not committed or aborted)
+    bool IsActive() const noexcept;
+
+    /// @brief Get the transaction state
+    enum class State {
+        kNone,        ///< Transaction not started
+        kInProgress,  ///< Transaction is active
+        kCommitted,   ///< Transaction has been committed
+        kAborted      ///< Transaction has been aborted
+    };
+
+    /// @brief Get current transaction state
+    State GetState() const noexcept;
+
+private:
+    std::unique_ptr<impl::TransactionImpl> impl_;
+};
+
+}  // namespace storages::mongo
+
+USERVER_NAMESPACE_END

mongo/src/formats/bson/document.cpp

@@ -2,6 +2,7 @@
 
 #include <formats/bson/value_impl.hpp>
 #include <userver/formats/bson/inline.hpp>
+#include <userver/formats/bson/serialize.hpp>
 #include <userver/utils/assert.hpp>
 
 USERVER_NAMESPACE_BEGIN
@@ -20,6 +21,11 @@ Document::Document(impl::BsonHolder bson)
     : Value(std::make_shared<impl::ValueImpl>(std::move(bson)))
 {}
 
+void PrintTo(const Document& value, std::ostream* os)
+{
+    *os << ToCanonicalJsonString(value);
+}
+
 }  // namespace formats::bson
 
 USERVER_NAMESPACE_END

mongo/src/formats/bson/value.cpp

@@ -9,6 +9,7 @@
 #include <userver/formats/bson/bson_builder.hpp>
 #include <userver/formats/bson/document.hpp>
 #include <userver/formats/bson/serialize.hpp>
+#include <userver/formats/json/value.hpp>
 #include <userver/utils/algo.hpp>
 
 USERVER_NAMESPACE_BEGIN