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

/* Copyright (C) 2014 InfiniDB, Inc.
   Copyright (C) 2019 MariaDB Corporation

   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_between.cpp 3954 2013-07-08 16:30:15Z bpaul $
 *
 *
 ****************************************************************************/

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

#include "functor_bool.h"
#include "functioncolumn.h"
#include "predicateoperator.h"
#include "constantcolumn.h"
using namespace execplan;

#include "rowgroup.h"

#include "errorcodes.h"
#include "idberrorinfo.h"
#include "errorids.h"

using namespace logging;

namespace
{
template <typename result_t>
inline bool numericGE(result_t op1, result_t op2)
{
  return op1 >= op2;
}

template <typename result_t>
inline bool numericLE(result_t op1, result_t op2)
{
  return op1 <= op2;
}

inline bool strGE(CHARSET_INFO& cs, const string& op1, const string& op2)
{
  return cs.strnncoll(op1.c_str(), op1.length(), op2.c_str(), op2.length()) >= 0;
}

inline bool strLE(CHARSET_INFO& cs, const string& op1, const string& op2)
{
  return cs.strnncoll(op1.c_str(), op1.length(), op2.c_str(), op2.length()) <= 0;
}

inline bool getBool(rowgroup::Row& row, funcexp::FunctionParm& pm, bool& isNull,
                    CalpontSystemCatalog::ColType& ct, bool notBetween)
{
  switch (ct.colDataType)
  {
    case execplan::CalpontSystemCatalog::BIGINT:
    case execplan::CalpontSystemCatalog::INT:
    case execplan::CalpontSystemCatalog::MEDINT:
    case execplan::CalpontSystemCatalog::TINYINT:
    case execplan::CalpontSystemCatalog::SMALLINT:
    {
      int64_t val = pm[0]->data()->getIntVal(row, isNull);

      if (notBetween)
      {
        if (!numericGE(val, pm[1]->data()->getIntVal(row, isNull)) && !isNull)
          return true;

        isNull = false;
        return (!numericLE(val, pm[2]->data()->getIntVal(row, isNull)) && !isNull);
      }

      return !isNull && numericGE(val, pm[1]->data()->getIntVal(row, isNull)) &&
             numericLE(val, pm[2]->data()->getIntVal(row, isNull)) && !isNull;
    }

    case execplan::CalpontSystemCatalog::UBIGINT:
    case execplan::CalpontSystemCatalog::UINT:
    case execplan::CalpontSystemCatalog::UMEDINT:
    case execplan::CalpontSystemCatalog::UTINYINT:
    case execplan::CalpontSystemCatalog::USMALLINT:
    {
      uint64_t val = pm[0]->data()->getUintVal(row, isNull);

      if (notBetween)
      {
        if (!numericGE(val, pm[1]->data()->getUintVal(row, isNull)) && !isNull)
          return true;

        isNull = false;
        return (!numericLE(val, pm[2]->data()->getUintVal(row, isNull)) && !isNull);
      }

      return !isNull && numericGE(val, pm[1]->data()->getUintVal(row, isNull)) &&
             numericLE(val, pm[2]->data()->getUintVal(row, isNull)) && !isNull;
    }

    case execplan::CalpontSystemCatalog::DATE:
    {
      int32_t val = pm[0]->data()->getDateIntVal(row, isNull);

      if (notBetween)
      {
        if (!numericGE(val, pm[1]->data()->getDateIntVal(row, isNull)) && !isNull)
          return true;

        isNull = false;
        return (!numericLE(val, pm[2]->data()->getDateIntVal(row, isNull)) && !isNull);
      }

      return !isNull && numericGE(val, pm[1]->data()->getDateIntVal(row, isNull)) &&
             numericLE(val, pm[2]->data()->getDateIntVal(row, isNull));
    }

    case execplan::CalpontSystemCatalog::DATETIME:
    {
      int64_t val = pm[0]->data()->getDatetimeIntVal(row, isNull);

      if (notBetween)
      {
        if (!numericGE(val, pm[1]->data()->getDatetimeIntVal(row, isNull)) && !isNull)
          return true;

        isNull = false;
        return (!numericLE(val, pm[2]->data()->getDatetimeIntVal(row, isNull)) && !isNull);
      }

      return !isNull && numericGE(val, pm[1]->data()->getDatetimeIntVal(row, isNull)) &&
             numericLE(val, pm[2]->data()->getDatetimeIntVal(row, isNull));
    }

    case execplan::CalpontSystemCatalog::TIMESTAMP:
    {
      int64_t val = pm[0]->data()->getTimestampIntVal(row, isNull);

      if (notBetween)
      {
        if (!numericGE(val, pm[1]->data()->getTimestampIntVal(row, isNull)) && !isNull)
          return true;

        isNull = false;
        return (!numericLE(val, pm[2]->data()->getTimestampIntVal(row, isNull)) && !isNull);
      }

      return !isNull && numericGE(val, pm[1]->data()->getTimestampIntVal(row, isNull)) &&
             numericLE(val, pm[2]->data()->getTimestampIntVal(row, isNull));
    }

    case execplan::CalpontSystemCatalog::TIME:
    {
      // Shift out unused day for compare
      int64_t val = pm[0]->data()->getTimeIntVal(row, isNull) << 12;

      if (notBetween)
      {
        if (!numericGE(val, pm[1]->data()->getTimeIntVal(row, isNull) << 12) && !isNull)
          return true;

        isNull = false;
        return (!numericLE(val, pm[2]->data()->getTimeIntVal(row, isNull) << 12) && !isNull);
      }

      return !isNull && numericGE(val, pm[1]->data()->getTimeIntVal(row, isNull) << 12) &&
             numericLE(val, pm[2]->data()->getTimeIntVal(row, isNull) << 12);
    }

    case execplan::CalpontSystemCatalog::DOUBLE:
    case execplan::CalpontSystemCatalog::UDOUBLE:
    case execplan::CalpontSystemCatalog::FLOAT:
    case execplan::CalpontSystemCatalog::UFLOAT:
    {
      double val = pm[0]->data()->getDoubleVal(row, isNull);

      if (notBetween)
      {
        if (!numericGE(val, pm[1]->data()->getDoubleVal(row, isNull)) && !isNull)
          return true;

        isNull = false;
        return (!numericLE(val, pm[2]->data()->getDoubleVal(row, isNull)) && !isNull);
      }

      return !isNull && numericGE(val, pm[1]->data()->getDoubleVal(row, isNull)) &&
             numericLE(val, pm[2]->data()->getDoubleVal(row, isNull));
    }

    case execplan::CalpontSystemCatalog::LONGDOUBLE:
    {
      long double val = pm[0]->data()->getLongDoubleVal(row, isNull);

      if (notBetween)
      {
        if (!numericGE(val, pm[1]->data()->getLongDoubleVal(row, isNull)) && !isNull)
          return true;

        isNull = false;
        return (!numericLE(val, pm[2]->data()->getLongDoubleVal(row, isNull)) && !isNull);
      }

      return !isNull && numericGE(val, pm[1]->data()->getLongDoubleVal(row, isNull)) &&
             numericLE(val, pm[2]->data()->getLongDoubleVal(row, isNull));
    }

    case execplan::CalpontSystemCatalog::DECIMAL:
    case execplan::CalpontSystemCatalog::UDECIMAL:
    {
      IDB_Decimal val = pm[0]->data()->getDecimalVal(row, isNull);

      if (isNull)
        return false;

      if (notBetween)
      {
        if (!numericGE(val, pm[1]->data()->getDecimalVal(row, isNull)) && !isNull)
          return true;

        isNull = false;
        return (!numericLE(val, pm[2]->data()->getDecimalVal(row, isNull)) && !isNull);
      }

      return numericGE(val, pm[1]->data()->getDecimalVal(row, isNull)) &&
             numericLE(val, pm[2]->data()->getDecimalVal(row, isNull));
    }

    case execplan::CalpontSystemCatalog::VARCHAR:  // including CHAR'
    case execplan::CalpontSystemCatalog::CHAR:
    case execplan::CalpontSystemCatalog::TEXT:
    {
      const string& val = pm[0]->data()->getStrVal(row, isNull);
      CHARSET_INFO& cs = datatypes::Charset(ct.charsetNumber).getCharset();

      if (notBetween)
      {
        if (!strGE(cs, val, pm[1]->data()->getStrVal(row, isNull)) && !isNull)
          return true;

        isNull = false;
        return (!strLE(cs, val, pm[2]->data()->getStrVal(row, isNull)) && !isNull);
      }

      return !isNull && strGE(cs, val, pm[1]->data()->getStrVal(row, isNull)) &&
             strLE(cs, val, pm[2]->data()->getStrVal(row, isNull));
    }

    default:
    {
      std::ostringstream oss;
      oss << "between: datatype of " << execplan::colDataTypeToString(ct.colDataType);
      throw logging::IDBExcept(oss.str(), ERR_DATATYPE_NOT_SUPPORT);
    }
  }
}

}  // namespace

