You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-11-03 17:13:17 +03:00
441 lines
9.0 KiB
C++
441 lines
9.0 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_truncate.cpp 3921 2013-06-19 18:59:56Z bwilkinson $
|
|
|
|
#include <cstdlib>
|
|
#include <iomanip>
|
|
#include <string>
|
|
using namespace std;
|
|
|
|
#include "functor_real.h"
|
|
#include "functioncolumn.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"
|
|
|
|
namespace
|
|
{
|
|
|
|
using namespace funcexp;
|
|
|
|
inline void decimalPlaceDouble(FunctionParm& fp, int64_t& s, double& p, Row& row, bool& isNull)
|
|
{
|
|
s = fp[1]->data()->getIntVal(row, isNull);
|
|
int64_t d = s;
|
|
if (isNull)
|
|
return;
|
|
|
|
int64_t i = (d >= 0) ? d : (-d);
|
|
int64_t r = 1;
|
|
while (i--)
|
|
r *= 10;
|
|
|
|
if (d >= 0)
|
|
p = (double) r;
|
|
else
|
|
p = 1.0 / ((double) r);
|
|
}
|
|
|
|
}
|
|
|
|
namespace funcexp
|
|
{
|
|
|
|
CalpontSystemCatalog::ColType Func_truncate::operationType(FunctionParm& fp, CalpontSystemCatalog::ColType& resultType)
|
|
{
|
|
if (resultType.colDataType == execplan::CalpontSystemCatalog::DECIMAL ||
|
|
resultType.colDataType == execplan::CalpontSystemCatalog::UDECIMAL)
|
|
{
|
|
CalpontSystemCatalog::ColType ct = fp[0]->data()->resultType();
|
|
switch (ct.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:
|
|
{
|
|
if (resultType.scale > ct.scale)
|
|
(resultType).scale = ct.scale;
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
return fp[0]->data()->resultType();
|
|
}
|
|
|
|
|
|
// truncate(X, D)
|
|
//
|
|
|
|
int64_t Func_truncate::getIntVal(Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
CalpontSystemCatalog::ColType& op_ct)
|
|
{
|
|
IDB_Decimal x = getDecimalVal(row, parm, isNull, op_ct);
|
|
|
|
if (x.scale > 0)
|
|
{
|
|
while (x.scale-- > 0)
|
|
x.value /= 10;
|
|
}
|
|
else
|
|
{
|
|
while (x.scale++ < 0)
|
|
x.value *= 10;
|
|
}
|
|
|
|
return x.value;
|
|
}
|
|
|
|
|
|
uint64_t Func_truncate::getUintVal(Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
CalpontSystemCatalog::ColType& op_ct)
|
|
{
|
|
return parm[0]->data()->getUintVal(row, isNull);
|
|
}
|
|
|
|
|
|
double Func_truncate::getDoubleVal(Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
CalpontSystemCatalog::ColType& op_ct)
|
|
{
|
|
if (execplan::CalpontSystemCatalog::DOUBLE == op_ct.colDataType ||
|
|
execplan::CalpontSystemCatalog::FLOAT == op_ct.colDataType)
|
|
{
|
|
int64_t d = 0;
|
|
double p = 1;
|
|
decimalPlaceDouble(parm, d, p, row, isNull);
|
|
|
|
if (isNull)
|
|
return 0.0;
|
|
|
|
double x = parm[0]->data()->getDoubleVal(row, isNull);
|
|
if (!isNull)
|
|
{
|
|
x *= p;
|
|
|
|
if (x > 0)
|
|
x = floor(x);
|
|
else
|
|
x = ceil(x);
|
|
|
|
if (p != 0.0)
|
|
x /= p;
|
|
else
|
|
x = 0.0;
|
|
}
|
|
|
|
return x;
|
|
}
|
|
|
|
IDB_Decimal x = getDecimalVal(row, parm, isNull, op_ct);
|
|
if (isNull)
|
|
return 0.0;
|
|
|
|
double d = x.value;
|
|
if (x.scale > 0)
|
|
{
|
|
while (x.scale-- > 0)
|
|
d /= 10.0;
|
|
}
|
|
else
|
|
{
|
|
while (x.scale++ < 0)
|
|
d *= 10.0;
|
|
}
|
|
|
|
return d;
|
|
}
|
|
|
|
|
|
IDB_Decimal Func_truncate::getDecimalVal(Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
CalpontSystemCatalog::ColType& op_ct)
|
|
{
|
|
IDB_Decimal decimal;
|
|
|
|
switch (op_ct.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:
|
|
{
|
|
int64_t d = 0;
|
|
//@Bug 3101 - GCC 4.5.1 optimizes too aggressively here. Mark as volatile.
|
|
volatile int64_t p = 1;
|
|
decimal = parm[0]->data()->getDecimalVal(row, isNull);
|
|
if (!isNull)
|
|
{
|
|
int64_t nvp = p;
|
|
d = parm[1]->data()->getIntVal(row, isNull);
|
|
if (!isNull)
|
|
helpers::decimalPlaceDec(d, nvp, decimal.scale);
|
|
p = nvp;
|
|
}
|
|
|
|
if (isNull)
|
|
break;
|
|
|
|
int64_t x = decimal.value;
|
|
if (d > 0)
|
|
{
|
|
x = x * p;
|
|
}
|
|
else if (d < 0)
|
|
{
|
|
if ((x >= p) || (x <= -p))
|
|
{
|
|
if (p != 0)
|
|
x = x / p;
|
|
else
|
|
x = 0;
|
|
}
|
|
else
|
|
{
|
|
x = 0;
|
|
}
|
|
}
|
|
|
|
// negative scale is not supported by CNX yet, set d to 0.
|
|
if (decimal.scale < 0)
|
|
{
|
|
do
|
|
x *= 10;
|
|
while (++decimal.scale < 0);
|
|
}
|
|
decimal.value = x;
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::DOUBLE:
|
|
case execplan::CalpontSystemCatalog::UDOUBLE:
|
|
case execplan::CalpontSystemCatalog::FLOAT:
|
|
case execplan::CalpontSystemCatalog::UFLOAT:
|
|
case execplan::CalpontSystemCatalog::VARCHAR:
|
|
case execplan::CalpontSystemCatalog::CHAR:
|
|
{
|
|
int64_t s = 0;
|
|
double p = 1;
|
|
decimalPlaceDouble(parm, s, p, row, isNull);
|
|
|
|
if (isNull)
|
|
break;
|
|
|
|
double x = parm[0]->data()->getDoubleVal(row, isNull);
|
|
if (!isNull)
|
|
{
|
|
x *= p;
|
|
decimal.value = (int64_t) x;
|
|
decimal.scale = s;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::DATE:
|
|
{
|
|
int32_t s = 0;
|
|
int64_t x = 0;
|
|
|
|
string value = DataConvert::dateToString1(parm[0]->data()->getDateIntVal(row, isNull));
|
|
|
|
s = parm[1]->data()->getIntVal(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;
|
|
}
|
|
}
|
|
|
|
decimal.value = x;
|
|
decimal.scale = s;
|
|
}
|
|
break;
|
|
|
|
case execplan::CalpontSystemCatalog::DATETIME:
|
|
{
|
|
int32_t s = 0;
|
|
int64_t x = 0;
|
|
|
|
string value =
|
|
DataConvert::datetimeToString1(parm[0]->data()->getDatetimeIntVal(row, isNull));
|
|
|
|
s = parm[1]->data()->getIntVal(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;
|
|
}
|
|
}
|
|
|
|
decimal.value = x;
|
|
decimal.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);
|
|
}
|
|
}
|
|
|
|
return decimal;
|
|
}
|
|
|
|
|
|
string Func_truncate::getStrVal(Row& row,
|
|
FunctionParm& parm,
|
|
bool& isNull,
|
|
CalpontSystemCatalog::ColType& op_ct)
|
|
{
|
|
IDB_Decimal x = getDecimalVal(row, parm, isNull, op_ct);
|
|
|
|
switch (op_ct.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:
|
|
if (x.scale != 0)
|
|
{
|
|
if (x.scale > 0 && x.scale < 19)
|
|
{
|
|
x.value /= IDB_pow[x.scale];
|
|
}
|
|
else if (x.scale < 0 && x.scale > -19)
|
|
{
|
|
x.value *= IDB_pow[-x.scale]; // may overflow
|
|
}
|
|
else if (x.scale > 0)
|
|
{
|
|
x.value = 0;
|
|
}
|
|
else // overflow may need throw exception
|
|
{
|
|
int64_t e = -x.scale % 18;
|
|
x.value *= IDB_pow[e];
|
|
e = -x.scale - e;
|
|
while (e > 0)
|
|
{
|
|
x.value *= IDB_pow[18];
|
|
e -= 18;
|
|
}
|
|
}
|
|
|
|
x.scale = 0;
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return dataconvert::DataConvert::decimalToString(x.value, x.scale, op_ct.colDataType);
|
|
}
|
|
|
|
|
|
} // namespace funcexp
|
|
// vim:ts=4 sw=4:
|