feat mongo: multimongo secdist update handling

commit_hash:8e41c046ec478914caf9f9cc605db9c1290c63e2
userver-framework · Dec 13, 2024 · 4c4701f · 4c4701f
1 parent d09b427
commit 4c4701f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/mongo/include/userver/storages/mongo/multi_mongo.hpp b/mongo/include/userver/storages/mongo/multi_mongo.hpp
@@ -53,7 +53,7 @@ class MultiMongo {
     /// @cond
     MultiMongo(
         std::string name,
-        const storages::secdist::Secdist& secdist,
+        storages::secdist::Secdist& secdist,
         storages::mongo::PoolConfig pool_config,
         clients::dns::Resolver* dns_resolver,
         dynamic_config::Source config_source
@@ -89,6 +89,8 @@ class MultiMongo {
 private:
     void OnConfigUpdate(const dynamic_config::Snapshot& config);
 
+    void OnSecdistUpdate(const storages::secdist::SecdistConfig& secdist);
+
     storages::mongo::PoolPtr FindPool(const std::string& dbalias) const;
 
     const std::string name_;
@@ -99,6 +101,7 @@ class MultiMongo {
     rcu::Variable<PoolMap> pool_map_;
     // config_subscriber_ must be the last field.
     concurrent::AsyncEventSubscriberScope config_subscriber_;
+    concurrent::AsyncEventSubscriberScope secdist_subscriber_;
 };
 
 }  // namespace storages::mongo

diff --git a/mongo/src/storages/mongo/multi_mongo.cpp b/mongo/src/storages/mongo/multi_mongo.cpp
@@ -72,7 +72,7 @@ void MultiMongo::PoolSet::Activate() { target_->pool_map_.Assign(*pool_map_ptr_)
 
 MultiMongo::MultiMongo(
     std::string name,
-    const storages::secdist::Secdist& secdist,
+    storages::secdist::Secdist& secdist,
     storages::mongo::PoolConfig pool_config,
     clients::dns::Resolver* dns_resolver,
     dynamic_config::Source config_source
@@ -83,6 +83,7 @@ MultiMongo::MultiMongo(
       pool_config_(std::move(pool_config)),
       dns_resolver_(dns_resolver) {
     config_subscriber_ = config_source_.UpdateAndListen(this, "multi_mongo", &MultiMongo::OnConfigUpdate);
+    secdist_subscriber_ = secdist.UpdateAndListen(this, "multi_mongo", &MultiMongo::OnSecdistUpdate);
 }
 
 storages::mongo::PoolPtr MultiMongo::GetPool(const std::string& dbalias) const {
@@ -142,6 +143,14 @@ storages::mongo::PoolPtr MultiMongo::FindPool(const std::string& dbalias) const
     return it->second;
 }
 
+void MultiMongo::OnSecdistUpdate(const storages::secdist::SecdistConfig& secdist) {
+    const auto pool_map = pool_map_.Read();
+    for (const auto& [dbalias, pool] : *pool_map) {
+        auto conn_string = secdist::GetSecdistConnectionString(secdist, dbalias);
+        pool->SetConnectionString(conn_string);
+    }
+}
+
 }  // namespace storages::mongo
 
 USERVER_NAMESPACE_END