in src/kudu/client/table_alterer-internal.cc [58:250]
Status KuduTableAlterer::Data::ToRequest(AlterTableRequestPB* req) {
if (!status_.ok()) {
return status_;
}
if (!rename_to_ &&
!new_extra_configs_ &&
!set_owner_to_ &&
!set_comment_to_ &&
!disk_size_limit_ &&
!row_count_limit_ &&
!set_replication_factor_to_ &&
steps_.empty()) {
return Status::InvalidArgument("No alter steps provided");
}
req->Clear();
req->set_modify_external_catalogs(modify_external_catalogs_);
req->mutable_table()->set_table_name(table_name_);
if (rename_to_) {
req->set_new_table_name(*rename_to_);
}
if (new_extra_configs_) {
req->mutable_new_extra_configs()->insert(new_extra_configs_->begin(),
new_extra_configs_->end());
}
if (set_owner_to_) {
req->set_new_table_owner(*set_owner_to_);
}
if (set_comment_to_) {
req->set_new_table_comment(*set_comment_to_);
}
if (set_replication_factor_to_) {
req->set_num_replicas(*set_replication_factor_to_);
}
if (schema_ != nullptr) {
RETURN_NOT_OK(SchemaToPB(*schema_, req->mutable_schema(),
SCHEMA_PB_WITHOUT_IDS |
SCHEMA_PB_WITHOUT_WRITE_DEFAULT |
SCHEMA_PB_WITHOUT_COMMENT));
}
if (disk_size_limit_) {
req->set_disk_size_limit(*disk_size_limit_);
}
if (row_count_limit_) {
req->set_row_count_limit(*row_count_limit_);
}
for (const Step& s : steps_) {
AlterTableRequestPB::Step* pb_step = req->add_alter_schema_steps();
pb_step->set_type(s.step_type);
switch (s.step_type) {
case AlterTableRequestPB::ADD_COLUMN:
{
if (s.spec->data_->name == Schema::GetAutoIncrementingColumnName()) {
return Status::InvalidArgument("can't add a column with reserved name",
s.spec->data_->name);
}
KuduColumnSchema col;
RETURN_NOT_OK(s.spec->ToColumnSchema(&col));
ColumnSchemaToPB(*col.col_,
pb_step->mutable_add_column()->mutable_schema(),
SCHEMA_PB_WITHOUT_WRITE_DEFAULT);
break;
}
case AlterTableRequestPB::DROP_COLUMN:
{
if (s.spec->data_->name == Schema::GetAutoIncrementingColumnName()) {
return Status::InvalidArgument("can't drop column", s.spec->data_->name);
}
pb_step->mutable_drop_column()->set_name(s.spec->data_->name);
break;
}
case AlterTableRequestPB::ALTER_COLUMN:
{
if (s.spec->data_->type ||
s.spec->data_->nullable ||
s.spec->data_->primary_key) {
return Status::NotSupported("unsupported alter operation",
s.spec->data_->name);
}
if (!s.spec->data_->rename_to &&
!s.spec->data_->default_val &&
!s.spec->data_->remove_default &&
!s.spec->data_->encoding &&
!s.spec->data_->compression &&
!s.spec->data_->block_size &&
!s.spec->data_->comment &&
!s.spec->data_->immutable) {
return Status::InvalidArgument("no alter operation specified",
s.spec->data_->name);
}
if (s.spec->data_->name == Schema::GetAutoIncrementingColumnName()) {
if (s.spec->data_->rename_to) {
return Status::InvalidArgument("can't change name for column",
Schema::GetAutoIncrementingColumnName());
} else if (s.spec->data_->remove_default) {
return Status::InvalidArgument("can't change remove default for column",
Schema::GetAutoIncrementingColumnName());
} else if (s.spec->data_->default_val) {
return Status::InvalidArgument("can't change default value for column",
Schema::GetAutoIncrementingColumnName());
} else if (s.spec->data_->immutable) {
return Status::InvalidArgument("can't change immutability for column",
Schema::GetAutoIncrementingColumnName());
}
}
// If the alter is solely a column rename, fall back to using
// RENAME_COLUMN, for backwards compatibility.
// TODO(wdb) Change this when compat can be broken.
if (!s.spec->data_->default_val &&
!s.spec->data_->remove_default &&
!s.spec->data_->encoding &&
!s.spec->data_->compression &&
!s.spec->data_->block_size &&
!s.spec->data_->comment &&
!s.spec->data_->immutable) {
pb_step->set_type(AlterTableRequestPB::RENAME_COLUMN);
pb_step->mutable_rename_column()->set_old_name(s.spec->data_->name);
pb_step->mutable_rename_column()->set_new_name(*(s.spec->data_->rename_to));
break;
}
ColumnSchemaDelta col_delta(s.spec->data_->name);
RETURN_NOT_OK(s.spec->ToColumnSchemaDelta(&col_delta));
auto* alter_pb = pb_step->mutable_alter_column();
ColumnSchemaDeltaToPB(col_delta, alter_pb->mutable_delta());
pb_step->set_type(AlterTableRequestPB::ALTER_COLUMN);
break;
}
case AlterTableRequestPB::ADD_RANGE_PARTITION:
{
RowOperationsPBEncoder encoder(pb_step->mutable_add_range_partition()
->mutable_range_bounds());
auto* partition_data = s.range_partition->data_;
RowOperationsPB_Type lower_bound_type =
partition_data->lower_bound_type_ == KuduTableCreator::INCLUSIVE_BOUND ?
RowOperationsPB::RANGE_LOWER_BOUND :
RowOperationsPB::EXCLUSIVE_RANGE_LOWER_BOUND;
RowOperationsPB_Type upper_bound_type =
partition_data->upper_bound_type_ == KuduTableCreator::EXCLUSIVE_BOUND ?
RowOperationsPB::RANGE_UPPER_BOUND :
RowOperationsPB::INCLUSIVE_RANGE_UPPER_BOUND;
encoder.Add(lower_bound_type, *partition_data->lower_bound_);
encoder.Add(upper_bound_type, *partition_data->upper_bound_);
for (const auto& hash_dimension : partition_data->hash_schema_) {
auto* custom_hash_schema_pb = pb_step->mutable_add_range_partition()->
mutable_custom_hash_schema()->add_hash_schema();
for (const auto& column_name : hash_dimension.column_names) {
custom_hash_schema_pb->add_columns()->set_name(column_name);
}
custom_hash_schema_pb->set_num_buckets(hash_dimension.num_buckets);
custom_hash_schema_pb->set_seed(hash_dimension.seed);
}
if (s.dimension_label) {
pb_step->mutable_add_range_partition()->set_dimension_label(
*s.dimension_label);
}
break;
}
case AlterTableRequestPB::DROP_RANGE_PARTITION:
{
RowOperationsPBEncoder encoder(pb_step->mutable_drop_range_partition()
->mutable_range_bounds());
auto* partition_data = s.range_partition->data_;
RowOperationsPB_Type lower_bound_type =
partition_data->lower_bound_type_ == KuduTableCreator::INCLUSIVE_BOUND ?
RowOperationsPB::RANGE_LOWER_BOUND :
RowOperationsPB::EXCLUSIVE_RANGE_LOWER_BOUND;
RowOperationsPB_Type upper_bound_type =
partition_data->upper_bound_type_ == KuduTableCreator::EXCLUSIVE_BOUND ?
RowOperationsPB::RANGE_UPPER_BOUND :
RowOperationsPB::INCLUSIVE_RANGE_UPPER_BOUND;
encoder.Add(lower_bound_type, *partition_data->lower_bound_);
encoder.Add(upper_bound_type, *partition_data->upper_bound_);
break;
}
default:
LOG(FATAL) << "unknown step type "
<< AlterTableRequestPB::StepType_Name(s.step_type);
}
}
return Status::OK();
}