ClickHouse/dbms/src/Interpreters/InterpreterAlterQuery.h

126 lines
3.5 KiB
C++
Raw Normal View History

2013-08-07 13:07:42 +00:00
#pragma once
#include <Storages/IStorage.h>
#include <Storages/AlterCommands.h>
#include <Interpreters/Context.h>
#include <Interpreters/IInterpreter.h>
#include <Parsers/ASTAlterQuery.h>
2013-08-07 13:07:42 +00:00
namespace DB
{
2017-06-02 21:37:28 +00:00
/** Allows you add or remove a column in the table.
* It also allows you to manipulate the partitions of the MergeTree family tables.
2013-08-07 13:07:42 +00:00
*/
2015-06-18 02:11:05 +00:00
class InterpreterAlterQuery : public IInterpreter
2013-08-07 13:07:42 +00:00
{
public:
2017-05-23 18:01:50 +00:00
InterpreterAlterQuery(const ASTPtr & query_ptr_, const Context & context_);
2013-08-07 13:07:42 +00:00
BlockIO execute() override;
2013-08-07 13:07:42 +00:00
private:
struct PartitionCommand
{
enum Type
{
DROP_PARTITION,
ATTACH_PARTITION,
FETCH_PARTITION,
FREEZE_PARTITION,
RESHARD_PARTITION,
CLEAR_COLUMN,
};
Type type;
Field partition;
Field column_name;
2017-06-02 21:37:28 +00:00
bool detach = false; /// true for DETACH PARTITION.
bool part = false;
2017-06-02 21:37:28 +00:00
String from; /// For FETCH PARTITION - path in ZK to the shard, from which to download the partition.
2017-06-02 21:37:28 +00:00
/// For RESHARD PARTITION.
WeightedZooKeeperPaths weighted_zookeeper_paths;
ASTPtr sharding_key_expr;
bool do_copy = false;
Field coordinator;
/// For FREEZE PARTITION
String with_name;
static PartitionCommand dropPartition(const Field & partition, bool detach)
{
PartitionCommand res;
res.type = DROP_PARTITION;
res.partition = partition;
res.detach = detach;
return res;
}
static PartitionCommand clearColumn(const Field & partition, const Field & column_name)
{
PartitionCommand res;
res.type = CLEAR_COLUMN;
res.partition = partition;
res.column_name = column_name;
return res;
}
static PartitionCommand attachPartition(const Field & partition, bool part)
{
PartitionCommand res;
res.type = ATTACH_PARTITION;
res.partition = partition;
res.part = part;
return res;
}
static PartitionCommand fetchPartition(const Field & partition, const String & from)
{
PartitionCommand res;
res.type = FETCH_PARTITION;
res.partition = partition;
res.from = from;
return res;
}
static PartitionCommand freezePartition(const Field & partition, const String & with_name)
{
PartitionCommand res;
res.type = FREEZE_PARTITION;
res.partition = partition;
res.with_name = with_name;
return res;
}
static PartitionCommand reshardPartitions(const Field & partition_,
const WeightedZooKeeperPaths & weighted_zookeeper_paths_, const ASTPtr & sharding_key_expr_,
bool do_copy_, const Field & coordinator_)
{
PartitionCommand res;
res.type = RESHARD_PARTITION;
res.partition = partition_;
res.weighted_zookeeper_paths = weighted_zookeeper_paths_;
res.sharding_key_expr = sharding_key_expr_;
res.do_copy = do_copy_;
res.coordinator = coordinator_;
return res;
}
};
using PartitionCommands = std::vector<PartitionCommand>;
ASTPtr query_ptr;
Context context;
static void parseAlter(const ASTAlterQuery::ParameterContainer & params,
AlterCommands & out_alter_commands, PartitionCommands & out_partition_commands);
2013-08-07 13:07:42 +00:00
};
2013-08-07 13:07:42 +00:00
}