mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-04-26 11:48:52 +03:00
630 lines
14 KiB
C++
630 lines
14 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_nullif.cpp 2485 2011-04-05 19:01:13Z zzhu $
|
|
*
|
|
*
|
|
****************************************************************************/
|
|
|
|
#include <cstdlib>
|
|
#include <string>
|
|
#include <sstream>
|
|
using namespace std;
|
|
|
|
#include "functor_all.h"
|
|
#include "functioncolumn.h"
|
|
#include "rowgroup.h"
|
|
using namespace execplan;
|
|
|
|
#include "rowgroup.h"
|
|
using namespace rowgroup;
|
|
|
|
#include "errorcodes.h"
|
|
#include "idberrorinfo.h"
|
|
#include "errorids.h"
|
|
using namespace logging;
|
|
using namespace dataconvert;
|
|
|
|
#include "funchelpers.h"
|
|
#include "utils_utf8.h"
|
|
using namespace funcexp;
|
|
|
|
namespace funcexp
|
|
{
|
|
|
|
CalpontSystemCatalog::ColType Func_nullif::operationType( FunctionParm& fp, CalpontSystemCatalog::ColType& resultType )
|
|
{
|
|
return resultType;
|
|
}
|
|
|
|
int64_t Func_nullif::getIntVal(rowgroup::Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
execplan::CalpontSystemCatalog::ColType& op_ct)
|
|
{
|
|
int64_t exp1 = parm[0]->data()->getIntVal(row, isNull);
|
|
int64_t exp2 = 0;
|
|
|
|
switch (parm[1]->data()->resultType().colDataType)
|
|
{
|
|
case execplan::CalpontSystemCatalog::BIGINT:
|
|
case execplan::CalpontSystemCatalog::INT:
|
|
case execplan::CalpontSystemCatalog::MEDINT:
|
|
case execplan::CalpontSystemCatalog::TINYINT:
|
|
case execplan::CalpontSystemCatalog::SMALLINT:
|
|
case execplan::CalpontSystemCatalog::DOUBLE:
|
|
case execplan::CalpontSystemCatalog::FLOAT:
|
|
case execplan::CalpontSystemCatalog::DECIMAL:
|
|
case execplan::CalpontSystemCatalog::UDECIMAL:
|
|
case execplan::CalpontSystemCatalog::VARCHAR:
|
|
case execplan::CalpontSystemCatalog::CHAR:
|
|
{
|
|
exp2 = parm[1]->data()->getIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
case execplan::CalpontSystemCatalog::UBIGINT:
|
|
case execplan::CalpontSystemCatalog::UINT:
|
|
case execplan::CalpontSystemCatalog::UMEDINT:
|
|
case execplan::CalpontSystemCatalog::UTINYINT:
|
|
case execplan::CalpontSystemCatalog::USMALLINT:
|
|
{
|
|
exp2 = parm[1]->data()->getUintVal(row, isNull);
|
|
break;
|
|
}
|
|
case execplan::CalpontSystemCatalog::DATE:
|
|
{
|
|
if (parm[0]->data()->resultType().colDataType ==
|
|
execplan::CalpontSystemCatalog::DATETIME)
|
|
{
|
|
// NULLIF arg0 is DATETIME, arg1 is DATE,
|
|
// Upgrade arg1 to datetime
|
|
exp2 = parm[1]->data()->getDatetimeIntVal(row, isNull);
|
|
}
|
|
else
|
|
{
|
|
// NULLIF arg0 is not DATETIME (might be date), arg1 is not DATE
|
|
// Use arg1 as DATE
|
|
exp2 = parm[1]->data()->getDateIntVal(row, isNull);
|
|
}
|
|
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::DATETIME:
|
|
{
|
|
exp2 = parm[1]->data()->getDatetimeIntVal(row, isNull);
|
|
if (parm[0]->data()->resultType().colDataType ==
|
|
execplan::CalpontSystemCatalog::DATE)
|
|
{
|
|
// NULLIF arg0 is DATE, arg1 is DATETIME,
|
|
// Upgrade arg1 to datetime
|
|
// When comparing exp1 as a Date, we can't simply promote. We have
|
|
// to be careful of the return value in case of not null return.
|
|
int64_t exp1 = parm[0]->data()->getDatetimeIntVal(row, isNull);
|
|
if ( exp1 == exp2 )
|
|
{
|
|
isNull = true;
|
|
return 0;
|
|
}
|
|
// since exp1 here is inside the block, when we leave the block, the
|
|
// original (Date) value is restored.
|
|
}
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
isNull = true;
|
|
}
|
|
}
|
|
|
|
if ( exp1 == exp2 ) {
|
|
isNull = true;
|
|
return 0;
|
|
}
|
|
|
|
return exp1;
|
|
}
|
|
|
|
uint64_t Func_nullif::getUintVal(rowgroup::Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
execplan::CalpontSystemCatalog::ColType& op_ct)
|
|
{
|
|
uint64_t exp1 = parm[0]->data()->getUintVal(row, isNull);
|
|
uint64_t exp2 = 0;
|
|
|
|
switch (parm[1]->data()->resultType().colDataType)
|
|
{
|
|
case execplan::CalpontSystemCatalog::BIGINT:
|
|
case execplan::CalpontSystemCatalog::INT:
|
|
case execplan::CalpontSystemCatalog::MEDINT:
|
|
case execplan::CalpontSystemCatalog::TINYINT:
|
|
case execplan::CalpontSystemCatalog::SMALLINT:
|
|
case execplan::CalpontSystemCatalog::DOUBLE:
|
|
case execplan::CalpontSystemCatalog::UDOUBLE:
|
|
case execplan::CalpontSystemCatalog::FLOAT:
|
|
case execplan::CalpontSystemCatalog::UFLOAT:
|
|
case execplan::CalpontSystemCatalog::DECIMAL:
|
|
case execplan::CalpontSystemCatalog::UDECIMAL:
|
|
case execplan::CalpontSystemCatalog::VARCHAR:
|
|
case execplan::CalpontSystemCatalog::CHAR:
|
|
{
|
|
int64_t iexp2 = parm[1]->data()->getIntVal(row, isNull);
|
|
if (isNull || iexp2 < 0) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
exp2 = iexp2;
|
|
break;
|
|
}
|
|
case execplan::CalpontSystemCatalog::UBIGINT:
|
|
case execplan::CalpontSystemCatalog::UINT:
|
|
case execplan::CalpontSystemCatalog::UMEDINT:
|
|
case execplan::CalpontSystemCatalog::UTINYINT:
|
|
case execplan::CalpontSystemCatalog::USMALLINT:
|
|
{
|
|
exp2 = parm[1]->data()->getUintVal(row, isNull);
|
|
break;
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::DATE:
|
|
{
|
|
exp2 = parm[1]->data()->getDateIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::DATETIME:
|
|
{
|
|
exp2 = parm[1]->data()->getDatetimeIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
isNull = true;
|
|
}
|
|
}
|
|
|
|
if ( exp1 == exp2 ) {
|
|
isNull = true;
|
|
return 0;
|
|
}
|
|
|
|
return exp1;
|
|
}
|
|
|
|
string Func_nullif::getStrVal(rowgroup::Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
CalpontSystemCatalog::ColType& op_ct)
|
|
{
|
|
string exp1 = parm[0]->data()->getStrVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return "";
|
|
}
|
|
|
|
string exp2 = parm[1]->data()->getStrVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
|
|
int datatype0 = parm[0]->data()->resultType().colDataType;
|
|
int datatype1 = parm[1]->data()->resultType().colDataType;
|
|
|
|
if ( datatype0 == execplan::CalpontSystemCatalog::DATE &&
|
|
datatype1 == execplan::CalpontSystemCatalog::DATETIME )
|
|
{
|
|
exp1 = exp1 + " 00:00:00";
|
|
}
|
|
|
|
if ( datatype1 == execplan::CalpontSystemCatalog::DATE &&
|
|
datatype0 == execplan::CalpontSystemCatalog::DATETIME )
|
|
{
|
|
exp2 = exp2 + " 00:00:00";
|
|
}
|
|
|
|
if ( utf8::idb_strcoll(exp1.c_str() , exp2.c_str()) == 0 ) {
|
|
isNull = true;
|
|
return "";
|
|
}
|
|
|
|
return parm[0]->data()->getStrVal(row, isNull);
|
|
}
|
|
|
|
int32_t Func_nullif::getDateIntVal(rowgroup::Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
CalpontSystemCatalog::ColType& ct)
|
|
{
|
|
int64_t exp1 = parm[0]->data()->getDateIntVal(row, isNull);
|
|
int64_t exp2 = 0;
|
|
|
|
switch (parm[1]->data()->resultType().colDataType)
|
|
{
|
|
case execplan::CalpontSystemCatalog::BIGINT:
|
|
case execplan::CalpontSystemCatalog::INT:
|
|
case execplan::CalpontSystemCatalog::MEDINT:
|
|
case execplan::CalpontSystemCatalog::TINYINT:
|
|
case execplan::CalpontSystemCatalog::SMALLINT:
|
|
case execplan::CalpontSystemCatalog::DOUBLE:
|
|
case execplan::CalpontSystemCatalog::FLOAT:
|
|
case execplan::CalpontSystemCatalog::DECIMAL:
|
|
case execplan::CalpontSystemCatalog::VARCHAR:
|
|
case execplan::CalpontSystemCatalog::CHAR:
|
|
{
|
|
exp2 = parm[1]->data()->getIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
case execplan::CalpontSystemCatalog::DATE:
|
|
{
|
|
exp2 = parm[1]->data()->getDateIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::DATETIME:
|
|
{
|
|
exp2 = parm[1]->data()->getDatetimeIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
isNull = true;
|
|
}
|
|
}
|
|
|
|
if ( exp1 == exp2 ) {
|
|
isNull = true;
|
|
return 0;
|
|
}
|
|
|
|
return exp1;
|
|
}
|
|
|
|
int64_t Func_nullif::getDatetimeIntVal(rowgroup::Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
CalpontSystemCatalog::ColType& ct)
|
|
{
|
|
int64_t exp1 = parm[0]->data()->getDatetimeIntVal(row, isNull);
|
|
int64_t exp2 = 0;
|
|
|
|
switch (parm[1]->data()->resultType().colDataType)
|
|
{
|
|
case execplan::CalpontSystemCatalog::BIGINT:
|
|
case execplan::CalpontSystemCatalog::INT:
|
|
case execplan::CalpontSystemCatalog::MEDINT:
|
|
case execplan::CalpontSystemCatalog::TINYINT:
|
|
case execplan::CalpontSystemCatalog::SMALLINT:
|
|
case execplan::CalpontSystemCatalog::DOUBLE:
|
|
case execplan::CalpontSystemCatalog::FLOAT:
|
|
case execplan::CalpontSystemCatalog::DECIMAL:
|
|
case execplan::CalpontSystemCatalog::VARCHAR:
|
|
case execplan::CalpontSystemCatalog::CHAR:
|
|
{
|
|
exp2 = parm[1]->data()->getIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
case execplan::CalpontSystemCatalog::DATE:
|
|
{
|
|
exp2 = parm[1]->data()->getDateIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::DATETIME:
|
|
{
|
|
exp2 = parm[1]->data()->getDatetimeIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
isNull = true;
|
|
}
|
|
}
|
|
|
|
if ( exp1 == exp2 ) {
|
|
isNull = true;
|
|
return 0;
|
|
}
|
|
|
|
return exp1;
|
|
}
|
|
|
|
double Func_nullif::getDoubleVal(rowgroup::Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
execplan::CalpontSystemCatalog::ColType& op_ct)
|
|
{
|
|
double exp1 = parm[0]->data()->getDoubleVal(row, isNull);
|
|
double exp2 = 0;
|
|
|
|
switch (parm[1]->data()->resultType().colDataType)
|
|
{
|
|
case execplan::CalpontSystemCatalog::BIGINT:
|
|
case execplan::CalpontSystemCatalog::INT:
|
|
case execplan::CalpontSystemCatalog::MEDINT:
|
|
case execplan::CalpontSystemCatalog::TINYINT:
|
|
case execplan::CalpontSystemCatalog::SMALLINT:
|
|
case execplan::CalpontSystemCatalog::DOUBLE:
|
|
case execplan::CalpontSystemCatalog::FLOAT:
|
|
case execplan::CalpontSystemCatalog::DECIMAL:
|
|
case execplan::CalpontSystemCatalog::VARCHAR:
|
|
case execplan::CalpontSystemCatalog::CHAR:
|
|
{
|
|
exp2 = parm[1]->data()->getDoubleVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
case execplan::CalpontSystemCatalog::DATE:
|
|
{
|
|
exp2 = parm[1]->data()->getDateIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::DATETIME:
|
|
{
|
|
exp2 = parm[1]->data()->getDatetimeIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
isNull = true;
|
|
}
|
|
}
|
|
|
|
if ( exp1 == exp2 ) {
|
|
isNull = true;
|
|
return 0;
|
|
}
|
|
|
|
return exp1;
|
|
}
|
|
|
|
|
|
execplan::IDB_Decimal Func_nullif::getDecimalVal(rowgroup::Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
execplan::CalpontSystemCatalog::ColType& op_ct)
|
|
{
|
|
IDB_Decimal exp1 = parm[0]->data()->getDecimalVal(row, isNull);
|
|
IDB_Decimal exp2;
|
|
|
|
switch (parm[1]->data()->resultType().colDataType)
|
|
{
|
|
case execplan::CalpontSystemCatalog::BIGINT:
|
|
case execplan::CalpontSystemCatalog::INT:
|
|
case execplan::CalpontSystemCatalog::MEDINT:
|
|
case execplan::CalpontSystemCatalog::TINYINT:
|
|
case execplan::CalpontSystemCatalog::SMALLINT:
|
|
case execplan::CalpontSystemCatalog::UBIGINT:
|
|
case execplan::CalpontSystemCatalog::UINT:
|
|
case execplan::CalpontSystemCatalog::UMEDINT:
|
|
case execplan::CalpontSystemCatalog::UTINYINT:
|
|
case execplan::CalpontSystemCatalog::USMALLINT:
|
|
case execplan::CalpontSystemCatalog::DECIMAL:
|
|
case execplan::CalpontSystemCatalog::UDECIMAL:
|
|
{
|
|
exp2 = parm[1]->data()->getDecimalVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::VARCHAR:
|
|
case execplan::CalpontSystemCatalog::CHAR:
|
|
{
|
|
int64_t value = parm[1]->data()->getIntVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
|
|
exp2.value = value;
|
|
exp2.scale = 0;
|
|
}
|
|
|
|
case execplan::CalpontSystemCatalog::DOUBLE:
|
|
case execplan::CalpontSystemCatalog::UDOUBLE:
|
|
case execplan::CalpontSystemCatalog::FLOAT:
|
|
case execplan::CalpontSystemCatalog::UFLOAT:
|
|
{
|
|
double value = parm[1]->data()->getDoubleVal(row, isNull);
|
|
if (isNull) {
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
|
|
exp2.value = (int64_t) value;
|
|
exp2.scale = 0;
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::DATE:
|
|
{
|
|
int32_t s = 0;
|
|
int64_t x = 0;
|
|
|
|
string value = DataConvert::dateToString1(parm[1]->data()->getDateIntVal(row, isNull));
|
|
if (!isNull)
|
|
{
|
|
x = atoll(value.c_str());
|
|
|
|
if ( s > 11 )
|
|
s = 0;
|
|
if ( s > 0 )
|
|
{
|
|
x *= helpers::powerOf10_c[s];
|
|
}
|
|
else if (s < 0)
|
|
{
|
|
s = -s;
|
|
if ( s >= (int32_t) value.size() )
|
|
{
|
|
x = 0;
|
|
}
|
|
else
|
|
{
|
|
x /= helpers::powerOf10_c[s];
|
|
x *= helpers::powerOf10_c[s];
|
|
}
|
|
|
|
s = 0;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
|
|
exp2.value = x;
|
|
exp2.scale = s;
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::DATETIME:
|
|
{
|
|
int32_t s = 0;
|
|
int64_t x = 0;
|
|
|
|
string value =
|
|
DataConvert::datetimeToString1(parm[1]->data()->getDatetimeIntVal(row, isNull));
|
|
if (!isNull)
|
|
{
|
|
//strip off micro seconds
|
|
value = value.substr(0,14);
|
|
int64_t x = atoll(value.c_str());
|
|
|
|
if ( s > 5 )
|
|
s = 0;
|
|
if ( s > 0 )
|
|
{
|
|
x *= helpers::powerOf10_c[s];
|
|
}
|
|
else if (s < 0)
|
|
{
|
|
s = -s;
|
|
if ( s >= (int32_t) value.size() )
|
|
{
|
|
x = 0;
|
|
}
|
|
else
|
|
{
|
|
x /= helpers::powerOf10_c[s];
|
|
x *= helpers::powerOf10_c[s];
|
|
}
|
|
|
|
s = 0;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
isNull = false;
|
|
return exp1;
|
|
}
|
|
|
|
exp2.value = x;
|
|
exp2.scale = s;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
{
|
|
std::ostringstream oss;
|
|
oss << "truncate: datatype of " << execplan::colDataTypeToString(op_ct.colDataType);
|
|
throw logging::IDBExcept(oss.str(), ERR_DATATYPE_NOT_SUPPORT);
|
|
}
|
|
}
|
|
|
|
if ( exp1 == exp2 ) {
|
|
isNull = true;
|
|
IDB_Decimal decimal;
|
|
decimal.value = 0;
|
|
decimal.scale = 0;
|
|
return decimal;
|
|
}
|
|
|
|
return exp1;
|
|
}
|
|
|
|
|
|
} // namespace funcexp
|
|
// vim:ts=4 sw=4:
|