/* 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_ltrim.cpp 3923 2013-06-19 21:43:06Z bwilkinson $ * * ****************************************************************************/ #include using namespace std; #include "functor_str.h" #include "functioncolumn.h" using namespace execplan; #include "rowgroup.h" using namespace rowgroup; #include "joblisttypes.h" using namespace joblist; namespace funcexp { CalpontSystemCatalog::ColType Func_ltrim_oracle::operationType(FunctionParm& fp, CalpontSystemCatalog::ColType& resultType) { // operation type is not used by this functor return fp[0]->data()->resultType(); } std::string Func_ltrim_oracle::getStrVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull, execplan::CalpontSystemCatalog::ColType& type) { CHARSET_INFO* cs = type.getCharset(); // The original string const auto& src = fp[0]->data()->getStrVal(row, isNull); if (isNull || src.length() < 1) { isNull = true; return ""; } // binLen represents the number of bytes in src size_t binLen = src.length(); const char* pos = src.str(); const char* end = pos + binLen; // strLen = the number of characters in src size_t strLen = cs->numchars(pos, end); // The trim characters. const string& trim = (fp.size() > 1 ? fp[1]->data()->getStrVal(row, isNull).safeString("") : " "); // binTLen represents the number of bytes in trim size_t binTLen = trim.length(); const char* posT = trim.c_str(); // strTLen = the number of characters in trim size_t strTLen = cs->numchars(posT, posT + binTLen); if (strTLen == 0 || strTLen > strLen) { isNull = src.length() < 1; return src.safeString(""); } if (binTLen == 1) { // If the trim string is 1 byte, don't waste cpu for memcmp while (pos < end && *pos == *posT) { ++pos; --binLen; } } else { while (pos + binTLen <= end && memcmp(pos, posT, binTLen) == 0) { pos += binTLen; binLen -= binTLen; } } // Turn back to a string std::string ret(pos, binLen); if (binLen == 0) { isNull = true; } return ret; } } // namespace funcexp