component.hpp

Go to the documentation of this file.

#pragma once
 
 
#include <chrono>
#include <vector>
 
#include <userver/components/loggable_component_base.hpp>
#include <userver/concurrent/async_event_source.hpp>
#include <userver/dynamic_config/snapshot.hpp>
#include <userver/engine/mutex.hpp>
#include <userver/utils/statistics/entry.hpp>
 
#include <userver/storages/postgres/database.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace components {
 
// clang-format off
 
 
// clang-format on
 
class Postgres : public LoggableComponentBase {
 public:
  static constexpr auto kDefaultMaxReplicationLag = std::chrono::seconds{60};
  static constexpr size_t kDefaultShardNumber = 0;
  static constexpr storages::postgres::CommandControl kDefaultCommandControl{
      std::chrono::milliseconds{500},  // network timeout
      std::chrono::milliseconds{250}   // statement timeout
  };
 
  Postgres(const ComponentConfig&, const ComponentContext&);
  ~Postgres() override;
 
  storages::postgres::ClusterPtr GetCluster() const;
 
  storages::postgres::ClusterPtr GetClusterForShard(size_t shard) const;
 
  size_t GetShardCount() const;
 
  storages::postgres::DatabasePtr GetDatabase() const { return database_; }
 
  void ExtendStatistics(utils::statistics::Writer& writer);
 
  static yaml_config::Schema GetStaticConfigSchema();
 
 private:
  void OnConfigUpdate(const dynamic_config::Snapshot& cfg);
 
  concurrent::AsyncEventSubscriberScope config_subscription_;
 
  utils::statistics::Entry statistics_holder_;
 
  std::string name_;
  std::string db_name_;
  storages::postgres::ClusterSettings initial_settings_;
  storages::postgres::DatabasePtr database_;
};
 
template <>
inline constexpr bool kHasValidate<Postgres> = true;
 
}  // namespace components
 
USERVER_NAMESPACE_END