mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 02:41:59 +00:00
71 lines
2.6 KiB
C++
71 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <Compression/CompressionInfo.h>
|
|
#include <Compression/ICompressionCodec.h>
|
|
#include <DataTypes/IDataType.h>
|
|
#include <Parsers/IAST_fwd.h>
|
|
#include <Common/IFactoryWithAliases.h>
|
|
|
|
#include <functional>
|
|
#include <memory>
|
|
#include <optional>
|
|
#include <unordered_map>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class ICompressionCodec;
|
|
|
|
using CompressionCodecPtr = std::shared_ptr<ICompressionCodec>;
|
|
|
|
using CodecNameWithLevel = std::pair<String, std::optional<int>>;
|
|
|
|
/** Creates a codec object by name of compression algorithm family and parameters.
|
|
*/
|
|
class CompressionCodecFactory final : private boost::noncopyable
|
|
{
|
|
protected:
|
|
using Creator = std::function<CompressionCodecPtr(const ASTPtr & parameters)>;
|
|
using CreatorWithType = std::function<CompressionCodecPtr(const ASTPtr & parameters, DataTypePtr column_type)>;
|
|
using SimpleCreator = std::function<CompressionCodecPtr()>;
|
|
using CompressionCodecsDictionary = std::unordered_map<String, CreatorWithType>;
|
|
using CompressionCodecsCodeDictionary = std::unordered_map<uint8_t, CreatorWithType>;
|
|
public:
|
|
|
|
static CompressionCodecFactory & instance();
|
|
|
|
/// Return default codec (currently LZ4)
|
|
CompressionCodecPtr getDefaultCodec() const;
|
|
|
|
/// Get codec by AST and possible column_type
|
|
/// some codecs can use information about type to improve inner settings
|
|
/// but every codec should be able to work without information about type
|
|
CompressionCodecPtr get(const ASTPtr & ast, DataTypePtr column_type = nullptr) const;
|
|
|
|
/// Get codec by method byte (no params available)
|
|
CompressionCodecPtr get(const uint8_t byte_code) const;
|
|
|
|
/// For backward compatibility with config settings
|
|
CompressionCodecPtr get(const String & family_name, std::optional<int> level) const;
|
|
|
|
/// Register codec with parameters and column type
|
|
void registerCompressionCodecWithType(const String & family_name, std::optional<uint8_t> byte_code, CreatorWithType creator);
|
|
/// Register codec with parameters
|
|
void registerCompressionCodec(const String & family_name, std::optional<uint8_t> byte_code, Creator creator);
|
|
|
|
/// Register codec without parameters
|
|
void registerSimpleCompressionCodec(const String & family_name, std::optional<uint8_t> byte_code, SimpleCreator creator);
|
|
|
|
protected:
|
|
CompressionCodecPtr getImpl(const String & family_name, const ASTPtr & arguments, DataTypePtr column_type) const;
|
|
|
|
private:
|
|
CompressionCodecsDictionary family_name_with_codec;
|
|
CompressionCodecsCodeDictionary family_code_with_codec;
|
|
CompressionCodecPtr default_codec;
|
|
|
|
CompressionCodecFactory();
|
|
};
|
|
|
|
}
|