mariadb-columnstore-engine/utils/funcexp/func_extract.cpp

/* 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 )
{
	uint32_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/4+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_SECOND_MICROSECOND: return (sec*1000000)+msec; 
        default:
			throw runtime_error("unit type is not supported: " + unit);
	};
}
}

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;	
	
	//@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::VARCHAR:
	case CalpontSystemCatalog::CHAR:
	{
		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 = dateGet( time, unit, false );

	return value;
}


} // namespace funcexp
// vim:ts=4 sw=4: