mariadb-columnstore-engine/utils/funcexp/func_get_format.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_get_format.cpp 2665 2011-06-01 20:42:52Z rdempsey $
 *
 *
 ****************************************************************************/

#include <cstdlib>
#include <string>
using namespace std;

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

#include "dataconvert.h"
using namespace dataconvert;

class to_upper
{
 public:
  char operator()(char c) const  // notice the return type
  {
    return toupper(c);
  }
};

namespace funcexp
{
string known_date_time_formats[5][4] = {{"USA", "%m.%d.%Y", "%Y-%m-%d %H.%i.%s", "%h:%i:%s %p"},
                                        {"JIS", "%Y-%m-%d", "%Y-%m-%d %H:%i:%s", "%H:%i:%s"},
                                        {"ISO", "%Y-%m-%d", "%Y-%m-%d %H:%i:%s", "%H:%i:%s"},
                                        {"EUR", "%d.%m.%Y", "%Y-%m-%d %H.%i.%s", "%H.%i.%s"},
                                        {"INTERNAL", "%Y%m%d", "%Y%m%d%H%i%s", "%H%i%s"}};

string know_types[3] = {"DATE", "DATETIME", "TIME"};

CalpontSystemCatalog::ColType Func_get_format::operationType(FunctionParm& /*fp*/,
                                                             CalpontSystemCatalog::ColType& resultType)
{
  return resultType;
}

string Func_get_format::getStrVal(rowgroup::Row& row, FunctionParm& parm, bool& isNull,
                                  CalpontSystemCatalog::ColType&)
{
  // parm[0] -- format
  // parm[1] -- type
  string format = parm[0]->data()->getStrVal(row, isNull).safeString("");

  if (isNull)
    return "";

  transform(format.begin(), format.end(), format.begin(), to_upper());

  string type = parm[1]->data()->getStrVal(row, isNull).safeString("");

  if (isNull)
    return "";

  transform(type.begin(), type.end(), type.begin(), to_upper());

  int itype = 0;

  for (; itype < 3; itype++)
  {
    if (know_types[itype] == type)
      break;
  }

  // check for match
  if (itype == 3)
    return "";

  for (int i = 0; i < 5; i++)
  {
    if (known_date_time_formats[i][0] == format)
    {
      switch (itype)
      {
        case 0: return known_date_time_formats[i][2]; break;

        default: isNull = true; return "";
      }
    }
  }

  return "";
}

}  // namespace funcexp