You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-11-03 17:13:17 +03:00
387 lines
8.3 KiB
C++
387 lines
8.3 KiB
C++
/* 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: aggregatecolumn.h 9679 2013-07-11 22:32:03Z zzhu $
|
|
*
|
|
*
|
|
***********************************************************************/
|
|
/** @file */
|
|
|
|
#ifndef AGGREGATECOLUMN_H
|
|
#define AGGREGATECOLUMN_H
|
|
#include <string>
|
|
|
|
#include "calpontselectexecutionplan.h"
|
|
#include "returnedcolumn.h"
|
|
|
|
namespace messageqcpp {
|
|
class ByteStream;
|
|
}
|
|
|
|
/**
|
|
* Namespace
|
|
*/
|
|
namespace execplan {
|
|
|
|
/**
|
|
* @brief A class to represent a aggregate return column
|
|
*
|
|
* This class is a specialization of class ReturnedColumn that
|
|
* handles an aggregate function call (e.g., SUM, COUNT, MIN, MAX).
|
|
*/
|
|
class AggregateColumn : public ReturnedColumn {
|
|
|
|
public:
|
|
/**
|
|
* AggOp enum
|
|
*/
|
|
enum AggOp
|
|
{
|
|
NOOP = 0,
|
|
COUNT_ASTERISK,
|
|
COUNT,
|
|
SUM,
|
|
AVG,
|
|
MIN,
|
|
MAX,
|
|
CONSTANT,
|
|
DISTINCT_COUNT,
|
|
DISTINCT_SUM,
|
|
DISTINCT_AVG,
|
|
STDDEV_POP,
|
|
STDDEV_SAMP,
|
|
VAR_POP,
|
|
VAR_SAMP,
|
|
BIT_AND,
|
|
BIT_OR,
|
|
BIT_XOR,
|
|
GROUP_CONCAT
|
|
};
|
|
|
|
/**
|
|
* typedef
|
|
*/
|
|
typedef std::vector<SRCP> ColumnList;
|
|
/*
|
|
* Constructors
|
|
*/
|
|
/**
|
|
* ctor
|
|
*/
|
|
AggregateColumn();
|
|
|
|
/**
|
|
* ctor
|
|
*/
|
|
AggregateColumn(const uint32_t sessionID);
|
|
|
|
/**
|
|
* ctor
|
|
*/
|
|
AggregateColumn(const AggOp aggop, ReturnedColumn* parm, const uint32_t sessionID=0);
|
|
|
|
/**
|
|
* ctor
|
|
*/
|
|
AggregateColumn(const AggOp aggop, const std::string& content, const uint32_t sessionID=0);
|
|
|
|
/**
|
|
* ctor
|
|
*/
|
|
AggregateColumn(const std::string& functionName, ReturnedColumn* parm, const uint32_t sessionID=0);
|
|
|
|
/**
|
|
* ctor
|
|
*/
|
|
AggregateColumn(const std::string& functionName, const std::string& content, const uint32_t sessionID=0);
|
|
|
|
/**
|
|
* ctor
|
|
*/
|
|
AggregateColumn( const AggregateColumn& rhs, const uint32_t sessionID=0 );
|
|
|
|
/**
|
|
* Destructors
|
|
*/
|
|
virtual ~AggregateColumn() { }
|
|
|
|
/**
|
|
* Accessor Methods
|
|
*/
|
|
virtual const std::string functionName() const
|
|
{
|
|
return fFunctionName;
|
|
}
|
|
|
|
/**
|
|
* accessor
|
|
*/
|
|
virtual void functionName(const std::string& functionName)
|
|
{
|
|
fFunctionName = functionName;
|
|
}
|
|
|
|
/**
|
|
* accessor
|
|
*/
|
|
virtual const uint8_t aggOp() const {return fAggOp;}
|
|
/**
|
|
* accessor
|
|
*/
|
|
virtual void aggOp(const uint8_t aggOp) {fAggOp = aggOp;}
|
|
|
|
/** get function parms
|
|
*
|
|
* set the function parms from this object
|
|
*/
|
|
virtual const SRCP functionParms() const
|
|
{
|
|
return fFunctionParms;
|
|
}
|
|
|
|
/** set function parms
|
|
*
|
|
* set the function parms for this object
|
|
*/
|
|
virtual void functionParms(const SRCP& functionParms)
|
|
{
|
|
fFunctionParms = functionParms;
|
|
}
|
|
|
|
/** return a copy of this pointer
|
|
*
|
|
* deep copy of this pointer and return the copy
|
|
*/
|
|
inline virtual AggregateColumn* clone() const
|
|
{
|
|
return new AggregateColumn (*this);
|
|
}
|
|
|
|
/**
|
|
* table alias name
|
|
*/
|
|
virtual const std::string tableAlias() const { return fTableAlias; }
|
|
/**
|
|
* table alias name
|
|
*/
|
|
virtual void tableAlias (const std::string& tableAlias) { fTableAlias = tableAlias; }
|
|
|
|
/**
|
|
* ASC flag
|
|
*/
|
|
inline virtual const bool asc() const { return fAsc; }
|
|
/**
|
|
* ASC flag
|
|
*/
|
|
inline virtual void asc(const bool asc) { fAsc = asc; }
|
|
|
|
/**
|
|
* fData: SQL representation of this object
|
|
*/
|
|
virtual const std::string data() const { return fData; }
|
|
/**
|
|
* fData: SQL representation of this object
|
|
*/
|
|
virtual void data(const std::string& data) { fData = data; }
|
|
|
|
/**
|
|
* Overloaded stream operator
|
|
*/
|
|
virtual const std::string toString() const;
|
|
|
|
/**
|
|
* Serialize interface
|
|
*/
|
|
virtual void serialize(messageqcpp::ByteStream&) const;
|
|
/**
|
|
* Serialize interface
|
|
*/
|
|
virtual void unserialize(messageqcpp::ByteStream&);
|
|
|
|
/** @brief Do a deep, strict (as opposed to semantic) equivalence test
|
|
*
|
|
* Do a deep, strict (as opposed to semantic) equivalence test.
|
|
* @return true iff every member of t is a duplicate copy of every member of this; false otherwise
|
|
*/
|
|
virtual bool operator==(const TreeNode* t) const;
|
|
|
|
/** @brief Do a deep, strict (as opposed to semantic) equivalence test
|
|
*
|
|
* Do a deep, strict (as opposed to semantic) equivalence test.
|
|
* @return true iff every member of t is a duplicate copy of every member of this; false otherwise
|
|
*/
|
|
virtual bool operator==(const AggregateColumn& t) const;
|
|
|
|
/** @brief Do a deep, strict (as opposed to semantic) equivalence test
|
|
*
|
|
* Do a deep, strict (as opposed to semantic) equivalence test.
|
|
* @return false iff every member of t is a duplicate copy of every member of this; true otherwise
|
|
*/
|
|
virtual bool operator!=(const TreeNode* t) const;
|
|
|
|
/** @brief Do a deep, strict (as opposed to semantic) equivalence test
|
|
*
|
|
* Do a deep, strict (as opposed to semantic) equivalence test.
|
|
* @return false iff every member of t is a duplicate copy of every member of this; true otherwise
|
|
*/
|
|
virtual bool operator!=(const AggregateColumn& t) const;
|
|
|
|
/** @brief push back arg to group by column list*/
|
|
virtual void addGroupByCol(SRCP ac) {fGroupByColList.push_back(ac);}
|
|
|
|
/** @brief push back arg to project by column list*/
|
|
virtual void addProjectCol(SRCP ac) {fProjectColList.push_back(ac);}
|
|
|
|
/**
|
|
* accessor
|
|
*/
|
|
virtual const ColumnList& groupByColList() const { return fGroupByColList;}
|
|
/**
|
|
* accessor
|
|
*/
|
|
virtual const ColumnList& projectColList() const { return fProjectColList;}
|
|
|
|
/** @brief constant argument for aggregate with constant */
|
|
inline const SRCP constCol() const { return fConstCol; }
|
|
/**
|
|
* accessor
|
|
*/
|
|
inline void constCol(const SRCP& constCol) { fConstCol = constCol; }
|
|
|
|
/**
|
|
* convert an aggregate name to an AggOp enum
|
|
*/
|
|
static AggOp agname2num(const std::string&);
|
|
|
|
virtual bool hasAggregate();
|
|
virtual bool hasWindowFunc() {return false;}
|
|
|
|
protected:
|
|
std::string fFunctionName; // deprecated field
|
|
uint8_t fAggOp;
|
|
|
|
/**
|
|
* A ReturnedColumn objects that are the arguments to this function
|
|
*/
|
|
SRCP fFunctionParms;
|
|
|
|
/** table alias
|
|
* A string to represent table alias name which contains this column
|
|
*/
|
|
std:: string fTableAlias;
|
|
|
|
/**
|
|
* Flag to indicate asc or desc order for order by column
|
|
*/
|
|
bool fAsc;
|
|
std::string fData;
|
|
ColumnList fGroupByColList;
|
|
ColumnList fProjectColList;
|
|
SRCP fConstCol;
|
|
|
|
public:
|
|
/***********************************************************
|
|
* F&E framework *
|
|
***********************************************************/
|
|
/**
|
|
* F&E
|
|
*/
|
|
virtual const std::string& getStrVal(rowgroup::Row& row, bool& isNull)
|
|
{
|
|
evaluate(row, isNull);
|
|
return TreeNode::getStrVal();
|
|
}
|
|
|
|
/**
|
|
* F&E
|
|
*/
|
|
virtual int64_t getIntVal(rowgroup::Row& row, bool& isNull)
|
|
{
|
|
evaluate(row, isNull);
|
|
return TreeNode::getIntVal();
|
|
}
|
|
|
|
/**
|
|
* F&E
|
|
*/
|
|
virtual uint64_t getUintVal(rowgroup::Row& row, bool& isNull)
|
|
{
|
|
evaluate(row, isNull);
|
|
return TreeNode::getUintVal();
|
|
}
|
|
|
|
/**
|
|
* F&E
|
|
*/
|
|
virtual float getFloatVal(rowgroup::Row& row, bool& isNull)
|
|
{
|
|
evaluate(row, isNull);
|
|
return TreeNode::getFloatVal();
|
|
}
|
|
|
|
/**
|
|
* F&E
|
|
*/
|
|
virtual double getDoubleVal(rowgroup::Row& row, bool& isNull)
|
|
{
|
|
evaluate(row, isNull);
|
|
return TreeNode::getDoubleVal();
|
|
}
|
|
|
|
/**
|
|
* F&E
|
|
*/
|
|
virtual IDB_Decimal getDecimalVal(rowgroup::Row& row, bool& isNull)
|
|
{
|
|
evaluate(row, isNull);
|
|
return TreeNode::getDecimalVal();
|
|
}
|
|
/**
|
|
* F&E
|
|
*/
|
|
virtual int32_t getDateIntVal(rowgroup::Row& row, bool& isNull)
|
|
{
|
|
evaluate(row, isNull);
|
|
return TreeNode::getDateIntVal();
|
|
}
|
|
/**
|
|
* F&E
|
|
*/
|
|
virtual int64_t getDatetimeIntVal(rowgroup::Row& row, bool& isNull)
|
|
{
|
|
evaluate(row, isNull);
|
|
return TreeNode::getDatetimeIntVal();
|
|
}
|
|
|
|
private:
|
|
void evaluate(rowgroup::Row& row, bool& isNull);
|
|
};
|
|
|
|
/**
|
|
* stream operator
|
|
*/
|
|
std::ostream& operator<<(std::ostream& os, const AggregateColumn& rhs);
|
|
/**
|
|
* utility function to extract all aggregate columns from a parse tree
|
|
*/
|
|
void getAggCols(ParseTree* n, void* obj);
|
|
|
|
} //namespace execplan
|
|
#endif //AGGREGATECOLUMN_H
|
|
|