ClickHouse/dbms/src/Interpreters/ExternalDictionariesLoader.h

#pragma once

#include <Dictionaries/IDictionary.h>
#include <Interpreters/ExternalLoader.h>
#include <memory>


namespace DB
{
class Context;
class IExternalLoaderConfigRepository;

/// Manages user-defined dictionaries.
class ExternalDictionariesLoader : public ExternalLoader
{
public:
    using DictPtr = std::shared_ptr<const IDictionaryBase>;

    /// Dictionaries will be loaded immediately and then will be updated in separate thread, each 'reload_period' seconds.
    ExternalDictionariesLoader(Context & context_);

    DictPtr getDictionary(const std::string & name) const
    {
        return std::static_pointer_cast<const IDictionaryBase>(getLoadable(name));
    }

    DictPtr tryGetDictionary(const std::string & name) const
    {
        return std::static_pointer_cast<const IDictionaryBase>(tryGetLoadable(name));
    }

    void addConfigRepository(
        const std::string & repository_name,
        std::unique_ptr<IExternalLoaderConfigRepository> config_repository);

protected:
    LoadablePtr create(const std::string & name, const Poco::Util::AbstractConfiguration & config,
                       const std::string & key_in_config) const override;

    friend class StorageSystemDictionaries;
    friend class DatabaseDictionary;

private:
    Context & context;
};

}