/* 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_nullif.cpp 2485 2011-04-05 19:01:13Z zzhu $ * * ****************************************************************************/ #include #include #include 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" 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: case execplan::CalpontSystemCatalog::TEXT: { 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) || (parm[0]->data()->resultType().colDataType == execplan::CalpontSystemCatalog::DATETIME)) { // 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; } case execplan::CalpontSystemCatalog::TIMESTAMP: { exp2 = parm[1]->data()->getTimestampIntVal(row, isNull); if ((parm[0]->data()->resultType().colDataType == execplan::CalpontSystemCatalog::TIMESTAMP) || (parm[0]->data()->resultType().colDataType == execplan::CalpontSystemCatalog::DATETIME)) { // NULLIF arg0 is DATETIME, arg1 is TIMESTAMP, // Upgrade arg0 to timestamp // 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()->getTimestampIntVal(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; } case execplan::CalpontSystemCatalog::TIME: { exp2 = parm[1]->data()->getTimeIntVal(row, isNull); if ((parm[0]->data()->resultType().colDataType == execplan::CalpontSystemCatalog::DATETIME) || (parm[0]->data()->resultType().colDataType == execplan::CalpontSystemCatalog::TIME)) { // NULLIF arg0 is DATETIME, arg1 is TIME, // Upgrade arg1 to time // 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()->getTimeIntVal(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: case execplan::CalpontSystemCatalog::TEXT: { 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; } case execplan::CalpontSystemCatalog::TIMESTAMP: { exp2 = parm[1]->data()->getTimestampIntVal(row, isNull); if (isNull) { isNull = false; return exp1; } break; } case execplan::CalpontSystemCatalog::TIME: { exp2 = parm[1]->data()->getTimeIntVal(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).safeString(""); CHARSET_INFO* cs = parm[0]->data()->resultType().getCharset(); if (isNull) { // NULLIF(NULL, ...) is NULL, according to server's results. return ""; } string exp2 = parm[1]->data()->getStrVal(row, isNull).safeString(""); 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 || datatype1 == execplan::CalpontSystemCatalog::TIMESTAMP)) { exp1 = exp1 + " 00:00:00"; } if (datatype1 == execplan::CalpontSystemCatalog::DATE && (datatype0 == execplan::CalpontSystemCatalog::DATETIME || datatype0 == execplan::CalpontSystemCatalog::TIMESTAMP)) { exp2 = exp2 + " 00:00:00"; } if (cs->strnncoll(exp1.c_str(), exp1.length(), exp2.c_str(), exp2.length()) == 0) { isNull = true; return ""; } return parm[0]->data()->getStrVal(row, isNull).safeString(""); } 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::UDECIMAL: case execplan::CalpontSystemCatalog::VARCHAR: case execplan::CalpontSystemCatalog::CHAR: case execplan::CalpontSystemCatalog::TEXT: { 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; } case execplan::CalpontSystemCatalog::TIMESTAMP: { exp2 = parm[1]->data()->getTimestampIntVal(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::UDECIMAL: case execplan::CalpontSystemCatalog::VARCHAR: case execplan::CalpontSystemCatalog::CHAR: case execplan::CalpontSystemCatalog::TEXT: case execplan::CalpontSystemCatalog::DATE: { // Upgrade to datetime for proper comparison exp2 = parm[1]->data()->getDatetimeIntVal(row, isNull); if (isNull) { isNull = false; return exp1; } break; } break; case execplan::CalpontSystemCatalog::TIME: case execplan::CalpontSystemCatalog::DATETIME: case execplan::CalpontSystemCatalog::TIMESTAMP: { 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::getTimeIntVal(rowgroup::Row& row, FunctionParm& parm, bool& isNull, CalpontSystemCatalog::ColType& ct) { int64_t exp1 = parm[0]->data()->getTimeIntVal(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: case execplan::CalpontSystemCatalog::TEXT: case execplan::CalpontSystemCatalog::TIME: case execplan::CalpontSystemCatalog::DATETIME: case execplan::CalpontSystemCatalog::TIMESTAMP: { exp2 = parm[1]->data()->getTimeIntVal(row, isNull); if (isNull) { isNull = false; return exp1; } break; } default: { isNull = false; return exp1; } } if (exp1 == exp2) { isNull = true; return 0; } return exp1; } int64_t Func_nullif::getTimestampIntVal(rowgroup::Row& row, FunctionParm& parm, bool& isNull, CalpontSystemCatalog::ColType& ct) { int64_t exp1 = parm[0]->data()->getTimestampIntVal(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: case execplan::CalpontSystemCatalog::TEXT: case execplan::CalpontSystemCatalog::TIME: case execplan::CalpontSystemCatalog::DATETIME: case execplan::CalpontSystemCatalog::TIMESTAMP: { exp2 = parm[1]->data()->getTimestampIntVal(row, isNull); if (isNull) { isNull = false; return exp1; } break; } default: { isNull = false; return exp1; } } 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::UDECIMAL: case execplan::CalpontSystemCatalog::VARCHAR: case execplan::CalpontSystemCatalog::CHAR: case execplan::CalpontSystemCatalog::TEXT: { exp2 = parm[1]->data()->getDoubleVal(row, isNull); if (isNull) { isNull = false; return exp1; } break; } case execplan::CalpontSystemCatalog::LONGDOUBLE: { exp2 = (double)parm[1]->data()->getLongDoubleVal(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::TIME: 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; } long double Func_nullif::getLongDoubleVal(rowgroup::Row& row, FunctionParm& parm, bool& isNull, execplan::CalpontSystemCatalog::ColType& op_ct) { long double exp1 = parm[0]->data()->getLongDoubleVal(row, isNull); long 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::UDECIMAL: case execplan::CalpontSystemCatalog::VARCHAR: case execplan::CalpontSystemCatalog::CHAR: case execplan::CalpontSystemCatalog::TEXT: { exp2 = parm[1]->data()->getDoubleVal(row, isNull); if (isNull) { isNull = false; return exp1; } break; } case execplan::CalpontSystemCatalog::LONGDOUBLE: { exp2 = parm[1]->data()->getLongDoubleVal(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::TIME: case execplan::CalpontSystemCatalog::DATETIME: { exp2 = parm[1]->data()->getDatetimeIntVal(row, isNull); if (isNull) { isNull = false; return exp1; } break; } case execplan::CalpontSystemCatalog::TIMESTAMP: { exp2 = parm[1]->data()->getTimestampIntVal(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: case execplan::CalpontSystemCatalog::TEXT: { int64_t value = parm[1]->data()->getIntVal(row, isNull); if (isNull) { isNull = false; return exp1; } exp2.value = value; exp2.scale = 0; } /* fall through */ 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::TIME: case execplan::CalpontSystemCatalog::DATETIME: case execplan::CalpontSystemCatalog::TIMESTAMP: { int32_t s = 0; int64_t x = 0; string value; if (parm[1]->data()->resultType().colDataType == execplan::CalpontSystemCatalog::TIMESTAMP) value = DataConvert::timestampToString1(parm[1]->data()->getTimestampIntVal(row, isNull), op_ct.getTimeZone()); else value = DataConvert::datetimeToString1(parm[1]->data()->getDatetimeIntVal(row, isNull)); if (!isNull) { // strip off micro seconds value = value.substr(0, 14); 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