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

/* Copyright (C) 2014 InfiniDB, Inc.
   Copyright (C) 2016 MariaDB Corporaton

   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_from_unixtime.cpp 3923 2013-06-19 21:43:06Z bwilkinson $
*
*
****************************************************************************/

#include <unistd.h>
#include <cstdlib>
#include <string>
using namespace std;

#include "functor_dtm.h"
#include "funchelpers.h"
#include "functioncolumn.h"
#include "rowgroup.h"
using namespace execplan;

#include "dataconvert.h"
using namespace dataconvert;


namespace
{
using namespace funcexp;

DateTime getDateTime(rowgroup::Row& row,
                     FunctionParm& parm,
                     bool& isNull)
{
    int64_t val;

    switch (parm[0]->data()->resultType().colDataType)
    {
        case execplan::CalpontSystemCatalog::FLOAT:
        case execplan::CalpontSystemCatalog::DOUBLE:
        case execplan::CalpontSystemCatalog::DECIMAL:
        {
            double value = parm[0]->data()->getDoubleVal(row, isNull);

            if (value > 0)
                value += 0.5;
            else if (value < 0)
                value -= 0.5;

            val = (int64_t)value;
            break;
        }

        default:
            val = parm[0]->data()->getDatetimeIntVal(row, isNull);
    }

    if (val < 0 || val > helpers::TIMESTAMP_MAX_VALUE)
        return 0;

    DateTime dt;

    struct tm tmp_tm;
    time_t tmp_t = (time_t)val;
    localtime_r(&tmp_t, &tmp_tm);

    //to->neg=0;
    dt.year = (int64_t) ((tmp_tm.tm_year + 1900) % 10000);
    dt.month = (int64_t) tmp_tm.tm_mon + 1;
    dt.day = (int64_t) tmp_tm.tm_mday;
    dt.hour = (int64_t) tmp_tm.tm_hour;
    dt.minute = (int64_t) tmp_tm.tm_min;
    dt.second = (int64_t) tmp_tm.tm_sec;
    dt.msecond = 0;
    return dt;
}
}

namespace funcexp
{

CalpontSystemCatalog::ColType Func_from_unixtime::operationType( FunctionParm& fp, CalpontSystemCatalog::ColType& resultType )
{
    CalpontSystemCatalog::ColType ct;
    ct.colDataType = CalpontSystemCatalog::VARCHAR;
    ct.colWidth = 255;
    return ct;
}

string Func_from_unixtime::getStrVal(rowgroup::Row& row,
                                     FunctionParm& parm,
                                     bool& isNull,
                                     CalpontSystemCatalog::ColType&)
{
    DateTime dt = getDateTime(row, parm, isNull);

    if (*reinterpret_cast<int64_t*>(&dt) == 0)
    {
        isNull = true;
        return "";
    }

    if (parm.size() == 2)
    {
        const string& format = parm[1]->data()->getStrVal(row, isNull);
        return helpers::IDB_date_format(dt, format);
    }

    char buf[256] = {0};
    DataConvert::datetimeToString(*(reinterpret_cast<int64_t*>(&dt)), buf, 255);
    return string(buf, 255);
}

int32_t Func_from_unixtime::getDateIntVal(rowgroup::Row& row,
        FunctionParm& parm,
        bool& isNull,
        CalpontSystemCatalog::ColType& ct)
{
    return (((getDatetimeIntVal(row, parm, isNull, ct) >> 32) & 0xFFFFFFC0) | 0x3E);
}

int64_t Func_from_unixtime::getDatetimeIntVal(rowgroup::Row& row,
        FunctionParm& parm,
        bool& isNull,
        CalpontSystemCatalog::ColType& ct)
{
    DateTime dt = getDateTime(row, parm, isNull);

    if (*reinterpret_cast<int64_t*>(&dt) == 0)
    {
        isNull = true;
        return 0;
    }

    return *reinterpret_cast<int64_t*>(&dt);
}

int64_t Func_from_unixtime::getIntVal(rowgroup::Row& row,
                                      FunctionParm& parm,
                                      bool& isNull,
                                      CalpontSystemCatalog::ColType& ct)
{
    DateTime dt = getDateTime(row, parm, isNull);

    if (*reinterpret_cast<int64_t*>(&dt) == 0)
    {
        isNull = true;
        return 0;
    }

    char buf[32];  // actual string guaranteed to be 22
    snprintf( buf, 32, "%04d%02d%02d%02d%02d%02d",
              dt.year, dt.month, dt.day, dt.hour,
              dt.minute, dt.second );
    return atoll(buf);
}

double Func_from_unixtime::getDoubleVal(rowgroup::Row& row,
                                        FunctionParm& parm,
                                        bool& isNull,
                                        CalpontSystemCatalog::ColType& ct)
{
    if (parm.size() == 1)
    {
        DateTime dt = getDateTime(row, parm, isNull);

        if (*reinterpret_cast<int64_t*>(&dt) == 0)
        {
            isNull = true;
            return 0;
        }

        char buf[32];  // actual string guaranteed to be 22
        snprintf( buf, 32, "%04d%02d%02d%02d%02d%02d.%06d",
                  dt.year, dt.month, dt.day, dt.hour,
                  dt.minute, dt.second, dt.msecond );
        return atof(buf);
    }

    return (double) atoi(getStrVal(row, parm, isNull, ct).c_str());
}


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