hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix the issue of direct loading getting stuck

Browse Source
This commit is contained in:
obdev
2023-03-17 09:10:58 +00:00
committed by ob-robot
parent 5257f801ff
commit a6fea4b8dc
1 changed files with 43 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
src/observer/table_load/ob_table_load_obj_cast.cpp
View File

@ -34,20 +34,50 @@ static int pad_obj(ObTableLoadCastObjCtx &cast_obj_ctx, const ObColumnSchemaV2 *
// is_pad = is_fixed_string;
//}
if (is_fixed_string) {
int32_t fixed_len = column_schema->get_data_length();
if (fixed_len > obj.val_len_) {
char *buf = (char *)cast_obj_ctx.cast_ctx_->allocator_v2_->alloc(fixed_len);
if (buf == nullptr) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("fail to allocate buf", K(fixed_len), KR(ret));
//Because NCHAR stores data in a format similar to Unicode, special treatment is required when padding.
if (column_schema->get_data_type() == ObNCharType) {
int32_t fixed_len = column_schema->get_data_length() * 2;
if (obj.val_len_ % 2 != 0) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("The length of data for NCHAR type must be even", K(obj.val_len_));
} else if (fixed_len > obj.val_len_) {
char *buf = (char *)cast_obj_ctx.cast_ctx_->allocator_v2_->alloc(fixed_len);
if (buf == nullptr) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("fail to allocate buf", K(fixed_len), KR(ret));
}
if (OB_SUCC(ret)) {
const ObCharsetType &cs = ObCharset::charset_type_by_coll(obj.get_collation_type());
char padding_char = (CHARSET_BINARY == cs) ? OB_PADDING_BINARY : OB_PADDING_CHAR;
MEMCPY(buf, obj.v_.ptr_, obj.val_len_);
if (CHARSET_BINARY == cs) {
MEMSET(buf + obj.val_len_, padding_char, fixed_len - obj.val_len_);
} else {
for (int i = 0; i < fixed_len - obj.val_len_; i += 2) {
*(buf + obj.val_len_ + i) = '\0';
*(buf + obj.val_len_ + i + 1) = padding_char;
}
}
obj.v_.ptr_ = buf;
obj.val_len_ = fixed_len;
}
}
if (OB_SUCC(ret)) {
const ObCharsetType &cs = ObCharset::charset_type_by_coll(obj.get_collation_type());
char padding_char = (CHARSET_BINARY == cs) ? OB_PADDING_BINARY : OB_PADDING_CHAR;
MEMCPY(buf, obj.v_.ptr_, obj.val_len_);
MEMSET(buf + obj.val_len_, padding_char, fixed_len - obj.val_len_);
obj.v_.ptr_ = buf;
obj.val_len_ = fixed_len;
} else {
int32_t fixed_len = column_schema->get_data_length();
if (fixed_len > obj.val_len_) {
char *buf = (char *)cast_obj_ctx.cast_ctx_->allocator_v2_->alloc(fixed_len);
if (buf == nullptr) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("fail to allocate buf", K(fixed_len), KR(ret));
}
if (OB_SUCC(ret)) {
const ObCharsetType &cs = ObCharset::charset_type_by_coll(obj.get_collation_type());
char padding_char = (CHARSET_BINARY == cs) ? OB_PADDING_BINARY : OB_PADDING_CHAR;
MEMCPY(buf, obj.v_.ptr_, obj.val_len_);
MEMSET(buf + obj.val_len_, padding_char, fixed_len - obj.val_len_);
obj.v_.ptr_ = buf;
obj.val_len_ = fixed_len;
}
}
}
}