mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-06-07 19:22:02 +03:00
88 lines
2.5 KiB
C++
88 lines
2.5 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_instr.cpp 3645 2013-03-19 13:10:24Z rdempsey $
|
|
*
|
|
*
|
|
****************************************************************************/
|
|
|
|
#include <cstdlib>
|
|
#include <string>
|
|
#include <sstream>
|
|
using namespace std;
|
|
|
|
#include "functor_int.h"
|
|
#include "functioncolumn.h"
|
|
#include "rowgroup.h"
|
|
using namespace execplan;
|
|
|
|
namespace funcexp
|
|
{
|
|
CalpontSystemCatalog::ColType Func_instr::operationType(FunctionParm& /*fp*/,
|
|
CalpontSystemCatalog::ColType& /*resultType*/)
|
|
{
|
|
CalpontSystemCatalog::ColType ct;
|
|
ct.colDataType = CalpontSystemCatalog::VARCHAR;
|
|
ct.colWidth = 255;
|
|
return ct;
|
|
}
|
|
|
|
int64_t Func_instr::getIntVal(rowgroup::Row& row, FunctionParm& parm, bool& isNull,
|
|
CalpontSystemCatalog::ColType& colType)
|
|
{
|
|
int64_t start = 0;
|
|
int64_t start0 = 0;
|
|
my_match_t match;
|
|
|
|
const auto& str = parm[0]->data()->getStrVal(row, isNull);
|
|
if (str.isNull())
|
|
return 0;
|
|
const char* s1 = str.str();
|
|
uint32_t l1 = (uint32_t)str.length();
|
|
|
|
const auto& substr = parm[1]->data()->getStrVal(row, isNull);
|
|
if (substr.isNull())
|
|
return 0;
|
|
|
|
const char* s2 = substr.str();
|
|
uint32_t l2 = (uint32_t)substr.length();
|
|
if (l2 < 1)
|
|
return start + 1;
|
|
|
|
CHARSET_INFO* cs = colType.getCharset();
|
|
|
|
if (parm.size() == 3)
|
|
{
|
|
start0 = start = parm[2]->data()->getIntVal(row, isNull) - 1;
|
|
|
|
if ((start < 0) || (start > l1))
|
|
return 0;
|
|
|
|
start = (int64_t)cs->charpos(s1, s1 + l1, start); // adjust start for multi-byte
|
|
|
|
if (start + l2 > l1) // Substring is longer than str at pos.
|
|
return 0;
|
|
}
|
|
|
|
if (!cs->instr(s1 + start, l1 - start, s2, l2, &match, 1))
|
|
return 0;
|
|
return (int64_t)match.mb_len + start0 + 1;
|
|
}
|
|
|
|
} // namespace funcexp
|