mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-04-26 11:48:52 +03:00
257 lines
7.9 KiB
C++
257 lines
7.9 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_extract.cpp 3923 2013-06-19 21:43:06Z bwilkinson $
|
|
*
|
|
*
|
|
****************************************************************************/
|
|
|
|
#include <cstdlib>
|
|
#include <string>
|
|
using namespace std;
|
|
|
|
#include "functor_int.h"
|
|
#include "funchelpers.h"
|
|
#include "functioncolumn.h"
|
|
#include "intervalcolumn.h"
|
|
#include "rowgroup.h"
|
|
using namespace execplan;
|
|
|
|
#include "dataconvert.h"
|
|
|
|
namespace
|
|
{
|
|
using namespace funcexp;
|
|
|
|
long long dateGet(uint64_t time, IntervalColumn::interval_type unit, bool dateType)
|
|
{
|
|
uint64_t year = 0, month = 0, day = 0, hour = 0, min = 0, sec = 0, msec = 0;
|
|
|
|
if (dateType)
|
|
{
|
|
year = (uint32_t)((time >> 16) & 0xffff);
|
|
month = (uint32_t)((time >> 12) & 0xf);
|
|
day = (uint32_t)((time >> 6) & 0x3f);
|
|
}
|
|
else
|
|
{
|
|
year = (uint32_t)((time >> 48) & 0xffff);
|
|
month = (uint32_t)((time >> 44) & 0xf);
|
|
day = (uint32_t)((time >> 38) & 0x3f);
|
|
hour = (uint32_t)((time >> 32) & 0x3f);
|
|
min = (uint32_t)((time >> 26) & 0x3f);
|
|
sec = (uint32_t)((time >> 20) & 0x3f);
|
|
msec = (uint32_t)((time & 0xfffff));
|
|
}
|
|
|
|
switch (unit)
|
|
{
|
|
case IntervalColumn::INTERVAL_YEAR: return year;
|
|
|
|
case IntervalColumn::INTERVAL_MONTH: return month;
|
|
|
|
case IntervalColumn::INTERVAL_DAY: return day;
|
|
|
|
case IntervalColumn::INTERVAL_HOUR: return hour;
|
|
|
|
case IntervalColumn::INTERVAL_MINUTE: return min;
|
|
|
|
case IntervalColumn::INTERVAL_SECOND: return sec;
|
|
|
|
case IntervalColumn::INTERVAL_MICROSECOND: return msec;
|
|
|
|
case IntervalColumn::INTERVAL_QUARTER: return (month - 1) / 3 + 1;
|
|
|
|
case IntervalColumn::INTERVAL_WEEK: return helpers::calc_mysql_week(year, month, day, 0);
|
|
|
|
case IntervalColumn::INTERVAL_YEAR_MONTH: return (year * 100) + month;
|
|
|
|
case IntervalColumn::INTERVAL_DAY_HOUR: return (day * 100) + hour;
|
|
|
|
case IntervalColumn::INTERVAL_DAY_MINUTE: return (day * 10000) + (hour * 100) + min;
|
|
|
|
case IntervalColumn::INTERVAL_DAY_SECOND: return (day * 1000000) + (hour * 10000) + (min * 100) + sec;
|
|
|
|
case IntervalColumn::INTERVAL_HOUR_MINUTE: return (hour * 100) + min;
|
|
|
|
case IntervalColumn::INTERVAL_HOUR_SECOND: return (hour * 10000) + (min * 100) + sec;
|
|
|
|
case IntervalColumn::INTERVAL_MINUTE_SECOND: return (min * 100) + sec;
|
|
|
|
case IntervalColumn::INTERVAL_DAY_MICROSECOND:
|
|
return (((day * 1000000) + (hour * 10000) + (min * 100) + sec) * 1000000) + msec;
|
|
|
|
case IntervalColumn::INTERVAL_HOUR_MICROSECOND:
|
|
return (((hour * 10000) + (min * 100) + sec) * 1000000) + msec;
|
|
|
|
case IntervalColumn::INTERVAL_MINUTE_MICROSECOND: return (((min * 100) + sec) * 1000000) + msec;
|
|
|
|
case IntervalColumn::INTERVAL_SECOND_MICROSECOND: return (sec * 1000000) + msec;
|
|
|
|
default: throw runtime_error(std::string("unit type is not supported: ") + std::to_string(unit));
|
|
};
|
|
}
|
|
|
|
long long timeGet(uint64_t time, IntervalColumn::interval_type unit)
|
|
{
|
|
int32_t hour = 0, min = 0, sec = 0, msec = 0, day = 0;
|
|
|
|
min = (int32_t)((time >> 32) & 0xff);
|
|
sec = (int32_t)((time >> 24) & 0xff);
|
|
msec = (int32_t)((time & 0xfffff));
|
|
|
|
// If negative, mask so it doesn't turn positive
|
|
int64_t mask = 0;
|
|
|
|
if ((time >> 40) & 0x800)
|
|
mask = 0xfffffffffffff000;
|
|
|
|
hour = mask | ((time >> 40) & 0xfff);
|
|
|
|
if ((hour >= 0) && (time >> 63))
|
|
hour *= -1;
|
|
|
|
// Always positive!
|
|
day = abs(hour / 24);
|
|
|
|
switch (unit)
|
|
{
|
|
case IntervalColumn::INTERVAL_YEAR:
|
|
case IntervalColumn::INTERVAL_MONTH: return 0;
|
|
|
|
case IntervalColumn::INTERVAL_DAY: return day;
|
|
|
|
case IntervalColumn::INTERVAL_HOUR: return hour;
|
|
|
|
case IntervalColumn::INTERVAL_MINUTE: return min;
|
|
|
|
case IntervalColumn::INTERVAL_SECOND: return sec;
|
|
|
|
case IntervalColumn::INTERVAL_MICROSECOND: return msec;
|
|
|
|
case IntervalColumn::INTERVAL_QUARTER:
|
|
case IntervalColumn::INTERVAL_WEEK:
|
|
case IntervalColumn::INTERVAL_YEAR_MONTH: return 0;
|
|
|
|
case IntervalColumn::INTERVAL_DAY_HOUR: return (day * 100) + hour;
|
|
|
|
case IntervalColumn::INTERVAL_DAY_MINUTE: return (day * 10000) + (hour * 100) + min;
|
|
|
|
case IntervalColumn::INTERVAL_DAY_SECOND: return (day * 1000000) + (hour * 10000) + (min * 100) + sec;
|
|
|
|
case IntervalColumn::INTERVAL_HOUR_MINUTE: return (hour * 100) + min;
|
|
|
|
case IntervalColumn::INTERVAL_HOUR_SECOND: return (hour * 10000) + (min * 100) + sec;
|
|
|
|
case IntervalColumn::INTERVAL_MINUTE_SECOND: return (min * 100) + sec;
|
|
|
|
case IntervalColumn::INTERVAL_DAY_MICROSECOND:
|
|
return (((day * 1000000) + (hour * 10000) + (min * 100) + sec) * 1000000) + msec;
|
|
|
|
case IntervalColumn::INTERVAL_HOUR_MICROSECOND:
|
|
return (((hour * 10000) + (min * 100) + sec) * 1000000) + msec;
|
|
|
|
case IntervalColumn::INTERVAL_MINUTE_MICROSECOND: return (((min * 100) + sec) * 1000000) + msec;
|
|
|
|
case IntervalColumn::INTERVAL_SECOND_MICROSECOND: return (sec * 1000000) + msec;
|
|
|
|
default: throw runtime_error(std::string("unit type is not supported: ") + std::to_string(unit));
|
|
};
|
|
};
|
|
} // namespace
|
|
|
|
namespace funcexp
|
|
{
|
|
CalpontSystemCatalog::ColType Func_extract::operationType(FunctionParm& fp,
|
|
CalpontSystemCatalog::ColType& resultType)
|
|
{
|
|
return resultType;
|
|
}
|
|
|
|
int64_t Func_extract::getIntVal(rowgroup::Row& row, FunctionParm& parm, bool& isNull,
|
|
CalpontSystemCatalog::ColType& ct)
|
|
{
|
|
IntervalColumn::interval_type unit =
|
|
static_cast<IntervalColumn::interval_type>(parm[1]->data()->getIntVal(row, isNull));
|
|
uint64_t time;
|
|
bool isTime = false;
|
|
|
|
//@bug4678 handle conversion from non date/datetime datatype
|
|
switch (parm[0]->data()->resultType().colDataType)
|
|
{
|
|
case CalpontSystemCatalog::DATE:
|
|
case CalpontSystemCatalog::DATETIME: time = parm[0]->data()->getDatetimeIntVal(row, isNull); break;
|
|
|
|
case CalpontSystemCatalog::TIMESTAMP:
|
|
{
|
|
dataconvert::TimeStamp timestamp(parm[0]->data()->getTimestampIntVal(row, isNull));
|
|
int64_t seconds = timestamp.second;
|
|
dataconvert::MySQLTime m_time;
|
|
dataconvert::gmtSecToMySQLTime(seconds, m_time, ct.getTimeZone());
|
|
dataconvert::DateTime dt;
|
|
dt.year = m_time.year;
|
|
dt.month = m_time.month;
|
|
dt.day = m_time.day;
|
|
dt.hour = m_time.hour;
|
|
dt.minute = m_time.minute;
|
|
dt.second = m_time.second;
|
|
dt.msecond = timestamp.msecond;
|
|
time = *(reinterpret_cast<uint64_t*>(&dt));
|
|
break;
|
|
}
|
|
|
|
case CalpontSystemCatalog::TIME:
|
|
time = parm[0]->data()->getTimeIntVal(row, isNull);
|
|
isTime = true;
|
|
break;
|
|
|
|
case CalpontSystemCatalog::VARCHAR:
|
|
case CalpontSystemCatalog::CHAR:
|
|
case CalpontSystemCatalog::TEXT:
|
|
{
|
|
const string& val = parm[0]->data()->getStrVal(row, isNull);
|
|
time = dataconvert::DataConvert::stringToDatetime(val);
|
|
break;
|
|
}
|
|
|
|
case CalpontSystemCatalog::INT:
|
|
case CalpontSystemCatalog::TINYINT:
|
|
case CalpontSystemCatalog::MEDINT:
|
|
case CalpontSystemCatalog::BIGINT:
|
|
case CalpontSystemCatalog::SMALLINT:
|
|
{
|
|
int64_t val = parm[0]->data()->getIntVal(row, isNull);
|
|
time = dataconvert::DataConvert::intToDatetime(val);
|
|
break;
|
|
}
|
|
|
|
default: time = parm[0]->data()->getIntVal(row, isNull);
|
|
}
|
|
|
|
long long value;
|
|
|
|
if (isTime)
|
|
value = timeGet(time, unit);
|
|
else
|
|
value = dateGet(time, unit, false);
|
|
|
|
return value;
|
|
}
|
|
|
|
} // namespace funcexp
|