mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-14 03:25:15 +00:00
118 lines
3.5 KiB
C++
118 lines
3.5 KiB
C++
#pragma once
|
||
|
||
#include <DB/IO/WriteBufferFromFile.h>
|
||
#include <DB/IO/CompressedWriteBuffer.h>
|
||
#include <DB/DataStreams/NativeBlockOutputStream.h>
|
||
|
||
#include <DB/Storages/IStorage.h>
|
||
#include <DB/Interpreters/Set.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
|
||
/** Общая часть StorageSet и StorageJoin.
|
||
*/
|
||
class StorageSetOrJoinBase : public IStorage
|
||
{
|
||
friend class SetOrJoinBlockOutputStream;
|
||
|
||
public:
|
||
String getTableName() const override { return name; }
|
||
const NamesAndTypesList & getColumnsListImpl() const override { return *columns; }
|
||
|
||
void rename(const String & new_path_to_db, const String & new_database_name, const String & new_table_name) override;
|
||
|
||
BlockOutputStreamPtr write(ASTPtr query, const Settings & settings) override;
|
||
|
||
protected:
|
||
StorageSetOrJoinBase(
|
||
const String & path_,
|
||
const String & name_,
|
||
NamesAndTypesListPtr columns_,
|
||
const NamesAndTypesList & materialized_columns_,
|
||
const NamesAndTypesList & alias_columns_,
|
||
const ColumnDefaults & column_defaults_);
|
||
|
||
String path;
|
||
String name;
|
||
NamesAndTypesListPtr columns;
|
||
|
||
UInt64 increment = 0; /// Для имён файлов бэкапа.
|
||
|
||
/// Восстановление из бэкапа.
|
||
void restore();
|
||
|
||
private:
|
||
void restoreFromFile(const String & file_path);
|
||
|
||
/// Вставить блок в состояние.
|
||
virtual void insertBlock(const Block & block) = 0;
|
||
virtual size_t getSize() const = 0;
|
||
};
|
||
|
||
|
||
class SetOrJoinBlockOutputStream : public IBlockOutputStream
|
||
{
|
||
public:
|
||
SetOrJoinBlockOutputStream(StorageSetOrJoinBase & table_,
|
||
const String & backup_path_, const String & backup_tmp_path_, const String & backup_file_name_);
|
||
|
||
void write(const Block & block) override;
|
||
void writeSuffix() override;
|
||
|
||
private:
|
||
StorageSetOrJoinBase & table;
|
||
String backup_path;
|
||
String backup_tmp_path;
|
||
String backup_file_name;
|
||
WriteBufferFromFile backup_buf;
|
||
CompressedWriteBuffer compressed_backup_buf;
|
||
NativeBlockOutputStream backup_stream;
|
||
};
|
||
|
||
|
||
/** Позволяет сохранить множество для последующего использования в правой части оператора IN.
|
||
* При вставке в таблицу, данные будут вставлены в множество,
|
||
* а также записаны в файл-бэкап, для восстановления после перезапуска.
|
||
* Чтение из таблицы напрямую невозможно - возможно лишь указание в правой части оператора IN.
|
||
*/
|
||
class StorageSet : public StorageSetOrJoinBase
|
||
{
|
||
public:
|
||
static StoragePtr create(
|
||
const String & path_,
|
||
const String & name_,
|
||
NamesAndTypesListPtr columns_,
|
||
const NamesAndTypesList & materialized_columns_,
|
||
const NamesAndTypesList & alias_columns_,
|
||
const ColumnDefaults & column_defaults_)
|
||
{
|
||
return (new StorageSet{
|
||
path_, name_, columns_,
|
||
materialized_columns_, alias_columns_, column_defaults_})->thisPtr();
|
||
}
|
||
|
||
String getName() const override { return "Set"; }
|
||
|
||
/// Получить доступ к внутренностям.
|
||
SetPtr & getSet() { return set; }
|
||
|
||
private:
|
||
SetPtr set { new Set{Limits{}} };
|
||
|
||
StorageSet(
|
||
const String & path_,
|
||
const String & name_,
|
||
NamesAndTypesListPtr columns_,
|
||
const NamesAndTypesList & materialized_columns_,
|
||
const NamesAndTypesList & alias_columns_,
|
||
const ColumnDefaults & column_defaults_);
|
||
|
||
void insertBlock(const Block & block) override { set->insertFromBlock(block); }
|
||
size_t getSize() const override { return set->getTotalRowCount(); };
|
||
};
|
||
|
||
}
|