namespace funcexp
{
CalpontSystemCatalog::ColType Func_between::operationType(FunctionParm& fp,
                                                          CalpontSystemCatalog::ColType& resultType)
{
  PredicateOperator op;
  CalpontSystemCatalog::ColType ct = fp[0]->data()->resultType();
  // op.operationType(fp[0]->data()->resultType());
  bool allString = true;

  for (uint32_t i = 1; i < fp.size(); i++)
  {
    // op.setOpType(op.operationType(), fp[i]->data()->resultType());
    op.setOpType(ct, fp[i]->data()->resultType());
    ct = op.operationType();

    if ((fp[i]->data()->resultType().colDataType != CalpontSystemCatalog::CHAR &&
         fp[i]->data()->resultType().colDataType != CalpontSystemCatalog::TEXT &&
         fp[i]->data()->resultType().colDataType != CalpontSystemCatalog::VARCHAR) ||
        ct.colDataType == CalpontSystemCatalog::DATE || ct.colDataType == CalpontSystemCatalog::DATETIME ||
        ct.colDataType == CalpontSystemCatalog::TIMESTAMP || ct.colDataType == CalpontSystemCatalog::TIME)
    {
      allString = false;
    }
  }

  if (allString)
  {
    ct.colDataType = CalpontSystemCatalog::VARCHAR;
    ct.colWidth = 255;
  }

  else if (op.operationType().colDataType == CalpontSystemCatalog::DATETIME)
  {
    ConstantColumn* cc = NULL;

    for (uint32_t i = 1; i < fp.size(); i++)
    {
      cc = dynamic_cast<ConstantColumn*>(fp[i]->data());

      if (cc)
      {
        Result result = cc->result();
        result.intVal = dataconvert::DataConvert::datetimeToInt(result.strVal);
        cc->result(result);
      }
    }
  }
  else if (op.operationType().colDataType == CalpontSystemCatalog::TIMESTAMP)
  {
    ConstantColumn* cc = NULL;

    for (uint32_t i = 1; i < fp.size(); i++)
    {
      cc = dynamic_cast<ConstantColumn*>(fp[i]->data());

      if (cc)
      {
        Result result = cc->result();
        result.intVal = dataconvert::DataConvert::timestampToInt(result.strVal, resultType.getTimeZone());
        cc->result(result);
      }
    }
  }
  else if (op.operationType().colDataType == CalpontSystemCatalog::TIME)
  {
    ConstantColumn* cc = NULL;

    for (uint32_t i = 1; i < fp.size(); i++)
    {
      cc = dynamic_cast<ConstantColumn*>(fp[i]->data());

      if (cc)
      {
        Result result = cc->result();
        result.intVal = dataconvert::DataConvert::timeToInt(result.strVal);
        cc->result(result);
      }
    }
  }

  return ct;
}

bool Func_between::getBoolVal(rowgroup::Row& row, FunctionParm& pm, bool& isNull,
                              CalpontSystemCatalog::ColType& ct)
{
  return getBool(row, pm, isNull, ct, false) && !isNull;
}

CalpontSystemCatalog::ColType Func_notbetween::operationType(FunctionParm& fp,
                                                             CalpontSystemCatalog::ColType& resultType)
{
  PredicateOperator* op = new PredicateOperator();
  CalpontSystemCatalog::ColType ct;
  op->setOpType(fp[0]->data()->resultType(), fp[1]->data()->resultType());
  op->setOpType(op->resultType(), fp[2]->data()->resultType());
  return op->operationType();
}

bool Func_notbetween::getBoolVal(rowgroup::Row& row, FunctionParm& pm, bool& isNull,
                                 CalpontSystemCatalog::ColType& ct)
{
  return getBool(row, pm, isNull, ct, true) && !isNull;
}

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