1
0
mirror of https://github.com/mariadb-corporation/mariadb-columnstore-engine.git synced 2025-04-18 21:44:02 +03:00
Sergey Zefirov 920607520c
feat(runtime)!: MCOL-678 A "GROUP BY ... WITH ROLLUP" support
Adds a special column which helps to differentiate data and rollups of
various depts and a simple logic to row aggregation to add processing of
subtotals.
2023-09-26 17:01:53 +03:00

413 lines
12 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: constantcolumn.cpp 9474 2013-05-02 15:28:09Z rdempsey $
*
*
***********************************************************************/
#include <string>
#include <iostream>
#include <sstream>
using namespace std;
#include "bytestream.h"
#include "constantcolumn.h"
#include "objectreader.h"
using namespace messageqcpp;
#include "dataconvert.h"
#include "calpontsystemcatalog.h"
namespace execplan
{
/**
* Constructors/Destructors
*/
ConstantColumn::ConstantColumn() : ReturnedColumn(), fType(NULLDATA)
{
}
ConstantColumn::ConstantColumn(const string& sql, TYPE type)
: ReturnedColumn(), fConstval(sql), fType(type), fData(sql)
{
fResult.strVal.assign(sql);
fResult.intVal = atoll(sql.c_str());
fResult.uintVal = strtoull(sql.c_str(), NULL, 0);
fResult.floatVal = atof(sql.c_str());
fResult.doubleVal = atof(sql.c_str());
fResult.longDoubleVal = strtold(sql.c_str(), NULL);
// decimal for constant should be constructed by the caller and call the decimal constructor
fResult.decimalVal = datatypes::Decimal(fResult.intVal, 0, 18, (int128_t)fResult.intVal);
// @bug 3381, default null item to integer type.
if (fType == ConstantColumn::NULLDATA)
{
if (fResult.uintVal > (uint64_t)MAX_BIGINT)
{
fResultType.colDataType = CalpontSystemCatalog::UBIGINT;
}
else
{
fResultType.colDataType = CalpontSystemCatalog::BIGINT;
}
fResultType.colWidth = 8;
}
else
{
fResultType.colDataType = CalpontSystemCatalog::VARCHAR;
fResultType.colWidth = sql.length();
}
}
ConstantColumn::ConstantColumn(const string& sql, const double val)
: ReturnedColumn(), fConstval(sql), fType(NUM), fData(sql)
{
fResult.strVal.assign(sql);
fResult.doubleVal = val;
fResult.intVal = (int64_t)val;
fResult.uintVal = (uint64_t)val;
fResult.floatVal = (float)val;
fResult.longDoubleVal = val;
// decimal for constant should be constructed by the caller and call the decimal constructor
fResult.decimalVal = datatypes::Decimal(fResult.intVal, 0, 18, (int128_t)fResult.intVal);
fResultType.colDataType = CalpontSystemCatalog::DOUBLE;
fResultType.colWidth = 8;
}
ConstantColumn::ConstantColumn(const string& sql, const long double val)
: ReturnedColumn(), fConstval(sql), fType(NUM), fData(sql)
{
fResult.strVal.assign(sql);
fResult.doubleVal = (double)val;
fResult.intVal = (int64_t)val;
fResult.uintVal = (uint64_t)val;
fResult.floatVal = (float)val;
fResult.longDoubleVal = val;
// decimal for constant should be constructed by the caller and call the decimal constructor
fResult.decimalVal = datatypes::Decimal(fResult.intVal, 0, 18, (int128_t)fResult.intVal);
fResultType.colDataType = CalpontSystemCatalog::DOUBLE;
fResultType.colWidth = 8;
}
ConstantColumn::ConstantColumn(const string& sql, const int64_t val, TYPE type)
: ReturnedColumn(), fConstval(sql), fType(type), fData(sql)
{
fResult.strVal.assign(sql);
fResult.intVal = val;
fResult.uintVal = (uint64_t)fResult.intVal;
fResult.floatVal = (float)fResult.intVal;
fResult.doubleVal = (double)fResult.intVal;
fResult.longDoubleVal = (long double)fResult.intVal;
fResult.decimalVal = datatypes::Decimal(fResult.intVal, 0, 0, (int128_t)fResult.intVal);
fResultType.colDataType = CalpontSystemCatalog::BIGINT;
fResultType.colWidth = 8;
}
ConstantColumn::ConstantColumn(const string& sql, const uint64_t val, TYPE type)
: ReturnedColumn(), fConstval(sql), fType(type), fData(sql)
{
fResult.strVal.assign(sql);
fResult.uintVal = val;
fResult.intVal = (int64_t)fResult.uintVal;
fResult.floatVal = (float)fResult.uintVal;
fResult.doubleVal = (double)fResult.uintVal;
fResult.longDoubleVal = (long double)fResult.uintVal;
fResult.decimalVal = datatypes::Decimal(fResult.uintVal, 0, 0, (int128_t)fResult.uintVal);
fResultType.colDataType = CalpontSystemCatalog::UBIGINT;
fResultType.colWidth = 8;
}
ConstantColumn::ConstantColumn(const string& sql, const IDB_Decimal& val)
: ReturnedColumn(), fConstval(sql), fType(NUM), fData(sql)
{
fResult.strVal.assign(sql);
fResult.intVal = (int64_t)atoll(sql.c_str());
fResult.uintVal = strtoull(sql.c_str(), NULL, 0);
fResult.floatVal = atof(sql.c_str());
fResult.doubleVal = atof(sql.c_str());
fResult.longDoubleVal = strtold(sql.c_str(), NULL);
fResult.decimalVal = val;
fResultType.colDataType = CalpontSystemCatalog::DECIMAL;
fResultType.setDecimalScalePrecision(val.precision, val.scale);
}
ConstantColumn::ConstantColumn(const ConstantColumn& rhs)
: ReturnedColumn(rhs)
, fConstval(rhs.constval())
, fType(rhs.type())
, fData(rhs.data())
, fTimeZone(rhs.timeZone())
{
sequence(rhs.sequence());
fAlias = rhs.alias();
fResult = rhs.fResult;
fResultType = rhs.fResultType;
}
ConstantColumn::ConstantColumn(const int64_t val, TYPE type) : ReturnedColumn(), fType(type)
{
ostringstream oss;
oss << val;
fConstval.assign(oss.str());
fData = oss.str();
fResult.strVal.assign(fData);
fResult.intVal = val;
fResult.uintVal = (uint64_t)fResult.intVal;
fResult.floatVal = (float)fResult.intVal;
fResult.doubleVal = (double)fResult.intVal;
fResult.longDoubleVal = (long double)fResult.intVal;
fResult.decimalVal = datatypes::Decimal(fResult.intVal, 0, 0, (int128_t)fResult.intVal);
fResultType.colDataType = CalpontSystemCatalog::BIGINT;
fResultType.colWidth = 8;
}
ConstantColumn::ConstantColumn(const uint64_t val, TYPE type, int8_t scale, uint8_t precision)
: ReturnedColumn(), fType(type)
{
ostringstream oss;
oss << val;
fConstval.assign(oss.str());
fData = oss.str();
fResult.strVal.assign(fData);
fResult.intVal = (int64_t)val;
fResult.uintVal = val;
fResult.floatVal = (float)fResult.uintVal;
fResult.doubleVal = (double)fResult.uintVal;
fResult.longDoubleVal = (long double)fResult.uintVal;
fResult.decimalVal = datatypes::Decimal(fResult.uintVal, scale, precision, (int128_t)fResult.uintVal);
fResultType.colDataType = CalpontSystemCatalog::UBIGINT;
fResultType.colWidth = 8;
}
ConstantColumn::~ConstantColumn()
{
}
const string ConstantColumn::toString() const
{
ostringstream oss;
oss << "ConstantColumn: " << fConstval.safeString("<<NuLL>>") << " intVal=" << fResult.intVal
<< " uintVal=" << fResult.uintVal;
oss << '(';
if (fType == LITERAL)
oss << 'l';
else if (fType == NUM)
oss << 'n';
else
oss << "null";
oss << ')';
oss << " resultType=" << colDataTypeToString(fResultType.colDataType);
if (fAlias.length() > 0)
oss << "/Alias: " << fAlias;
return oss.str();
}
std::string ConstantColumn::toCppCode(IncludeSet& includes) const
{
includes.insert("constantcolumn.h");
std::stringstream ss;
ss << "ConstantColumn(" << std::quoted(fData) << ", " << fConstval.safeString() << ")";
return ss.str();
}
const string ConstantColumn::data() const
{
return fData;
}
ostream& operator<<(ostream& output, const ConstantColumn& rhs)
{
output << rhs.toString();
return output;
}
void ConstantColumn::serialize(messageqcpp::ByteStream& b) const
{
b << (ObjectReader::id_t)ObjectReader::CONSTANTCOLUMN;
ReturnedColumn::serialize(b);
b << fConstval;
b << (uint32_t)fType;
// b << fAlias;
b << fData;
messageqcpp::ByteStream::octbyte timeZone = fTimeZone;
b << timeZone;
b << static_cast<ByteStream::doublebyte>(fReturnAll);
b << (uint64_t)fResult.intVal;
b << fResult.uintVal;
b << fResult.doubleVal;
b << fResult.longDoubleVal;
b << fResult.floatVal;
b << (uint8_t)fResult.boolVal;
b << fResult.strVal;
b << (uint64_t)fResult.decimalVal.value;
b << fResult.decimalVal.s128Value;
b << (uint8_t)fResult.decimalVal.scale;
b << (uint8_t)fResult.decimalVal.precision;
}
void ConstantColumn::unserialize(messageqcpp::ByteStream& b)
{
ObjectReader::checkType(b, ObjectReader::CONSTANTCOLUMN);
ReturnedColumn::unserialize(b);
// uint64_t val;
b >> fConstval;
b >> (uint32_t&)fType;
b >> fData;
messageqcpp::ByteStream::octbyte timeZone;
b >> timeZone;
fTimeZone = timeZone;
b >> reinterpret_cast<ByteStream::doublebyte&>(fReturnAll);
b >> (uint64_t&)fResult.intVal;
b >> fResult.uintVal;
b >> fResult.doubleVal;
b >> fResult.longDoubleVal;
b >> fResult.floatVal;
b >> (uint8_t&)fResult.boolVal;
b >> fResult.strVal;
b >> (uint64_t&)fResult.decimalVal.value;
b >> fResult.decimalVal.s128Value;
b >> (uint8_t&)fResult.decimalVal.scale;
b >> (uint8_t&)fResult.decimalVal.precision;
}
RollupMarkColumn::RollupMarkColumn()
{
fExpressionId = 0x55667788ULL;
fResultType.colDataType = CalpontSystemCatalog::INT;
fResultType.colWidth = 4;
// no-op.
}
RollupMarkColumn::~RollupMarkColumn()
{
// no-op
}
void RollupMarkColumn::serialize(messageqcpp::ByteStream& b) const
{
b << (ObjectReader::id_t)ObjectReader::ROLLUPMARKCOLUMN;
ReturnedColumn::serialize(b);
messageqcpp::ByteStream::octbyte timeZone = fTimeZone;
b << timeZone;
b << static_cast<ByteStream::doublebyte>(fReturnAll);
b << (uint64_t)fResult.intVal;
b << fResult.uintVal;
b << fResult.doubleVal;
b << fResult.longDoubleVal;
b << fResult.floatVal;
b << (uint8_t)fResult.boolVal;
b << fResult.strVal;
b << (uint64_t)fResult.decimalVal.value;
b << fResult.decimalVal.s128Value;
b << (uint8_t)fResult.decimalVal.scale;
b << (uint8_t)fResult.decimalVal.precision;
}
void RollupMarkColumn::unserialize(messageqcpp::ByteStream& b)
{
ObjectReader::checkType(b, ObjectReader::ROLLUPMARKCOLUMN);
ReturnedColumn::unserialize(b);
// uint64_t val;
messageqcpp::ByteStream::octbyte timeZone;
b >> timeZone;
fTimeZone = timeZone;
b >> reinterpret_cast<ByteStream::doublebyte&>(fReturnAll);
b >> (uint64_t&)fResult.intVal;
b >> fResult.uintVal;
b >> fResult.doubleVal;
b >> fResult.longDoubleVal;
b >> fResult.floatVal;
b >> (uint8_t&)fResult.boolVal;
b >> fResult.strVal;
b >> (uint64_t&)fResult.decimalVal.value;
b >> fResult.decimalVal.s128Value;
b >> (uint8_t&)fResult.decimalVal.scale;
b >> (uint8_t&)fResult.decimalVal.precision;
}
static utils::NullString ns;
const utils::NullString& RollupMarkColumn::getStrVal(rowgroup::Row& row, bool& isNull)
{
return ns;
}
bool ConstantColumn::operator==(const ConstantColumn& t) const
{
const ReturnedColumn *rc1, *rc2;
rc1 = static_cast<const ReturnedColumn*>(this);
rc2 = static_cast<const ReturnedColumn*>(&t);
if (*rc1 != *rc2)
return false;
if (fConstval.isNull() != t.fConstval.isNull())
return false;
if (!fConstval.isNull() && fConstval.unsafeStringRef() != t.fConstval.unsafeStringRef())
return false;
if (fType != t.fType)
return false;
if (fData != t.fData)
return false;
if (fReturnAll != t.fReturnAll)
return false;
if (fTimeZone != t.fTimeZone)
return false;
return true;
}
bool ConstantColumn::operator==(const TreeNode* t) const
{
const ConstantColumn* o;
o = dynamic_cast<const ConstantColumn*>(t);
if (o == NULL)
return false;
return *this == *o;
}
bool ConstantColumn::operator!=(const ConstantColumn& t) const
{
return (!(*this == t));
}
bool ConstantColumn::operator!=(const TreeNode* t) const
{
return (!(*this == t));
}
} // namespace execplan