mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 17:32:32 +00:00
1010 lines
35 KiB
C++
1010 lines
35 KiB
C++
#pragma once
|
|
|
|
#include <Core/SettingsFields.h>
|
|
#include <Common/SettingsChanges.h>
|
|
#include <Common/FieldVisitorToString.h>
|
|
#include <IO/Operators.h>
|
|
#include <base/range.h>
|
|
#include <boost/blank.hpp>
|
|
#include <unordered_map>
|
|
|
|
|
|
namespace boost::program_options
|
|
{
|
|
class options_description;
|
|
}
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class ReadBuffer;
|
|
class WriteBuffer;
|
|
|
|
enum class SettingsWriteFormat
|
|
{
|
|
BINARY, /// Part of the settings are serialized as strings, and other part as variants. This is the old behaviour.
|
|
STRINGS_WITH_FLAGS, /// All settings are serialized as strings. Before each value the flag `is_important` is serialized.
|
|
DEFAULT = STRINGS_WITH_FLAGS,
|
|
};
|
|
|
|
/** Template class to define collections of settings.
|
|
* Example of usage:
|
|
*
|
|
* mysettings.h:
|
|
* #define APPLY_FOR_MYSETTINGS(M) \
|
|
* M(UInt64, a, 100, "Description of a", 0) \
|
|
* M(Float, f, 3.11, "Description of f", IMPORTANT) // IMPORTANT - means the setting can't be ignored by older versions) \
|
|
* M(String, s, "default", "Description of s", 0)
|
|
*
|
|
* DECLARE_SETTINGS_TRAITS(MySettingsTraits, APPLY_FOR_MYSETTINGS)
|
|
|
|
* struct MySettings : public BaseSettings<MySettingsTraits>
|
|
* {
|
|
* };
|
|
*
|
|
* mysettings.cpp:
|
|
* IMPLEMENT_SETTINGS_TRAITS(MySettingsTraits, APPLY_FOR_MYSETTINGS)
|
|
*/
|
|
template <class TTraits>
|
|
class BaseSettings : public TTraits::Data
|
|
{
|
|
using CustomSettingMap = std::unordered_map<std::string_view, std::pair<std::shared_ptr<const String>, SettingFieldCustom>>;
|
|
public:
|
|
BaseSettings() = default;
|
|
BaseSettings(const BaseSettings &) = default;
|
|
BaseSettings(BaseSettings &&) noexcept = default;
|
|
BaseSettings & operator=(const BaseSettings &) = default;
|
|
BaseSettings & operator=(BaseSettings &&) noexcept = default;
|
|
virtual ~BaseSettings() = default;
|
|
|
|
using Traits = TTraits;
|
|
|
|
virtual void set(std::string_view name, const Field & value);
|
|
Field get(std::string_view name) const;
|
|
|
|
void setString(std::string_view name, const String & value);
|
|
String getString(std::string_view name) const;
|
|
|
|
bool tryGet(std::string_view name, Field & value) const;
|
|
bool tryGetString(std::string_view name, String & value) const;
|
|
|
|
bool isChanged(std::string_view name) const;
|
|
SettingsChanges changes() const;
|
|
void applyChange(const SettingChange & change);
|
|
void applyChanges(const SettingsChanges & changes);
|
|
void applyChanges(const BaseSettings & changes); /// NOLINT
|
|
|
|
/// Resets all the settings to their default values.
|
|
void resetToDefault();
|
|
/// Resets specified setting to its default value.
|
|
void resetToDefault(std::string_view name);
|
|
|
|
bool has(std::string_view name) const { return hasBuiltin(name) || hasCustom(name); }
|
|
static bool hasBuiltin(std::string_view name);
|
|
bool hasCustom(std::string_view name) const;
|
|
|
|
const char * getTypeName(std::string_view name) const;
|
|
const char * getDescription(std::string_view name) const;
|
|
|
|
/// Checks if it's possible to assign a field to a specified value and throws an exception if not.
|
|
/// This function doesn't change the fields, it performs check only.
|
|
static void checkCanSet(std::string_view name, const Field & value);
|
|
static void checkCanSetString(std::string_view name, const String & str);
|
|
|
|
/// Conversions without changing the fields.
|
|
static Field castValueUtil(std::string_view name, const Field & value);
|
|
static String valueToStringUtil(std::string_view name, const Field & value);
|
|
static Field stringToValueUtil(std::string_view name, const String & str);
|
|
|
|
void write(WriteBuffer & out, SettingsWriteFormat format = SettingsWriteFormat::DEFAULT) const;
|
|
void read(ReadBuffer & in, SettingsWriteFormat format = SettingsWriteFormat::DEFAULT);
|
|
|
|
// A debugging aid.
|
|
std::string toString() const;
|
|
|
|
/// Represents a reference to a setting field.
|
|
class SettingFieldRef
|
|
{
|
|
public:
|
|
const String & getName() const;
|
|
Field getValue() const;
|
|
Field getDefaultValue() const;
|
|
String getValueString() const;
|
|
String getDefaultValueString() const;
|
|
bool isValueChanged() const;
|
|
const char * getTypeName() const;
|
|
const char * getDescription() const;
|
|
bool isCustom() const;
|
|
bool isObsolete() const;
|
|
|
|
bool operator==(const SettingFieldRef & other) const { return (getName() == other.getName()) && (getValue() == other.getValue()); }
|
|
bool operator!=(const SettingFieldRef & other) const { return !(*this == other); }
|
|
|
|
private:
|
|
friend class BaseSettings;
|
|
const BaseSettings * settings;
|
|
const typename Traits::Accessor * accessor;
|
|
size_t index;
|
|
std::conditional_t<Traits::allow_custom_settings, const CustomSettingMap::mapped_type*, boost::blank> custom_setting;
|
|
};
|
|
|
|
enum SkipFlags
|
|
{
|
|
SKIP_NONE = 0,
|
|
SKIP_CHANGED = 0x01,
|
|
SKIP_UNCHANGED = 0x02,
|
|
SKIP_BUILTIN = 0x04,
|
|
SKIP_CUSTOM = 0x08,
|
|
SKIP_ALL = SKIP_CHANGED | SKIP_UNCHANGED | SKIP_CUSTOM | SKIP_BUILTIN,
|
|
};
|
|
|
|
class Iterator
|
|
{
|
|
public:
|
|
Iterator & operator++();
|
|
Iterator operator++(int); /// NOLINT
|
|
const SettingFieldRef & operator *() const { return field_ref; }
|
|
|
|
bool operator ==(const Iterator & other) const;
|
|
bool operator !=(const Iterator & other) const { return !(*this == other); }
|
|
|
|
private:
|
|
friend class BaseSettings;
|
|
Iterator(const BaseSettings & settings_, const typename Traits::Accessor & accessor_, SkipFlags skip_flags_);
|
|
void doSkip();
|
|
void setPointerToCustomSetting();
|
|
|
|
SettingFieldRef field_ref;
|
|
std::conditional_t<Traits::allow_custom_settings, CustomSettingMap::const_iterator, boost::blank> custom_settings_iterator;
|
|
SkipFlags skip_flags;
|
|
};
|
|
|
|
class Range
|
|
{
|
|
public:
|
|
Range(const BaseSettings & settings_, SkipFlags skip_flags_) : settings(settings_), accessor(Traits::Accessor::instance()), skip_flags(skip_flags_) {}
|
|
Iterator begin() const { return Iterator(settings, accessor, skip_flags); }
|
|
Iterator end() const { return Iterator(settings, accessor, SKIP_ALL); }
|
|
|
|
private:
|
|
const BaseSettings & settings;
|
|
const typename Traits::Accessor & accessor;
|
|
SkipFlags skip_flags;
|
|
};
|
|
|
|
Range all(SkipFlags skip_flags = SKIP_NONE) const { return Range{*this, skip_flags}; }
|
|
Range allChanged() const { return all(SKIP_UNCHANGED); }
|
|
Range allUnchanged() const { return all(SKIP_CHANGED); }
|
|
Range allBuiltin() const { return all(SKIP_CUSTOM); }
|
|
Range allCustom() const { return all(SKIP_BUILTIN); }
|
|
|
|
Iterator begin() const { return allChanged().begin(); }
|
|
Iterator end() const { return allChanged().end(); }
|
|
|
|
private:
|
|
SettingFieldCustom & getCustomSetting(std::string_view name);
|
|
const SettingFieldCustom & getCustomSetting(std::string_view name) const;
|
|
const SettingFieldCustom * tryGetCustomSetting(std::string_view name) const;
|
|
|
|
std::conditional_t<Traits::allow_custom_settings, CustomSettingMap, boost::blank> custom_settings_map;
|
|
};
|
|
|
|
struct BaseSettingsHelpers
|
|
{
|
|
[[noreturn]] static void throwSettingNotFound(std::string_view name);
|
|
static void warningSettingNotFound(std::string_view name);
|
|
|
|
static void writeString(std::string_view str, WriteBuffer & out);
|
|
static String readString(ReadBuffer & in);
|
|
|
|
enum Flags : UInt64
|
|
{
|
|
IMPORTANT = 0x01,
|
|
CUSTOM = 0x02,
|
|
OBSOLETE = 0x04,
|
|
};
|
|
static void writeFlags(Flags flags, WriteBuffer & out);
|
|
static Flags readFlags(ReadBuffer & in);
|
|
};
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::set(std::string_view name, const Field & value)
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
accessor.setValue(*this, index, value);
|
|
else
|
|
getCustomSetting(name) = value;
|
|
}
|
|
|
|
template <typename TTraits>
|
|
Field BaseSettings<TTraits>::get(std::string_view name) const
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
return accessor.getValue(*this, index);
|
|
else
|
|
return static_cast<Field>(getCustomSetting(name));
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::setString(std::string_view name, const String & value)
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
accessor.setValueString(*this, index, value);
|
|
else
|
|
getCustomSetting(name).parseFromString(value);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
String BaseSettings<TTraits>::getString(std::string_view name) const
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
return accessor.getValueString(*this, index);
|
|
else
|
|
return getCustomSetting(name).toString();
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool BaseSettings<TTraits>::tryGet(std::string_view name, Field & value) const
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
{
|
|
value = accessor.getValue(*this, index);
|
|
return true;
|
|
}
|
|
if (const auto * custom_setting = tryGetCustomSetting(name))
|
|
{
|
|
value = static_cast<Field>(*custom_setting);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool BaseSettings<TTraits>::tryGetString(std::string_view name, String & value) const
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
{
|
|
value = accessor.getValueString(*this, index);
|
|
return true;
|
|
}
|
|
if (const auto * custom_setting = tryGetCustomSetting(name))
|
|
{
|
|
value = custom_setting->toString();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool BaseSettings<TTraits>::isChanged(std::string_view name) const
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
return accessor.isValueChanged(*this, index);
|
|
return tryGetCustomSetting(name) != nullptr;
|
|
}
|
|
|
|
template <typename TTraits>
|
|
SettingsChanges BaseSettings<TTraits>::changes() const
|
|
{
|
|
SettingsChanges res;
|
|
for (const auto & field : *this)
|
|
res.emplace_back(field.getName(), field.getValue());
|
|
return res;
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::applyChange(const SettingChange & change)
|
|
{
|
|
set(change.name, change.value);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::applyChanges(const SettingsChanges & changes)
|
|
{
|
|
for (const auto & change : changes)
|
|
applyChange(change);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::applyChanges(const BaseSettings & other_settings)
|
|
{
|
|
for (const auto & field : other_settings)
|
|
set(field.getName(), field.getValue());
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::resetToDefault()
|
|
{
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
for (size_t i : collections::range(accessor.size()))
|
|
{
|
|
if (accessor.isValueChanged(*this, i))
|
|
accessor.resetValueToDefault(*this, i);
|
|
}
|
|
|
|
if constexpr (Traits::allow_custom_settings)
|
|
custom_settings_map.clear();
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::resetToDefault(std::string_view name)
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
accessor.resetValueToDefault(*this, index);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool BaseSettings<TTraits>::hasBuiltin(std::string_view name)
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
return (accessor.find(name) != static_cast<size_t>(-1));
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool BaseSettings<TTraits>::hasCustom(std::string_view name) const
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
return tryGetCustomSetting(name);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
const char * BaseSettings<TTraits>::getTypeName(std::string_view name) const
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
return accessor.getTypeName(index);
|
|
else if (tryGetCustomSetting(name))
|
|
return "Custom";
|
|
else
|
|
BaseSettingsHelpers::throwSettingNotFound(name);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
const char * BaseSettings<TTraits>::getDescription(std::string_view name) const
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
return accessor.getDescription(index);
|
|
else if (tryGetCustomSetting(name))
|
|
return "Custom";
|
|
else
|
|
BaseSettingsHelpers::throwSettingNotFound(name);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::checkCanSet(std::string_view name, const Field & value)
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
castValueUtil(name, value);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::checkCanSetString(std::string_view name, const String & str)
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
stringToValueUtil(name, str);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
Field BaseSettings<TTraits>::castValueUtil(std::string_view name, const Field & value)
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
return accessor.castValueUtil(index, value);
|
|
if constexpr (Traits::allow_custom_settings)
|
|
return value;
|
|
else
|
|
BaseSettingsHelpers::throwSettingNotFound(name);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
String BaseSettings<TTraits>::valueToStringUtil(std::string_view name, const Field & value)
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
return accessor.valueToStringUtil(index, value);
|
|
if constexpr (Traits::allow_custom_settings)
|
|
return value.dump();
|
|
else
|
|
BaseSettingsHelpers::throwSettingNotFound(name);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
Field BaseSettings<TTraits>::stringToValueUtil(std::string_view name, const String & str)
|
|
{
|
|
name = TTraits::resolveName(name);
|
|
try
|
|
{
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
if (size_t index = accessor.find(name); index != static_cast<size_t>(-1))
|
|
return accessor.stringToValueUtil(index, str);
|
|
if constexpr (Traits::allow_custom_settings)
|
|
return Field::restoreFromDump(str);
|
|
else
|
|
BaseSettingsHelpers::throwSettingNotFound(name);
|
|
}
|
|
catch (Exception & e)
|
|
{
|
|
e.addMessage("while parsing value '{}' for setting '{}'", str, name);
|
|
throw;
|
|
}
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::write(WriteBuffer & out, SettingsWriteFormat format) const
|
|
{
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
|
|
for (const auto & field : *this)
|
|
{
|
|
bool is_custom = field.isCustom();
|
|
bool is_important = !is_custom && accessor.isImportant(field.index);
|
|
|
|
BaseSettingsHelpers::writeString(field.getName(), out);
|
|
|
|
if ((format >= SettingsWriteFormat::STRINGS_WITH_FLAGS) || is_custom)
|
|
{
|
|
using Flags = BaseSettingsHelpers::Flags;
|
|
Flags flags{0};
|
|
if (is_custom)
|
|
flags = static_cast<Flags>(flags | Flags::CUSTOM);
|
|
else if (is_important)
|
|
flags = static_cast<Flags>(flags | Flags::IMPORTANT);
|
|
BaseSettingsHelpers::writeFlags(flags, out);
|
|
|
|
BaseSettingsHelpers::writeString(field.getValueString(), out);
|
|
}
|
|
else
|
|
accessor.writeBinary(*this, field.index, out);
|
|
}
|
|
|
|
/// Empty string is a marker of the end of settings.
|
|
BaseSettingsHelpers::writeString(std::string_view{}, out);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::read(ReadBuffer & in, SettingsWriteFormat format)
|
|
{
|
|
resetToDefault();
|
|
const auto & accessor = Traits::Accessor::instance();
|
|
while (true)
|
|
{
|
|
String read_name = BaseSettingsHelpers::readString(in);
|
|
if (read_name.empty() /* empty string is a marker of the end of settings */)
|
|
break;
|
|
|
|
std::string_view name = TTraits::resolveName(read_name);
|
|
size_t index = accessor.find(name);
|
|
|
|
using Flags = BaseSettingsHelpers::Flags;
|
|
Flags flags{0};
|
|
if (format >= SettingsWriteFormat::STRINGS_WITH_FLAGS)
|
|
flags = BaseSettingsHelpers::readFlags(in);
|
|
bool is_important = (flags & Flags::IMPORTANT);
|
|
bool is_custom = (flags & Flags::CUSTOM);
|
|
|
|
if (index != static_cast<size_t>(-1))
|
|
{
|
|
if (is_custom)
|
|
{
|
|
SettingFieldCustom temp;
|
|
temp.parseFromString(BaseSettingsHelpers::readString(in));
|
|
accessor.setValue(*this, index, static_cast<Field>(temp));
|
|
}
|
|
else if (format >= SettingsWriteFormat::STRINGS_WITH_FLAGS)
|
|
accessor.setValueString(*this, index, BaseSettingsHelpers::readString(in));
|
|
else
|
|
accessor.readBinary(*this, index, in);
|
|
}
|
|
else if (is_custom && Traits::allow_custom_settings)
|
|
{
|
|
getCustomSetting(name).parseFromString(BaseSettingsHelpers::readString(in));
|
|
}
|
|
else if (is_important)
|
|
{
|
|
BaseSettingsHelpers::throwSettingNotFound(name);
|
|
}
|
|
else
|
|
{
|
|
BaseSettingsHelpers::warningSettingNotFound(name);
|
|
BaseSettingsHelpers::readString(in);
|
|
}
|
|
}
|
|
}
|
|
|
|
template <typename TTraits>
|
|
String BaseSettings<TTraits>::toString() const
|
|
{
|
|
WriteBufferFromOwnString out;
|
|
bool first = true;
|
|
for (const auto & setting : *this)
|
|
{
|
|
if (!first)
|
|
out << ", ";
|
|
out << setting.getName() << " = " << applyVisitor(FieldVisitorToString(), setting.getValue());
|
|
first = false;
|
|
}
|
|
return out.str();
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool operator==(const BaseSettings<TTraits> & left, const BaseSettings<TTraits> & right)
|
|
{
|
|
auto l = left.begin();
|
|
for (const auto & r : right)
|
|
{
|
|
if ((l == left.end()) || (*l != r))
|
|
return false;
|
|
++l;
|
|
}
|
|
return l == left.end();
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool operator!=(const BaseSettings<TTraits> & left, const BaseSettings<TTraits> & right)
|
|
{
|
|
return !(left == right);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
SettingFieldCustom & BaseSettings<TTraits>::getCustomSetting(std::string_view name)
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
auto it = custom_settings_map.find(name);
|
|
if (it == custom_settings_map.end())
|
|
{
|
|
auto new_name = std::make_shared<String>(name);
|
|
it = custom_settings_map.emplace(*new_name, std::make_pair(new_name, SettingFieldCustom{})).first;
|
|
}
|
|
return it->second.second;
|
|
}
|
|
BaseSettingsHelpers::throwSettingNotFound(name);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
const SettingFieldCustom & BaseSettings<TTraits>::getCustomSetting(std::string_view name) const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
auto it = custom_settings_map.find(name);
|
|
if (it != custom_settings_map.end())
|
|
return it->second.second;
|
|
}
|
|
BaseSettingsHelpers::throwSettingNotFound(name);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
const SettingFieldCustom * BaseSettings<TTraits>::tryGetCustomSetting(std::string_view name) const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
auto it = custom_settings_map.find(name);
|
|
if (it != custom_settings_map.end())
|
|
return &it->second.second;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
template <typename TTraits>
|
|
BaseSettings<TTraits>::Iterator::Iterator(const BaseSettings & settings_, const typename Traits::Accessor & accessor_, SkipFlags skip_flags_)
|
|
: skip_flags(skip_flags_)
|
|
{
|
|
field_ref.settings = &settings_;
|
|
field_ref.accessor = &accessor_;
|
|
auto & index = field_ref.index;
|
|
|
|
if (skip_flags == SKIP_ALL)
|
|
{
|
|
index = accessor_.size();
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
custom_settings_iterator = settings_.custom_settings_map.end();
|
|
field_ref.custom_setting = nullptr;
|
|
}
|
|
return;
|
|
}
|
|
|
|
if (skip_flags & SKIP_CHANGED)
|
|
{
|
|
skip_flags = static_cast<SkipFlags>(skip_flags | SKIP_CUSTOM);
|
|
if (skip_flags & SKIP_UNCHANGED)
|
|
skip_flags = static_cast<SkipFlags>(skip_flags | SKIP_BUILTIN);
|
|
}
|
|
|
|
if (skip_flags & SKIP_BUILTIN)
|
|
index = accessor_.size();
|
|
else
|
|
index = 0;
|
|
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
if (skip_flags & SKIP_CUSTOM)
|
|
custom_settings_iterator = settings_.custom_settings_map.end();
|
|
else
|
|
custom_settings_iterator = settings_.custom_settings_map.begin();
|
|
}
|
|
|
|
doSkip();
|
|
setPointerToCustomSetting();
|
|
}
|
|
|
|
template <typename TTraits>
|
|
typename BaseSettings<TTraits>::Iterator & BaseSettings<TTraits>::Iterator::operator++()
|
|
{
|
|
const auto & accessor = *field_ref.accessor;
|
|
auto & index = field_ref.index;
|
|
if (index != accessor.size())
|
|
++index;
|
|
else
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
++custom_settings_iterator;
|
|
}
|
|
doSkip();
|
|
setPointerToCustomSetting();
|
|
return *this;
|
|
}
|
|
|
|
template <typename TTraits>
|
|
typename BaseSettings<TTraits>::Iterator BaseSettings<TTraits>::Iterator::operator++(int) /// NOLINT
|
|
{
|
|
auto res = *this;
|
|
++*this;
|
|
return res;
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::Iterator::doSkip()
|
|
{
|
|
const auto & accessor = *field_ref.accessor;
|
|
const auto & settings = *field_ref.settings;
|
|
auto & index = field_ref.index;
|
|
if (skip_flags & SKIP_CHANGED)
|
|
{
|
|
while ((index != accessor.size()) && accessor.isValueChanged(settings, index))
|
|
++index;
|
|
}
|
|
else if (skip_flags & SKIP_UNCHANGED)
|
|
{
|
|
while ((index != accessor.size()) && !accessor.isValueChanged(settings, index))
|
|
++index;
|
|
}
|
|
}
|
|
|
|
template <typename TTraits>
|
|
void BaseSettings<TTraits>::Iterator::setPointerToCustomSetting()
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
const auto & accessor = *field_ref.accessor;
|
|
const auto & settings = *field_ref.settings;
|
|
const auto & index = field_ref.index;
|
|
if ((index == accessor.size()) && (custom_settings_iterator != settings.custom_settings_map.end()))
|
|
field_ref.custom_setting = &custom_settings_iterator->second;
|
|
else
|
|
field_ref.custom_setting = nullptr;
|
|
}
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool BaseSettings<TTraits>::Iterator::operator ==(const typename BaseSettings<TTraits>::Iterator & other) const
|
|
{
|
|
if constexpr (TTraits::allow_custom_settings)
|
|
{
|
|
if (custom_settings_iterator != other.custom_settings_iterator)
|
|
return false;
|
|
}
|
|
return ((field_ref.index == other.field_ref.index) && (field_ref.settings == other.field_ref.settings));
|
|
}
|
|
|
|
template <typename TTraits>
|
|
const String & BaseSettings<TTraits>::SettingFieldRef::getName() const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
if (custom_setting)
|
|
return *custom_setting->first;
|
|
}
|
|
return accessor->getName(index);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
Field BaseSettings<TTraits>::SettingFieldRef::getValue() const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
if (custom_setting)
|
|
return static_cast<Field>(custom_setting->second);
|
|
}
|
|
return accessor->getValue(*settings, index);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
Field BaseSettings<TTraits>::SettingFieldRef::getDefaultValue() const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
if (custom_setting)
|
|
return static_cast<Field>(custom_setting->second);
|
|
}
|
|
return accessor->getDefaultValue(index);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
String BaseSettings<TTraits>::SettingFieldRef::getValueString() const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
if (custom_setting)
|
|
return custom_setting->second.toString();
|
|
}
|
|
return accessor->getValueString(*settings, index);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
String BaseSettings<TTraits>::SettingFieldRef::getDefaultValueString() const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
if (custom_setting)
|
|
return custom_setting->second.toString();
|
|
}
|
|
return accessor->getDefaultValueString(index);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool BaseSettings<TTraits>::SettingFieldRef::isValueChanged() const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
if (custom_setting)
|
|
return true;
|
|
}
|
|
return accessor->isValueChanged(*settings, index);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
const char * BaseSettings<TTraits>::SettingFieldRef::getTypeName() const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
if (custom_setting)
|
|
return "Custom";
|
|
}
|
|
return accessor->getTypeName(index);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
const char * BaseSettings<TTraits>::SettingFieldRef::getDescription() const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
if (custom_setting)
|
|
return "Custom";
|
|
}
|
|
return accessor->getDescription(index);
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool BaseSettings<TTraits>::SettingFieldRef::isCustom() const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
return custom_setting != nullptr;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
template <typename TTraits>
|
|
bool BaseSettings<TTraits>::SettingFieldRef::isObsolete() const
|
|
{
|
|
if constexpr (Traits::allow_custom_settings)
|
|
{
|
|
if (custom_setting)
|
|
return false;
|
|
}
|
|
return accessor->isObsolete(index);
|
|
}
|
|
|
|
using AliasMap = std::unordered_map<std::string_view, std::string_view>;
|
|
|
|
/// NOLINTNEXTLINE
|
|
#define DECLARE_SETTINGS_TRAITS(SETTINGS_TRAITS_NAME, LIST_OF_SETTINGS_MACRO) \
|
|
DECLARE_SETTINGS_TRAITS_COMMON(SETTINGS_TRAITS_NAME, LIST_OF_SETTINGS_MACRO, 0)
|
|
|
|
/// NOLINTNEXTLINE
|
|
#define DECLARE_SETTINGS_TRAITS_ALLOW_CUSTOM_SETTINGS(SETTINGS_TRAITS_NAME, LIST_OF_SETTINGS_MACRO) \
|
|
DECLARE_SETTINGS_TRAITS_COMMON(SETTINGS_TRAITS_NAME, LIST_OF_SETTINGS_MACRO, 1)
|
|
|
|
/// NOLINTNEXTLINE
|
|
#define DECLARE_SETTINGS_TRAITS_COMMON(SETTINGS_TRAITS_NAME, LIST_OF_SETTINGS_MACRO, ALLOW_CUSTOM_SETTINGS) \
|
|
struct SETTINGS_TRAITS_NAME \
|
|
{ \
|
|
struct Data \
|
|
{ \
|
|
LIST_OF_SETTINGS_MACRO(DECLARE_SETTINGS_TRAITS_, SKIP_ALIAS) \
|
|
}; \
|
|
\
|
|
class Accessor \
|
|
{ \
|
|
public: \
|
|
static const Accessor & instance(); \
|
|
size_t size() const { return field_infos.size(); } \
|
|
size_t find(std::string_view name) const; \
|
|
const String & getName(size_t index) const { return field_infos[index].name; } \
|
|
const char * getTypeName(size_t index) const { return field_infos[index].type; } \
|
|
const char * getDescription(size_t index) const { return field_infos[index].description; } \
|
|
bool isImportant(size_t index) const { return field_infos[index].is_important; } \
|
|
bool isObsolete(size_t index) const { return field_infos[index].is_obsolete; } \
|
|
Field castValueUtil(size_t index, const Field & value) const { return field_infos[index].cast_value_util_function(value); } \
|
|
String valueToStringUtil(size_t index, const Field & value) const { return field_infos[index].value_to_string_util_function(value); } \
|
|
Field stringToValueUtil(size_t index, const String & str) const { return field_infos[index].string_to_value_util_function(str); } \
|
|
void setValue(Data & data, size_t index, const Field & value) const { return field_infos[index].set_value_function(data, value); } \
|
|
Field getValue(const Data & data, size_t index) const { return field_infos[index].get_value_function(data); } \
|
|
void setValueString(Data & data, size_t index, const String & str) const { return field_infos[index].set_value_string_function(data, str); } \
|
|
String getValueString(const Data & data, size_t index) const { return field_infos[index].get_value_string_function(data); } \
|
|
bool isValueChanged(const Data & data, size_t index) const { return field_infos[index].is_value_changed_function(data); } \
|
|
void resetValueToDefault(Data & data, size_t index) const { return field_infos[index].reset_value_to_default_function(data); } \
|
|
void writeBinary(const Data & data, size_t index, WriteBuffer & out) const { return field_infos[index].write_binary_function(data, out); } \
|
|
void readBinary(Data & data, size_t index, ReadBuffer & in) const { return field_infos[index].read_binary_function(data, in); } \
|
|
Field getDefaultValue(size_t index) const { return field_infos[index].get_default_value_function(); } \
|
|
String getDefaultValueString(size_t index) const { return field_infos[index].get_default_value_string_function(); } \
|
|
private: \
|
|
Accessor(); \
|
|
struct FieldInfo \
|
|
{ \
|
|
String name; \
|
|
const char * type; \
|
|
const char * description; \
|
|
bool is_important; \
|
|
bool is_obsolete; \
|
|
Field (*cast_value_util_function)(const Field &); \
|
|
String (*value_to_string_util_function)(const Field &); \
|
|
Field (*string_to_value_util_function)(const String &); \
|
|
void (*set_value_function)(Data &, const Field &) ; \
|
|
Field (*get_value_function)(const Data &) ; \
|
|
void (*set_value_string_function)(Data &, const String &) ; \
|
|
String (*get_value_string_function)(const Data &) ; \
|
|
bool (*is_value_changed_function)(const Data &); \
|
|
void (*reset_value_to_default_function)(Data &) ; \
|
|
void (*write_binary_function)(const Data &, WriteBuffer &) ; \
|
|
void (*read_binary_function)(Data &, ReadBuffer &) ; \
|
|
Field (*get_default_value_function)() ; \
|
|
String (*get_default_value_string_function)() ; \
|
|
}; \
|
|
std::vector<FieldInfo> field_infos; \
|
|
std::unordered_map<std::string_view, size_t> name_to_index_map; \
|
|
}; \
|
|
static constexpr bool allow_custom_settings = ALLOW_CUSTOM_SETTINGS; \
|
|
\
|
|
static inline const AliasMap aliases_to_settings = \
|
|
DefineAliases() LIST_OF_SETTINGS_MACRO(ALIAS_TO, ALIAS_FROM); \
|
|
\
|
|
using SettingsToAliasesMap = std::unordered_map<std::string_view, std::vector<std::string_view>>; \
|
|
static inline const SettingsToAliasesMap & settingsToAliases() \
|
|
{ \
|
|
static SettingsToAliasesMap setting_to_aliases_mapping = [] \
|
|
{ \
|
|
std::unordered_map<std::string_view, std::vector<std::string_view>> map; \
|
|
for (const auto & [alias, destination] : aliases_to_settings) \
|
|
map[destination].push_back(alias); \
|
|
return map; \
|
|
}(); \
|
|
return setting_to_aliases_mapping; \
|
|
} \
|
|
\
|
|
static std::string_view resolveName(std::string_view name) \
|
|
{ \
|
|
if (auto it = aliases_to_settings.find(name); it != aliases_to_settings.end()) \
|
|
return it->second; \
|
|
return name; \
|
|
} \
|
|
};
|
|
|
|
|
|
/// NOLINTNEXTLINE
|
|
#define SKIP_ALIAS(ALIAS_NAME)
|
|
/// NOLINTNEXTLINE
|
|
#define ALIAS_TO(TYPE, NAME, DEFAULT, DESCRIPTION, FLAGS) .setName(#NAME)
|
|
/// NOLINTNEXTLINE
|
|
#define ALIAS_FROM(ALIAS) .addAlias(#ALIAS)
|
|
|
|
struct DefineAliases
|
|
{
|
|
std::string_view name;
|
|
AliasMap map;
|
|
|
|
DefineAliases & setName(std::string_view value)
|
|
{
|
|
name = value;
|
|
return *this;
|
|
}
|
|
|
|
DefineAliases & addAlias(std::string_view value)
|
|
{
|
|
map.emplace(value, name);
|
|
return *this;
|
|
}
|
|
|
|
/// NOLINTNEXTLINE(google-explicit-constructor)
|
|
operator AliasMap() { return std::move(map); }
|
|
};
|
|
|
|
/// NOLINTNEXTLINE
|
|
#define DECLARE_SETTINGS_TRAITS_(TYPE, NAME, DEFAULT, DESCRIPTION, FLAGS) \
|
|
SettingField##TYPE NAME {DEFAULT};
|
|
|
|
/// NOLINTNEXTLINE
|
|
#define IMPLEMENT_SETTINGS_TRAITS(SETTINGS_TRAITS_NAME, LIST_OF_SETTINGS_MACRO) \
|
|
const SETTINGS_TRAITS_NAME::Accessor & SETTINGS_TRAITS_NAME::Accessor::instance() \
|
|
{ \
|
|
static const Accessor the_instance = [] \
|
|
{ \
|
|
Accessor res; \
|
|
constexpr int IMPORTANT = 0x01; \
|
|
UNUSED(IMPORTANT); \
|
|
LIST_OF_SETTINGS_MACRO(IMPLEMENT_SETTINGS_TRAITS_, SKIP_ALIAS) \
|
|
for (size_t i : collections::range(res.field_infos.size())) \
|
|
{ \
|
|
const auto & info = res.field_infos[i]; \
|
|
res.name_to_index_map.emplace(info.name, i); \
|
|
} \
|
|
return res; \
|
|
}(); \
|
|
return the_instance; \
|
|
} \
|
|
\
|
|
SETTINGS_TRAITS_NAME::Accessor::Accessor() {} \
|
|
\
|
|
size_t SETTINGS_TRAITS_NAME::Accessor::find(std::string_view name) const \
|
|
{ \
|
|
auto it = name_to_index_map.find(name); \
|
|
if (it != name_to_index_map.end()) \
|
|
return it->second; \
|
|
return static_cast<size_t>(-1); \
|
|
} \
|
|
\
|
|
template class BaseSettings<SETTINGS_TRAITS_NAME>;
|
|
|
|
/// NOLINTNEXTLINE
|
|
#define IMPLEMENT_SETTINGS_TRAITS_(TYPE, NAME, DEFAULT, DESCRIPTION, FLAGS) \
|
|
res.field_infos.emplace_back( \
|
|
FieldInfo{#NAME, #TYPE, DESCRIPTION, (FLAGS) & IMPORTANT, \
|
|
static_cast<bool>((FLAGS) & BaseSettingsHelpers::Flags::OBSOLETE), \
|
|
[](const Field & value) -> Field { return static_cast<Field>(SettingField##TYPE{value}); }, \
|
|
[](const Field & value) -> String { return SettingField##TYPE{value}.toString(); }, \
|
|
[](const String & str) -> Field { SettingField##TYPE temp; temp.parseFromString(str); return static_cast<Field>(temp); }, \
|
|
[](Data & data, const Field & value) { data.NAME = value; }, \
|
|
[](const Data & data) -> Field { return static_cast<Field>(data.NAME); }, \
|
|
[](Data & data, const String & str) { data.NAME.parseFromString(str); }, \
|
|
[](const Data & data) -> String { return data.NAME.toString(); }, \
|
|
[](const Data & data) -> bool { return data.NAME.changed; }, \
|
|
[](Data & data) { data.NAME = SettingField##TYPE{DEFAULT}; }, \
|
|
[](const Data & data, WriteBuffer & out) { data.NAME.writeBinary(out); }, \
|
|
[](Data & data, ReadBuffer & in) { data.NAME.readBinary(in); }, \
|
|
[]() -> Field { return static_cast<Field>(SettingField##TYPE{DEFAULT}); }, \
|
|
[]() -> String { return SettingField##TYPE{DEFAULT}.toString(); } \
|
|
});
|
|
}
|