mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-18 04:12:19 +00:00
105 lines
3.2 KiB
C++
105 lines
3.2 KiB
C++
#include <DB/IO/CompressedStream.h>
|
||
#include <DB/IO/ReadHelpers.h>
|
||
#include <DB/Common/Exception.h>
|
||
#include <DB/Common/StringUtils.h>
|
||
#include <Poco/Util/AbstractConfiguration.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
namespace ErrorCodes
|
||
{
|
||
extern const int UNKNOWN_COMPRESSION_METHOD;
|
||
extern const int UNKNOWN_ELEMENT_IN_CONFIG;
|
||
}
|
||
|
||
|
||
/** Позволяет выбрать метод сжатия по указанным в конфигурационном файле условиям.
|
||
* Конфиг выглядит примерно так:
|
||
|
||
<compression>
|
||
|
||
<!-- Набор вариантов. Варианты проверяются подряд. Побеждает последний сработавший вариант. Если ни один не сработал, то используется lz4. -->
|
||
<case>
|
||
|
||
<!-- Условия. Должны сработать одновременно все. Часть условий могут быть не указаны. -->
|
||
<min_part_size>10000000000</min_part_size> <!-- Минимальный размер куска в байтах. -->
|
||
<min_part_size_ratio>0.01</min_part_size_ratio> <!-- Минимальный размер куска относительно всех данных таблицы. -->
|
||
|
||
<!-- Какой метод сжатия выбрать. -->
|
||
<method>zstd</method>
|
||
</case>
|
||
|
||
<case>
|
||
...
|
||
</case>
|
||
</compression>
|
||
*/
|
||
class CompressionMethodSelector
|
||
{
|
||
private:
|
||
struct Element
|
||
{
|
||
size_t min_part_size = 0;
|
||
double min_part_size_ratio = 0;
|
||
CompressionMethod method = CompressionMethod::LZ4;
|
||
|
||
void setMethod(const std::string & name)
|
||
{
|
||
if (name == "lz4")
|
||
method = CompressionMethod::LZ4;
|
||
else if (name == "zstd")
|
||
method = CompressionMethod::ZSTD;
|
||
else
|
||
throw Exception("Unknown compression method " + name, ErrorCodes::UNKNOWN_COMPRESSION_METHOD);
|
||
}
|
||
|
||
Element(Poco::Util::AbstractConfiguration & config, const std::string & config_prefix)
|
||
{
|
||
min_part_size = parse<size_t>(config.getString(config_prefix + ".min_part_size", "0"));
|
||
min_part_size_ratio = config.getDouble(config_prefix + ".min_part_size_ratio", 0);
|
||
|
||
setMethod(config.getString(config_prefix + ".method"));
|
||
}
|
||
|
||
bool check(size_t part_size, double part_size_ratio) const
|
||
{
|
||
return part_size >= min_part_size
|
||
&& part_size_ratio >= min_part_size_ratio;
|
||
}
|
||
};
|
||
|
||
std::vector<Element> elements;
|
||
|
||
public:
|
||
CompressionMethodSelector() {} /// Всегда возвращает метод по-умолчанию.
|
||
|
||
CompressionMethodSelector(Poco::Util::AbstractConfiguration & config, const std::string & config_prefix)
|
||
{
|
||
Poco::Util::AbstractConfiguration::Keys keys;
|
||
config.keys(config_prefix, keys);
|
||
|
||
for (const auto & name : keys)
|
||
{
|
||
if (!startsWith(name.data(), "case"))
|
||
throw Exception("Unknown element in config: " + config_prefix + "." + name + ", must be 'case'", ErrorCodes::UNKNOWN_ELEMENT_IN_CONFIG);
|
||
|
||
elements.emplace_back(config, config_prefix + "." + name);
|
||
}
|
||
}
|
||
|
||
CompressionMethod choose(size_t part_size, double part_size_ratio) const
|
||
{
|
||
CompressionMethod res = CompressionMethod::LZ4;
|
||
|
||
for (const auto & element : elements)
|
||
if (element.check(part_size, part_size_ratio))
|
||
res = element.method;
|
||
|
||
return res;
|
||
}
|
||
};
|
||
|
||
}
|