mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-05-31 11:41:14 +03:00
114 lines
2.9 KiB
C++
114 lines
2.9 KiB
C++
/* Copyright (C) 2014 InfiniDB, Inc.
|
|
|
|
This program is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU General Public License
|
|
as published by the Free Software Foundation; version 2 of
|
|
the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
MA 02110-1301, USA. */
|
|
|
|
/****************************************************************************
|
|
* $Id: func_concat_ws.cpp 3714 2013-04-18 16:29:25Z bpaul $
|
|
*
|
|
*
|
|
****************************************************************************/
|
|
|
|
#include <string>
|
|
using namespace std;
|
|
|
|
#include "functor_str.h"
|
|
#include "functioncolumn.h"
|
|
#include "utils_utf8.h"
|
|
using namespace execplan;
|
|
|
|
#include "rowgroup.h"
|
|
using namespace rowgroup;
|
|
|
|
#define STRCOLL_ENH__
|
|
|
|
namespace funcexp
|
|
{
|
|
|
|
CalpontSystemCatalog::ColType Func_concat_ws::operationType(FunctionParm& fp, CalpontSystemCatalog::ColType& resultType )
|
|
{
|
|
// operation type is not used by this functor
|
|
return fp[0]->data()->resultType();
|
|
}
|
|
|
|
|
|
string Func_concat_ws::getStrVal(Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
CalpontSystemCatalog::ColType&)
|
|
{
|
|
string delim = stringValue(parm[0], row, isNull);
|
|
if (isNull)
|
|
return "";
|
|
|
|
#ifdef STRCOLL_ENH__
|
|
wstring wstr;
|
|
size_t strwclen = utf8::idb_mbstowcs(0,delim.c_str(),0) + 1;
|
|
wchar_t* wcbuf = (wchar_t*)alloca(strwclen * sizeof(wchar_t));
|
|
strwclen = utf8::idb_mbstowcs(wcbuf, delim.c_str(), strwclen);
|
|
wstring wdelim(wcbuf, strwclen);
|
|
|
|
for ( unsigned int id = 1 ; id < parm.size() ; id++)
|
|
{
|
|
string tstr = stringValue(parm[id], row, isNull);
|
|
if (isNull)
|
|
{
|
|
isNull = false;
|
|
continue;
|
|
}
|
|
if (!wstr.empty())
|
|
wstr += wdelim;
|
|
size_t strwclen1 = utf8::idb_mbstowcs(0, tstr.c_str(), 0) + 1;
|
|
wchar_t* wcbuf1 = (wchar_t*)alloca(strwclen1 * sizeof(wchar_t));
|
|
strwclen1 = utf8::idb_mbstowcs(wcbuf1, tstr.c_str(), strwclen1);
|
|
wstring str1(wcbuf1, strwclen1);
|
|
wstr += str1;
|
|
}
|
|
|
|
size_t strmblen = utf8::idb_wcstombs(0, wstr.c_str(), 0) + 1;
|
|
char* outbuf = (char*)alloca(strmblen * sizeof(char));
|
|
strmblen = utf8::idb_wcstombs(outbuf, wstr.c_str(), strmblen);
|
|
if (strmblen == 0)
|
|
isNull = true;
|
|
else
|
|
isNull = false;
|
|
return string(outbuf, strmblen);
|
|
|
|
#else
|
|
string str;
|
|
for ( uint32_t i = 1 ; i < parm.size() ; i++)
|
|
{
|
|
str += string(stringValue(parm[i], row, isNull).c_str());
|
|
|
|
if (isNull)
|
|
{
|
|
isNull = false;
|
|
continue;
|
|
}
|
|
if (!str.empty() && !isNull)
|
|
str += delim;
|
|
}
|
|
if (str.empty())
|
|
isNull = true;
|
|
else
|
|
isNull = false;
|
|
return str;
|
|
#endif
|
|
}
|
|
|
|
|
|
} // namespace funcexp
|
|
// vim:ts=4 sw=4:
|