pool.hpp

Go to the documentation of this file.

#pragma once
 
/// @file userver/storages/mongo/pool.hpp
/// @brief @copybrief storages::mongo::Pool
 
#include <memory>
#include <string>
#include <vector>
 
#include <userver/clients/dns/resolver_fwd.hpp>
#include <userver/dynamic_config/fwd.hpp>
#include <userver/storages/mongo/collection.hpp>
#include <userver/storages/mongo/pool_config.hpp>
#include <userver/utils/statistics/fwd.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace storages::mongo {
 
namespace impl {
class PoolImpl;
}  // namespace impl
 
/// @ingroup userver_clients
///
/// @brief MongoDB client pool.
///
/// Use constructor only for tests, in production the pool should be retrieved
/// from @ref userver_components "the components" via
/// components::Mongo::GetPool() or components::MultiMongo::GetPool().
///
/// ## Example usage:
///
/// @snippet storages/mongo/collection_mongotest.hpp  Sample Mongo usage
class Pool {
 public:
  /// @cond
 
  /// Client pool constructor, for internal use only
  /// @param id pool identification string
  /// @param uri database connection string
  /// @param pool_config static config
  /// @param dns_resolver asynchronous resolver or `nullptr`
  /// @param config_source dynamic config
  Pool(std::string id, const std::string& uri, const PoolConfig& pool_config,
       clients::dns::Resolver* dns_resolver,
       dynamic_config::Source config_source);
 
  ~Pool();
 
  void Start();
 
  void Stop();
  /// @endcond
 
  /// Checks whether a collection exists
  bool HasCollection(const std::string& name) const;
 
  /// Returns a handle for the specified collection
  Collection GetCollection(std::string name) const;
 
  /// Drops the associated database if it exists. New modifications of
  /// collections will attempt to re-create the database automatically.
  void DropDatabase();
 
  /// Get a list of all the collection names in the associated database
  std::vector<std::string> ListCollectionNames() const;
 
  void Ping();
 
  /// Writes pool statistics
  friend void DumpMetric(utils::statistics::Writer& writer, const Pool& pool);
 
 private:
  std::shared_ptr<impl::PoolImpl> impl_;
};
 
using PoolPtr = std::shared_ptr<Pool>;
 
}  // namespace storages::mongo
 
USERVER_NAMESPACE_END