1
0
mirror of https://github.com/mariadb-corporation/mariadb-columnstore-engine.git synced 2025-04-18 21:44:02 +03:00
2025-02-21 20:02:38 +04:00

729 lines
20 KiB
C++

/* 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: functor_int.h 3642 2013-03-18 19:09:53Z bpaul $
/** @file */
#pragma once
#include "functor.h"
namespace funcexp
{
/** @brief Func_Int class
* For function that returns a integer result.
* Must implement getIntVal()
* Implement any other methods that behave differently from the default.
*/
class Func_Int : public Func
{
public:
Func_Int() = default;
explicit Func_Int(const std::string& funcName) : Func(funcName)
{
}
~Func_Int() override = default;
/*
int64_t getIntVal(rowgroup::Row& row,
FunctionParm& fp,
bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) = 0;
*/
double getDoubleVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override
{
return ((double)getIntVal(row, fp, isNull, op_ct));
}
long double getLongDoubleVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override
{
return ((long double)getIntVal(row, fp, isNull, op_ct));
}
std::string getStrVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override
{
return intToString(getIntVal(row, fp, isNull, op_ct));
}
};
class Func_BitOp : public Func_Int
{
public:
explicit Func_BitOp(const std::string& funcName) : Func_Int(funcName)
{
}
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override
{
return resultType;
}
bool validateArgCount(execplan::FunctionColumn& col, uint expected) const;
void setFunctorByParm(execplan::FunctionColumn& col, const execplan::SPTP& parm,
Func_Int& return_uint64_from_uint64, Func_Int& return_uint64_from_sint64,
Func_Int& return_uint64_from_generic) const;
// Fix for << and >>
bool fixForBitShift(execplan::FunctionColumn& col, Func_Int& return_uint64_from_uint64,
Func_Int& return_uint64_from_sint64, Func_Int& return_uint64_from_generic) const;
// Fix for & | ^
bool fixForBitOp2(execplan::FunctionColumn& col, Func_Int& return_uint64_from_uint64_uint64,
Func_Int& return_uint64_from_sint64_sint64,
Func_Int& return_uint64_from_generic_generic) const;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override
{
isNull = true;
return 0;
}
};
/** @brief Func_instr class
*/
class Func_instr : public Func_Int
{
public:
Func_instr() : Func_Int("instr")
{
}
~Func_instr() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_length class
*/
class Func_length : public Func_Int
{
public:
Func_length() : Func_Int("length")
{
}
~Func_length() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_sign class
*/
class Func_sign : public Func_Int
{
public:
Func_sign() : Func_Int("sign")
{
}
~Func_sign() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
std::string getStrVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_day class
*/
class Func_day : public Func_Int
{
public:
Func_day() : Func_Int("day")
{
}
~Func_day() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_minute class
*/
class Func_minute : public Func_Int
{
public:
Func_minute() : Func_Int("minute")
{
}
~Func_minute() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_month class
*/
class Func_month : public Func_Int
{
public:
Func_month() : Func_Int("month")
{
}
~Func_month() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_week class
*/
class Func_week : public Func_Int
{
public:
Func_week() : Func_Int("week")
{
}
~Func_week() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_year class
*/
class Func_year : public Func_Int
{
public:
Func_year() : Func_Int("year")
{
}
~Func_year() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_to_days class
*/
class Func_to_days : public Func_Int
{
public:
Func_to_days() : Func_Int("to_days")
{
}
~Func_to_days() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_char_length class
*/
class Func_char_length : public Func_Int
{
public:
Func_char_length() : Func_Int("length")
{
}
~Func_char_length() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_extract class
*/
class Func_extract : public Func_Int
{
public:
Func_extract() : Func_Int("extract")
{
}
~Func_extract() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_cast_signed class
*/
class Func_cast_signed : public Func_Int
{
public:
Func_cast_signed() : Func_Int("cast_signed")
{
}
~Func_cast_signed() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_cast_unsigned class
*/
class Func_cast_unsigned : public Func_Int
{
public:
Func_cast_unsigned() : Func_Int("cast_unsigned")
{
}
~Func_cast_unsigned() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override
{
return (int64_t)(getUintVal(row, fp, isNull, op_ct));
}
uint64_t getUintVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_bitand class
*/
class Func_bitand : public Func_BitOp
{
public:
Func_bitand() : Func_BitOp("bitand")
{
}
~Func_bitand() override = default;
bool fix(execplan::FunctionColumn& col) const override;
};
/** @brief Func_bitor class
*/
class Func_bitor : public Func_BitOp
{
public:
Func_bitor() : Func_BitOp("bitor")
{
}
~Func_bitor() override = default;
bool fix(execplan::FunctionColumn& col) const override;
uint64_t getUintVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_bitxor class
*/
class Func_bitxor : public Func_BitOp
{
public:
Func_bitxor() : Func_BitOp("bitxor")
{
}
~Func_bitxor() override = default;
bool fix(execplan::FunctionColumn& col) const override;
};
/** @brief Func_bit_count class
*/
class Func_bit_count : public Func_BitOp
{
public:
Func_bit_count() : Func_BitOp("bit_count")
{
}
~Func_bit_count() override = default;
bool fix(execplan::FunctionColumn& col) const override;
};
/** @brief Func_hour class
*/
class Func_hour : public Func_Int
{
public:
Func_hour() : Func_Int("hour")
{
}
~Func_hour() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_second class
*/
class Func_second : public Func_Int
{
public:
Func_second() : Func_Int("second")
{
}
~Func_second() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_dayofweek class
*/
class Func_dayofweek : public Func_Int
{
public:
Func_dayofweek() : Func_Int("dayofweek")
{
}
~Func_dayofweek() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_dayofyear class
*/
class Func_dayofyear : public Func_Int
{
public:
Func_dayofyear() : Func_Int("dayofyear")
{
}
~Func_dayofyear() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_leftshift class
*/
class Func_leftshift : public Func_BitOp
{
public:
Func_leftshift() : Func_BitOp("leftshift")
{
}
~Func_leftshift() override = default;
bool fix(execplan::FunctionColumn& col) const override;
};
/** @brief Func_rightshift class
*/
class Func_rightshift : public Func_BitOp
{
public:
Func_rightshift() : Func_BitOp("rightshift")
{
}
~Func_rightshift() override = default;
bool fix(execplan::FunctionColumn& col) const override;
};
/** @brief Func_quarter class
*/
class Func_quarter : public Func_Int
{
public:
Func_quarter() : Func_Int("quarter")
{
}
~Func_quarter() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_ascii class
*/
class Func_ascii : public Func_Int
{
public:
Func_ascii() : Func_Int("ascii")
{
}
~Func_ascii() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_dayname class
*/
class Func_dayname : public Func_Int
{
public:
Func_dayname() : Func_Int("dayname")
{
}
~Func_dayname() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
std::string getStrVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_weekday class
*/
class Func_weekday : public Func_Int
{
public:
Func_weekday() : Func_Int("weekday")
{
}
~Func_weekday() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_yearweek class
*/
class Func_yearweek : public Func_Int
{
public:
Func_yearweek() : Func_Int("yearweek")
{
}
~Func_yearweek() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_last_day class
*/
class Func_last_day : public Func_Int
{
public:
Func_last_day() : Func_Int("last_day")
{
}
~Func_last_day() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_time_to_sec class
*/
class Func_time_to_sec : public Func_Int
{
public:
Func_time_to_sec() : Func_Int("time_to_sec")
{
}
~Func_time_to_sec() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_microsecond class
*/
class Func_microsecond : public Func_Int
{
public:
Func_microsecond() : Func_Int("microsecond")
{
}
~Func_microsecond() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_crc32 class
*/
class Func_crc32 : public Func_Int
{
public:
Func_crc32() : Func_Int("crc32")
{
}
~Func_crc32() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_period_add class
*/
class Func_period_add : public Func_Int
{
public:
Func_period_add() : Func_Int("period_add")
{
}
~Func_period_add() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_period_diff class
*/
class Func_period_diff : public Func_Int
{
public:
Func_period_diff() : Func_Int("period_diff")
{
}
~Func_period_diff() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_strcmp class
*/
class Func_strcmp : public Func_Int
{
public:
Func_strcmp() : Func_Int("strcmp")
{
}
~Func_strcmp() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
std::string getStrVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_unix_timestamp class
*/
class Func_unix_timestamp : public Func_Int
{
public:
Func_unix_timestamp() : Func_Int("unix_timestamp")
{
}
~Func_unix_timestamp() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
std::string getStrVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
/** @brief Func_strcmp class
*/
class Func_find_in_set : public Func_Int
{
public:
Func_find_in_set() : Func_Int("find_in_set")
{
}
~Func_find_in_set() override = default;
execplan::CalpontSystemCatalog::ColType operationType(
FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& resultType) override;
int64_t getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
std::string getStrVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
double getDoubleVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
execplan::IDB_Decimal getDecimalVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
execplan::CalpontSystemCatalog::ColType& op_ct) override;
};
} // namespace funcexp