database/mariadb-columnstore-engine
1
0
Fork 0
mirror of https://github.com/mariadb-corporation/mariadb-columnstore-engine.git synced 2025-07-29 08:21:15 +03:00
Code Activity

MCOL-3536 Collation

Browse Source
This commit is contained in:
David Hall
2020-06-03 19:43:53 -05:00
parent 39a93ef753
commit 889094a23d
16 changed files with 211 additions and 151 deletions
Download Patch File Download Diff File Expand all files Collapse all files

83
utils/funcexp/func_insert.cpp
View File

@ -20,6 +20,9 @@
*
*
****************************************************************************/
#include <mariadb.h>
#undef set_bits // mariadb.h defines set_bits, which is incompatible with boost
#include <my_sys.h>
#include <string>
using namespace std;
@ -48,69 +51,61 @@ CalpontSystemCatalog::ColType Func_insert::operationType(FunctionParm& fp, Calpo
return fp[0]->data()->resultType();
}
string insertStr(const string& src, int pos, int len, const string& targ)
{
int64_t strLen = static_cast<int64_t>(src.length());
if ((pos <= 0) || ((pos - 1) >= strLen))
return src;
if ((len < 0) || (len > strLen))
len = strLen;
const char* srcptr = src.c_str();
advance(srcptr, pos - 1, srcptr + strLen);
// srcptr now pointing to where we need to insert targ string
uint32_t srcPos = srcptr - src.c_str();
uint32_t finPos = strLen;
const char* finptr = src.c_str();
if ((strLen - (pos - 1 + len)) >= 0)
{
advance(finptr, (pos - 1 + len), finptr + strLen);
// finptr now pointing to the end of the string to replace
finPos = finptr - src.c_str();
}
string out;
out.reserve(srcPos + targ.length() + strLen - finPos + 1);
out.append( src.c_str(), srcPos );
out.append( targ.c_str(), targ.length() );
out.append( src.c_str() + finPos, strLen - finPos );
return out;
}
std::string Func_insert::getStrVal(rowgroup::Row& row,
FunctionParm& fp,
bool& isNull,
execplan::CalpontSystemCatalog::ColType&)
{
string tstr;
string src;
string tnewstr;
stringValue(fp[0], row, isNull, tstr);
int64_t start, length;
stringValue(fp[0], row, isNull, src);
if (isNull)
{
return "";
}
stringValue(fp[3], row, isNull, tnewstr);
if (isNull)
return "";
int64_t pos = fp[1]->data()->getIntVal(row, isNull);
start = fp[1]->data()->getIntVal(row, isNull);
if (isNull)
return "";
int64_t len = fp[2]->data()->getIntVal(row, isNull);
length = fp[2]->data()->getIntVal(row, isNull);
if (isNull)
return "";
return insertStr( tstr, pos, len, tnewstr );
start--; // Because SQL syntax is 1 based and we want 0 based.
CHARSET_INFO* cs = fp[0]->data()->resultType().getCharset();
// binLen represents the number of bytes
int64_t binLen = static_cast<int64_t>(src.length());
const char* pos = src.c_str();
const char* end = pos + binLen;
// strLen is number of characters
int64_t strLen = cs->numchars(pos, end);
// Return the original string if start isn't within the string.
if ((start <= 1) || start >= strLen)
return src;
if ((length < 0) || (length > strLen))
length = strLen;
// Convert start and length from characters to bytes.
start = cs->charpos(pos, end, start);
length = cs->charpos(pos+start, end, length);
string out;
out.reserve(binLen - length + tnewstr.length() + 1);
out.append(src.c_str(), start);
out.append(tnewstr.c_str(), tnewstr.length());
out.append(src.c_str() + start + length, binLen - start - length);
return out;
}