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

Add new interface tolower toupper in ObCharset

Browse Source
This commit is contained in:
wjhh2008
2023-01-11 03:24:50 +00:00
committed by ob-robot
parent c88aab6b2d
commit 5830b64c41
4 changed files with 152 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
deps/oblib/src/lib/charset/ob_charset.cpp vendored
View File

@ -2336,6 +2336,74 @@ size_t ObCharset::caseup(const ObCollationType collation_type, ObString &src)
return size;
}
int ObCharset::toupper(const ObCollationType collation_type,
const ObString &src, ObString &dst,
ObIAllocator &allocator)
{
int ret = OB_SUCCESS;
const ObCharsetInfo *cs_info = NULL;
if (OB_ISNULL(cs_info = get_charset(collation_type))) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid collation type", K(ret), K(collation_type));
} else {
int casemulti = cs_info->caseup_multiply;
if (1 == casemulti) {
if (OB_FAIL(ob_write_string(allocator, src, dst))) {
LOG_WARN("fail to copy string", K(ret), K(src));
} else {
size_t size = cs_info->cset->caseup(cs_info, dst.ptr(), dst.length(), dst.ptr(), dst.length());
dst.assign_ptr(dst.ptr(), static_cast<ObString::obstr_size_t>(size));
}
} else {
char *buf = NULL;
int64_t buf_len = src.length() * casemulti;
if (OB_ISNULL(buf = static_cast<char*>(allocator.alloc(buf_len)))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("fail to alloc memory", K(ret));
} else {
size_t size = cs_info->cset->caseup(cs_info, const_cast<char*>(src.ptr()), src.length(), buf, buf_len);
dst.assign_ptr(buf, static_cast<ObString::obstr_size_t>(size));
}
}
}
return ret;
}
int ObCharset::tolower(const ObCollationType collation_type,
const ObString &src, ObString &dst,
ObIAllocator &allocator)
{
int ret = OB_SUCCESS;
const ObCharsetInfo *cs_info = NULL;
if (OB_ISNULL(cs_info = get_charset(collation_type))) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid collation type", K(ret), K(collation_type));
} else {
int casemulti = cs_info->casedn_multiply;
if (1 == casemulti) {
if (OB_FAIL(ob_write_string(allocator, src, dst))) {
LOG_WARN("fail to copy string", K(ret), K(src));
} else {
size_t size = cs_info->cset->casedn(cs_info, dst.ptr(), dst.length(), dst.ptr(), dst.length());
dst.assign_ptr(dst.ptr(), static_cast<ObString::obstr_size_t>(size));
}
} else {
char *buf = NULL;
int64_t buf_len = src.length() * casemulti;
if (OB_ISNULL(buf = static_cast<char*>(allocator.alloc(buf_len)))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("fail to alloc memory", K(ret));
} else {
size_t size = cs_info->cset->casedn(cs_info, const_cast<char*>(src.ptr()), src.length(), buf, buf_len);
dst.assign_ptr(buf, static_cast<ObString::obstr_size_t>(size));
}
}
}
return ret;
}
bool ObCharset::case_insensitive_equal(const ObString &one,
const ObString &another,
const ObCollationType &collation_type) {

11
deps/oblib/src/lib/charset/ob_charset.h vendored
View File

@ -405,9 +405,20 @@ public:
static int strcmp(const ObCollationType collation_type,
const ObString &l_str,
const ObString &r_str);
//these interface is not safe:
//when invoke this, if ObString a = "134"; this func will core; so avoid passing src as a style
//if collation type is gb18030, this func will die
//**Please** use toupper and tolower instead of casedn and caseup
static size_t casedn(const ObCollationType collation_type, ObString &src);
static size_t caseup(const ObCollationType collation_type, ObString &src);
static int toupper(const ObCollationType collation_type,
const ObString &src, ObString &dst,
ObIAllocator &allocator);
static int tolower(const ObCollationType collation_type,
const ObString &src, ObString &dst,
ObIAllocator &allocator);
static bool case_insensitive_equal(const ObString &one,
const ObString &another,
const ObCollationType &collation_type = CS_TYPE_UTF8MB4_GENERAL_CI);