lru_cache_component_base.hpp

Go to the documentation of this file.

#pragma once
 
/// @file userver/cache/lru_cache_component_base.hpp
/// @brief @copybrief cache::LruCacheComponent
 
#include <functional>
 
#include <userver/cache/expirable_lru_cache.hpp>
#include <userver/cache/lru_cache_config.hpp>
#include <userver/components/component_base.hpp>
#include <userver/concurrent/async_event_source.hpp>
#include <userver/dump/dumper.hpp>
#include <userver/dump/meta.hpp>
#include <userver/dump/operations.hpp>
#include <userver/dynamic_config/source.hpp>
#include <userver/logging/log.hpp>
#include <userver/testsuite/cache_control.hpp>
#include <userver/utils/statistics/entry.hpp>
#include <userver/yaml_config/schema.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace cache {
 
namespace impl {
 
void RegisterOnStatisticsStorage(
    const components::ComponentContext& context,
    const std::string& name,
    std::function<void(utils::statistics::Writer&)> func
);
 
dynamic_config::Source FindDynamicConfigSource(const components::ComponentContext& context);
 
bool IsDumpSupportEnabled(const components::ComponentConfig& config);
 
yaml_config::Schema GetLruCacheComponentBaseSchema();
 
}  // namespace impl
 
/// @ingroup userver_components userver_base_classes
///
/// @brief Base class for @ref scripts/docs/en/userver/lru_cache.md "LRU and Expirable LRU cache" components.
///
/// Provides facilities for creating @ref scripts/docs/en/userver/lru_cache.md "LRU and Expirable LRU caches".
/// You need to override LruCacheComponent::DoGetByKey to handle cache misses.
///
/// Caching components must be configured in service config (see options below)
/// and may be reconfigured dynamically via @ref components::DynamicConfig.
///
/// ## LruCacheComponent Dynamic config
/// * @ref USERVER_LRU_CACHES
///
/// ## Static options of cache::LruCacheComponent :
/// @include{doc} scripts/docs/en/components_schema/core/src/cache/lru_cache_component_base.md
///
/// Options inherited from @ref components::ComponentBase :
/// @include{doc} scripts/docs/en/components_schema/core/src/components/impl/component_base.md
///
/// ## Example usage:
///
/// @snippet cache/lru_cache_component_base_test.hpp  Sample lru cache component
///
/// Do not forget to @ref userver_components "add the component to component list":
/// @snippet cache/lru_cache_component_base_test.cpp  Sample lru cache component registration
///
/// ## Example config:
/// @snippet cache/lru_cache_component_base_test.cpp  Sample lru cache component config
template <typename Key, typename Value, typename Hash = std::hash<Key>, typename Equal = std::equal_to<Key>>
// NOLINTNEXTLINE(fuchsia-multiple-inheritance)
class LruCacheComponent : public components::ComponentBase, private dump::DumpableEntity {
public:
    using Cache = ExpirableLruCache<Key, Value, Hash, Equal>;
    using CacheWrapper = LruCacheWrapper<Key, Value, Hash, Equal>;
 
    LruCacheComponent(const components::ComponentConfig&, const components::ComponentContext&);
 
    ~LruCacheComponent() override;
 
    CacheWrapper GetCache();
 
    static yaml_config::Schema GetStaticConfigSchema();
 
protected:
    virtual Value DoGetByKey(const Key& key) = 0;
 
    std::shared_ptr<Cache> GetCacheRaw() { return cache_; }
 
private:
    void DropCache();
 
    Value GetByKey(const Key& key);
 
    void OnConfigUpdate(const dynamic_config::Snapshot& cfg);
 
    void UpdateConfig(const LruCacheConfig& config);
 
    static constexpr bool kCacheIsDumpable = dump::kIsDumpable<Key> && dump::kIsDumpable<Value>;
 
    void GetAndWrite(dump::Writer& writer) const override;
    void ReadAndSet(dump::Reader& reader) override;
 
    const std::string name_;
    const LruCacheConfigStatic static_config_;
    const std::shared_ptr<Cache> cache_;
    std::shared_ptr<dump::Dumper> dumper_;
 
    // Subscriptions must be the last fields.
    concurrent::AsyncEventSubscriberScope config_subscription_;
};
 
