mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-04-23 07:05:36 +03:00
381 lines
12 KiB
C++
381 lines
12 KiB
C++
/* Copyright (C) 2014 InfiniDB, Inc.
|
|
Copyright (C) 2019 MariaDB Corporation
|
|
|
|
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_between.cpp 3954 2013-07-08 16:30:15Z bpaul $
|
|
*
|
|
*
|
|
****************************************************************************/
|
|
|
|
#include <cstdlib>
|
|
#include <string>
|
|
using namespace std;
|
|
|
|
#include "functor_bool.h"
|
|
#include "functioncolumn.h"
|
|
#include "predicateoperator.h"
|
|
#include "constantcolumn.h"
|
|
using namespace execplan;
|
|
|
|
#include "rowgroup.h"
|
|
|
|
#include "errorcodes.h"
|
|
#include "idberrorinfo.h"
|
|
#include "errorids.h"
|
|
|
|
using namespace logging;
|
|
|
|
namespace
|
|
{
|
|
template <typename result_t>
|
|
inline bool numericGE(result_t op1, result_t op2)
|
|
{
|
|
return op1 >= op2;
|
|
}
|
|
|
|
template <typename result_t>
|
|
inline bool numericLE(result_t op1, result_t op2)
|
|
{
|
|
return op1 <= op2;
|
|
}
|
|
|
|
inline bool strGE(CHARSET_INFO& cs, const string& op1, const string& op2)
|
|
{
|
|
return cs.strnncoll(op1.c_str(), op1.length(), op2.c_str(), op2.length()) >= 0;
|
|
}
|
|
|
|
inline bool strLE(CHARSET_INFO& cs, const string& op1, const string& op2)
|
|
{
|
|
return cs.strnncoll(op1.c_str(), op1.length(), op2.c_str(), op2.length()) <= 0;
|
|
}
|
|
|
|
inline bool getBool(rowgroup::Row& row, funcexp::FunctionParm& pm, bool& isNull,
|
|
CalpontSystemCatalog::ColType& ct, bool notBetween)
|
|
{
|
|
switch (ct.colDataType)
|
|
{
|
|
case execplan::CalpontSystemCatalog::BIGINT:
|
|
case execplan::CalpontSystemCatalog::INT:
|
|
case execplan::CalpontSystemCatalog::MEDINT:
|
|
case execplan::CalpontSystemCatalog::TINYINT:
|
|
case execplan::CalpontSystemCatalog::SMALLINT:
|
|
{
|
|
int64_t val = pm[0]->data()->getIntVal(row, isNull);
|
|
|
|
if (notBetween)
|
|
{
|
|
if (!numericGE(val, pm[1]->data()->getIntVal(row, isNull)) && !isNull)
|
|
return true;
|
|
|
|
isNull = false;
|
|
return (!numericLE(val, pm[2]->data()->getIntVal(row, isNull)) && !isNull);
|
|
}
|
|
|
|
return !isNull && numericGE(val, pm[1]->data()->getIntVal(row, isNull)) &&
|
|
numericLE(val, pm[2]->data()->getIntVal(row, isNull)) && !isNull;
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::UBIGINT:
|
|
case execplan::CalpontSystemCatalog::UINT:
|
|
case execplan::CalpontSystemCatalog::UMEDINT:
|
|
case execplan::CalpontSystemCatalog::UTINYINT:
|
|
case execplan::CalpontSystemCatalog::USMALLINT:
|
|
{
|
|
uint64_t val = pm[0]->data()->getUintVal(row, isNull);
|
|
|
|
if (notBetween)
|
|
{
|
|
if (!numericGE(val, pm[1]->data()->getUintVal(row, isNull)) && !isNull)
|
|
return true;
|
|
|
|
isNull = false;
|
|
return (!numericLE(val, pm[2]->data()->getUintVal(row, isNull)) && !isNull);
|
|
}
|
|
|
|
return !isNull && numericGE(val, pm[1]->data()->getUintVal(row, isNull)) &&
|
|
numericLE(val, pm[2]->data()->getUintVal(row, isNull)) && !isNull;
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::DATE:
|
|
{
|
|
int32_t val = pm[0]->data()->getDateIntVal(row, isNull);
|
|
|
|
if (notBetween)
|
|
{
|
|
if (!numericGE(val, pm[1]->data()->getDateIntVal(row, isNull)) && !isNull)
|
|
return true;
|
|
|
|
isNull = false;
|
|
return (!numericLE(val, pm[2]->data()->getDateIntVal(row, isNull)) && !isNull);
|
|
}
|
|
|
|
return !isNull && numericGE(val, pm[1]->data()->getDateIntVal(row, isNull)) &&
|
|
numericLE(val, pm[2]->data()->getDateIntVal(row, isNull));
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::DATETIME:
|
|
{
|
|
int64_t val = pm[0]->data()->getDatetimeIntVal(row, isNull);
|
|
|
|
if (notBetween)
|
|
{
|
|
if (!numericGE(val, pm[1]->data()->getDatetimeIntVal(row, isNull)) && !isNull)
|
|
return true;
|
|
|
|
isNull = false;
|
|
return (!numericLE(val, pm[2]->data()->getDatetimeIntVal(row, isNull)) && !isNull);
|
|
}
|
|
|
|
return !isNull && numericGE(val, pm[1]->data()->getDatetimeIntVal(row, isNull)) &&
|
|
numericLE(val, pm[2]->data()->getDatetimeIntVal(row, isNull));
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::TIMESTAMP:
|
|
{
|
|
int64_t val = pm[0]->data()->getTimestampIntVal(row, isNull);
|
|
|
|
if (notBetween)
|
|
{
|
|
if (!numericGE(val, pm[1]->data()->getTimestampIntVal(row, isNull)) && !isNull)
|
|
return true;
|
|
|
|
isNull = false;
|
|
return (!numericLE(val, pm[2]->data()->getTimestampIntVal(row, isNull)) && !isNull);
|
|
}
|
|
|
|
return !isNull && numericGE(val, pm[1]->data()->getTimestampIntVal(row, isNull)) &&
|
|
numericLE(val, pm[2]->data()->getTimestampIntVal(row, isNull));
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::TIME:
|
|
{
|
|
// Shift out unused day for compare
|
|
int64_t val = pm[0]->data()->getTimeIntVal(row, isNull) << 12;
|
|
|
|
if (notBetween)
|
|
{
|
|
if (!numericGE(val, pm[1]->data()->getTimeIntVal(row, isNull) << 12) && !isNull)
|
|
return true;
|
|
|
|
isNull = false;
|
|
return (!numericLE(val, pm[2]->data()->getTimeIntVal(row, isNull) << 12) && !isNull);
|
|
}
|
|
|
|
return !isNull && numericGE(val, pm[1]->data()->getTimeIntVal(row, isNull) << 12) &&
|
|
numericLE(val, pm[2]->data()->getTimeIntVal(row, isNull) << 12);
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::DOUBLE:
|
|
case execplan::CalpontSystemCatalog::UDOUBLE:
|
|
case execplan::CalpontSystemCatalog::FLOAT:
|
|
case execplan::CalpontSystemCatalog::UFLOAT:
|
|
{
|
|
double val = pm[0]->data()->getDoubleVal(row, isNull);
|
|
|
|
if (notBetween)
|
|
{
|
|
if (!numericGE(val, pm[1]->data()->getDoubleVal(row, isNull)) && !isNull)
|
|
return true;
|
|
|
|
isNull = false;
|
|
return (!numericLE(val, pm[2]->data()->getDoubleVal(row, isNull)) && !isNull);
|
|
}
|
|
|
|
return !isNull && numericGE(val, pm[1]->data()->getDoubleVal(row, isNull)) &&
|
|
numericLE(val, pm[2]->data()->getDoubleVal(row, isNull));
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::LONGDOUBLE:
|
|
{
|
|
long double val = pm[0]->data()->getLongDoubleVal(row, isNull);
|
|
|
|
if (notBetween)
|
|
{
|
|
if (!numericGE(val, pm[1]->data()->getLongDoubleVal(row, isNull)) && !isNull)
|
|
return true;
|
|
|
|
isNull = false;
|
|
return (!numericLE(val, pm[2]->data()->getLongDoubleVal(row, isNull)) && !isNull);
|
|
}
|
|
|
|
return !isNull && numericGE(val, pm[1]->data()->getLongDoubleVal(row, isNull)) &&
|
|
numericLE(val, pm[2]->data()->getLongDoubleVal(row, isNull));
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::DECIMAL:
|
|
case execplan::CalpontSystemCatalog::UDECIMAL:
|
|
{
|
|
IDB_Decimal val = pm[0]->data()->getDecimalVal(row, isNull);
|
|
|
|
if (isNull)
|
|
return false;
|
|
|
|
if (notBetween)
|
|
{
|
|
if (!numericGE(val, pm[1]->data()->getDecimalVal(row, isNull)) && !isNull)
|
|
return true;
|
|
|
|
isNull = false;
|
|
return (!numericLE(val, pm[2]->data()->getDecimalVal(row, isNull)) && !isNull);
|
|
}
|
|
|
|
return numericGE(val, pm[1]->data()->getDecimalVal(row, isNull)) &&
|
|
numericLE(val, pm[2]->data()->getDecimalVal(row, isNull));
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::VARCHAR: // including CHAR'
|
|
case execplan::CalpontSystemCatalog::CHAR:
|
|
case execplan::CalpontSystemCatalog::TEXT:
|
|
{
|
|
const string& val = pm[0]->data()->getStrVal(row, isNull).safeString("");
|
|
auto charset = datatypes::Charset(ct.charsetNumber);
|
|
CHARSET_INFO& cs = charset.getCharset();
|
|
|
|
if (notBetween)
|
|
{
|
|
if (!strGE(cs, val, pm[1]->data()->getStrVal(row, isNull).safeString("")) && !isNull)
|
|
return true;
|
|
|
|
isNull = false;
|
|
return (!strLE(cs, val, pm[2]->data()->getStrVal(row, isNull).safeString("")) && !isNull);
|
|
}
|
|
|
|
return !isNull && strGE(cs, val, pm[1]->data()->getStrVal(row, isNull).safeString("")) &&
|
|
strLE(cs, val, pm[2]->data()->getStrVal(row, isNull).safeString(""));
|
|
}
|
|
break; // XXX: gcc falsely complains here.
|
|
|
|
default:
|
|
{
|
|
std::ostringstream oss;
|
|
oss << "between: datatype of " << execplan::colDataTypeToString(ct.colDataType);
|
|
throw logging::IDBExcept(oss.str(), ERR_DATATYPE_NOT_SUPPORT);
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|
|
namespace funcexp
|
|
{
|
|
CalpontSystemCatalog::ColType Func_between::operationType(FunctionParm& fp,
|
|
CalpontSystemCatalog::ColType& resultType)
|
|
{
|
|
PredicateOperator op;
|
|
CalpontSystemCatalog::ColType ct = fp[0]->data()->resultType();
|
|
// op.operationType(fp[0]->data()->resultType());
|
|
bool allString = true;
|
|
|
|
for (uint32_t i = 1; i < fp.size(); i++)
|
|
{
|
|
// op.setOpType(op.operationType(), fp[i]->data()->resultType());
|
|
op.setOpType(ct, fp[i]->data()->resultType());
|
|
ct = op.operationType();
|
|
|
|
if ((fp[i]->data()->resultType().colDataType != CalpontSystemCatalog::CHAR &&
|
|
fp[i]->data()->resultType().colDataType != CalpontSystemCatalog::TEXT &&
|
|
fp[i]->data()->resultType().colDataType != CalpontSystemCatalog::VARCHAR) ||
|
|
ct.colDataType == CalpontSystemCatalog::DATE || ct.colDataType == CalpontSystemCatalog::DATETIME ||
|
|
ct.colDataType == CalpontSystemCatalog::TIMESTAMP || ct.colDataType == CalpontSystemCatalog::TIME)
|
|
{
|
|
allString = false;
|
|
}
|
|
}
|
|
|
|
if (allString)
|
|
{
|
|
ct.colDataType = CalpontSystemCatalog::VARCHAR;
|
|
ct.colWidth = 255;
|
|
}
|
|
|
|
else if (op.operationType().colDataType == CalpontSystemCatalog::DATETIME)
|
|
{
|
|
ConstantColumn* cc = NULL;
|
|
|
|
for (uint32_t i = 1; i < fp.size(); i++)
|
|
{
|
|
cc = dynamic_cast<ConstantColumn*>(fp[i]->data());
|
|
|
|
if (cc)
|
|
{
|
|
Result result = cc->result();
|
|
result.intVal = dataconvert::DataConvert::datetimeToInt(result.strVal.safeString(""));
|
|
cc->result(result);
|
|
}
|
|
}
|
|
}
|
|
else if (op.operationType().colDataType == CalpontSystemCatalog::TIMESTAMP)
|
|
{
|
|
ConstantColumn* cc = NULL;
|
|
|
|
for (uint32_t i = 1; i < fp.size(); i++)
|
|
{
|
|
cc = dynamic_cast<ConstantColumn*>(fp[i]->data());
|
|
|
|
if (cc)
|
|
{
|
|
Result result = cc->result();
|
|
result.intVal = dataconvert::DataConvert::timestampToInt(result.strVal.safeString(""), resultType.getTimeZone());
|
|
cc->result(result);
|
|
}
|
|
}
|
|
}
|
|
else if (op.operationType().colDataType == CalpontSystemCatalog::TIME)
|
|
{
|
|
ConstantColumn* cc = NULL;
|
|
|
|
for (uint32_t i = 1; i < fp.size(); i++)
|
|
{
|
|
cc = dynamic_cast<ConstantColumn*>(fp[i]->data());
|
|
|
|
if (cc)
|
|
{
|
|
Result result = cc->result();
|
|
result.intVal = dataconvert::DataConvert::timeToInt(result.strVal.safeString(""));
|
|
cc->result(result);
|
|
}
|
|
}
|
|
}
|
|
|
|
return ct;
|
|
}
|
|
|
|
bool Func_between::getBoolVal(rowgroup::Row& row, FunctionParm& pm, bool& isNull,
|
|
CalpontSystemCatalog::ColType& ct)
|
|
{
|
|
return getBool(row, pm, isNull, ct, false) && !isNull;
|
|
}
|
|
|
|
CalpontSystemCatalog::ColType Func_notbetween::operationType(FunctionParm& fp,
|
|
CalpontSystemCatalog::ColType& resultType)
|
|
{
|
|
PredicateOperator* op = new PredicateOperator();
|
|
CalpontSystemCatalog::ColType ct;
|
|
op->setOpType(fp[0]->data()->resultType(), fp[1]->data()->resultType());
|
|
op->setOpType(op->resultType(), fp[2]->data()->resultType());
|
|
return op->operationType();
|
|
}
|
|
|
|
bool Func_notbetween::getBoolVal(rowgroup::Row& row, FunctionParm& pm, bool& isNull,
|
|
CalpontSystemCatalog::ColType& ct)
|
|
{
|
|
return getBool(row, pm, isNull, ct, true) && !isNull;
|
|
}
|
|
|
|
} // namespace funcexp
|