component.hpp

Go to the documentation of this file.

#pragma once
 
/// @file userver/storages/redis/component.hpp
/// @brief @copybrief components::Redis
 
#include <memory>
#include <string>
#include <unordered_map>
 
#include <userver/components/component_fwd.hpp>
#include <userver/components/loggable_component_base.hpp>
#include <userver/dynamic_config/source.hpp>
#include <userver/rcu/rcu.hpp>
#include <userver/storages/redis/impl/base.hpp>
#include <userver/storages/redis/impl/wait_connected_mode.hpp>
#include <userver/testsuite/redis_control.hpp>
#include <userver/utils/statistics/entry.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace redis {
class Sentinel;
class ThreadPools;
}  // namespace redis
 
/// Components, clients and helpers for different databases and storages
namespace storages {}
 
/// Redis client
namespace storages::redis {
class Client;
class SubscribeClient;
class SubscribeClientImpl;
}  // namespace storages::redis
 
namespace components {
 
// clang-format off
 
/// @ingroup userver_components
///
/// @brief Redis client component
///
/// Provides access to a redis cluster.
///
/// ## Dynamic options:
/// * @ref REDIS_DEFAULT_COMMAND_CONTROL
/// * @ref REDIS_SUBSCRIBER_DEFAULT_COMMAND_CONTROL
/// * @ref REDIS_SUBSCRIPTIONS_REBALANCE_MIN_INTERVAL_SECONDS
/// * @ref REDIS_WAIT_CONNECTED
/// * @ref REDIS_COMMANDS_BUFFERING_SETTINGS
/// * @ref REDIS_METRICS_SETTINGS
/// * @ref REDIS_PUBSUB_METRICS_SETTINGS
///
/// ## Static options:
/// Name | Description | Default value
/// ---- | ----------- | -------------
/// thread_pools.redis_thread_pool_size | thread count to serve Redis requests | -
/// thread_pools.sentinel_thread_pool_size | thread count to serve sentinel requests. | -
/// groups | array of redis clusters to work with excluding subscribers | -
/// groups.[].config_name | key name in secdist with options for this cluster | -
/// groups.[].db | name to refer to the cluster in components::Redis::GetClient() | -
/// groups.[].sharding_strategy | one of RedisCluster, KeyShardCrc32, KeyShardTaximeterCrc32 or KeyShardGpsStorageDriver | "KeyShardTaximeterCrc32"
/// groups.[].allow_reads_from_master | allows read requests from master instance | false
/// subscribe_groups | array of redis clusters to work with in subscribe mode | -
/// subscribe_groups.[].config_name | key name in secdist with options for this cluster | -
/// subscribe_groups.[].db | name to refer to the cluster in components::Redis::GetSubscribeClient() | -
/// subscribe_groups.[].sharding_strategy | either RedisCluster or KeyShardTaximeterCrc32 | "KeyShardTaximeterCrc32"
///
/// ## Static configuration example:
///
/// ```
///    redis:
///        groups:
///          - config_name: taxi-tmp
///            db: taxi-tmp
///            sharding_strategy: "RedisCluster"
///          - config_name: taxi-tmp-pubsub
///            db: taxi-tmp-pubsub
///        subscribe_groups:
///          - config_name: taxi-tmp-pubsub
///            db: taxi-tmp-pubsub
///        thread_pools:
///            redis_thread_pool_size: 8
///            sentinel_thread_pool_size: 1
/// ```
 
// clang-format on
class Redis : public LoggableComponentBase {
 public:
  Redis(const ComponentConfig& config,
        const ComponentContext& component_context);
 
  ~Redis() override;
 
  /// @ingroup userver_component_names
  /// @brief The default name of components::Redis
  static constexpr std::string_view kName = "redis";
 
  std::shared_ptr<storages::redis::Client> GetClient(
      const std::string& name,
      USERVER_NAMESPACE::redis::RedisWaitConnected wait_connected = {}) const;
  [[deprecated("use GetClient()")]] std::shared_ptr<redis::Sentinel> Client(
      const std::string& name) const;
  std::shared_ptr<storages::redis::SubscribeClient> GetSubscribeClient(
      const std::string& name,
      USERVER_NAMESPACE::redis::RedisWaitConnected wait_connected = {}) const;
 
  static yaml_config::Schema GetStaticConfigSchema();
 
 private:
  void OnConfigUpdate(const dynamic_config::Snapshot& cfg);
 
  void Connect(const ComponentConfig& config,
               const ComponentContext& component_context,
               const testsuite::RedisControl& testsuite_redis_control);
 
  void WriteStatistics(utils::statistics::Writer& writer);
  void WriteStatisticsPubsub(utils::statistics::Writer& writer);
 
  std::shared_ptr<redis::ThreadPools> thread_pools_;
  std::unordered_map<std::string, std::shared_ptr<redis::Sentinel>> sentinels_;
  std::unordered_map<std::string, std::shared_ptr<storages::redis::Client>>
      clients_;
  std::unordered_map<std::string,
                     std::shared_ptr<storages::redis::SubscribeClientImpl>>
      subscribe_clients_;
 
  dynamic_config::Source config_;
  concurrent::AsyncEventSubscriberScope config_subscription_;
 
  utils::statistics::Entry statistics_holder_;
  utils::statistics::Entry subscribe_statistics_holder_;
 
  rcu::Variable<redis::MetricsSettings> metrics_settings_;
  rcu::Variable<redis::PubsubMetricsSettings> pubsub_metrics_settings_;
};
 
template <>
inline constexpr bool kHasValidate<Redis> = true;
 
}  // namespace components
 
USERVER_NAMESPACE_END