template <typename Key, typename Value, typename Hash, typename Equal>
LruCacheComponent<
    Key,
    Value,
    Hash,
    Equal>::LruCacheComponent(const components::ComponentConfig& config, const components::ComponentContext& context)
    : ComponentBase(config, context),
      name_(components::GetCurrentComponentName(context)),
      static_config_(config),
      cache_(std::make_shared<Cache>(static_config_.ways, static_config_.GetWaySize()))
{
    if (impl::IsDumpSupportEnabled(config)) {
        dumper_ = std::make_shared<dump::Dumper>(config, context, static_cast<dump::DumpableEntity&>(*this));
        cache_->SetDumper(dumper_);
        dumper_->ReadDump();
    }
 
    cache_->SetMaxLifetime(static_config_.config.lifetime);
    cache_->SetBackgroundUpdate(static_config_.config.background_update);
 
    if (static_config_.use_dynamic_config) {
        LOG_INFO()
            << "Dynamic LRU cache config is enabled, subscribing on "
               "dynamic-config updates, cache="
            << name_;
 
        config_subscription_ =
            impl::FindDynamicConfigSource(context)
                .UpdateAndListen(this, "cache." + name_, &LruCacheComponent::OnConfigUpdate);
    } else {
        LOG_INFO() << "Dynamic LRU cache config is disabled, cache=" << name_;
    }
 
    impl::RegisterOnStatisticsStorage(context, name_, [this](utils::statistics::Writer& writer) { writer = *cache_; });
 
    testsuite::RegisterCacheScope(context, this, &LruCacheComponent::DropCache);
}
 
template <typename Key, typename Value, typename Hash, typename Equal>
LruCacheComponent<Key, Value, Hash, Equal>::~LruCacheComponent() {
    config_subscription_.Unsubscribe();
 
    if (dumper_) {
        dumper_->CancelWriteTaskAndWait();
    }
}
 
template <typename Key, typename Value, typename Hash, typename Equal>
typename LruCacheComponent<Key, Value, Hash, Equal>::CacheWrapper LruCacheComponent<
    Key,
    Value,
    Hash,
    Equal>::GetCache() {
    return CacheWrapper(cache_, [this](const Key& key) { return GetByKey(key); });
}
 
template <typename Key, typename Value, typename Hash, typename Equal>
void LruCacheComponent<Key, Value, Hash, Equal>::DropCache() {
    cache_->Invalidate();
}
 
template <typename Key, typename Value, typename Hash, typename Equal>
Value LruCacheComponent<Key, Value, Hash, Equal>::GetByKey(const Key& key) {
    return DoGetByKey(key);
}
 
template <typename Key, typename Value, typename Hash, typename Equal>
void LruCacheComponent<Key, Value, Hash, Equal>::OnConfigUpdate(const dynamic_config::Snapshot& cfg) {
    const auto config = GetLruConfig(cfg, name_);
    if (config) {
        LOG_DEBUG() << "Using dynamic config for LRU cache";
        UpdateConfig(*config);
    } else {
        LOG_DEBUG() << "Using static config for LRU cache";
        UpdateConfig(static_config_.config);
    }
}
 
template <typename Key, typename Value, typename Hash, typename Equal>
void LruCacheComponent<Key, Value, Hash, Equal>::UpdateConfig(const LruCacheConfig& config) {
    cache_->SetWaySize(config.GetWaySize(static_config_.ways));
    cache_->SetMaxLifetime(config.lifetime);
    cache_->SetBackgroundUpdate(config.background_update);
}
 
template <typename Key, typename Value, typename Hash, typename Equal>
yaml_config::Schema LruCacheComponent<Key, Value, Hash, Equal>::GetStaticConfigSchema() {
    return impl::GetLruCacheComponentBaseSchema();
}
 
template <typename Key, typename Value, typename Hash, typename Equal>
void LruCacheComponent<Key, Value, Hash, Equal>::GetAndWrite(dump::Writer& writer) const {
    if constexpr (kCacheIsDumpable) {
        cache_->Write(writer);
    } else {
        dump::ThrowDumpUnimplemented(name_);
    }
}
 
template <typename Key, typename Value, typename Hash, typename Equal>
void LruCacheComponent<Key, Value, Hash, Equal>::ReadAndSet(dump::Reader& reader) {
    if constexpr (kCacheIsDumpable) {
        cache_->Read(reader);
    } else {
        dump::ThrowDumpUnimplemented(name_);
    }
}
 
}  // namespace cache
 
USERVER_NAMESPACE_END