database/mariadb-columnstore-engine
1
0
Fork 0
mirror of https://github.com/mariadb-corporation/mariadb-columnstore-engine.git synced 2025-08-08 14:22:09 +03:00
Code Activity

clang format apply

Browse Source
This commit is contained in:
Leonid Fedorov
2022-02-11 12:24:40 +00:00
parent 509f005be7
commit 7c808317dc
1367 changed files with 394342 additions and 413129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

209
utils/udfsdk/avg_mode.cpp
View File

@@ -24,159 +24,156 @@
using namespace mcsv1sdk;
mcsv1_UDAF::ReturnCode avg_mode::init(mcsv1Context* context,
ColumnDatum* colTypes)
mcsv1_UDAF::ReturnCode avg_mode::init(mcsv1Context* context, ColumnDatum* colTypes)
{
if (context->getParameterCount() < 1)
{
// The error message will be prepended with
// "The storage engine for the table doesn't support "
context->setErrorMessage("avg_mode() with 0 arguments");
return mcsv1_UDAF::ERROR;
}
if (context->getParameterCount() < 1)
{
// The error message will be prepended with
// "The storage engine for the table doesn't support "
context->setErrorMessage("avg_mode() with 0 arguments");
return mcsv1_UDAF::ERROR;
}
if (context->getParameterCount() > 1)
{
context->setErrorMessage("avg_mode() with more than 1 argument");
return mcsv1_UDAF::ERROR;
}
if (context->getParameterCount() > 1)
{
context->setErrorMessage("avg_mode() with more than 1 argument");
return mcsv1_UDAF::ERROR;
}
if (!(isNumeric(colTypes[0].dataType)))
{
// The error message will be prepended with
// "The storage engine for the table doesn't support "
context->setErrorMessage("avg_mode() with non-numeric argument");
return mcsv1_UDAF::ERROR;
}
context->setResultType(execplan::CalpontSystemCatalog::DOUBLE);
context->setColWidth(8);
context->setScale(context->getScale() * 2);
context->setPrecision(19);
context->setRunFlag(mcsv1sdk::UDAF_IGNORE_NULLS);
return mcsv1_UDAF::SUCCESS;
if (!(isNumeric(colTypes[0].dataType)))
{
// The error message will be prepended with
// "The storage engine for the table doesn't support "
context->setErrorMessage("avg_mode() with non-numeric argument");
return mcsv1_UDAF::ERROR;
}
context->setResultType(execplan::CalpontSystemCatalog::DOUBLE);
context->setColWidth(8);
context->setScale(context->getScale() * 2);
context->setPrecision(19);
context->setRunFlag(mcsv1sdk::UDAF_IGNORE_NULLS);
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode avg_mode::reset(mcsv1Context* context)
{
ModeData* data = static_cast<ModeData*>(context->getUserData());
data->mData.clear();
return mcsv1_UDAF::SUCCESS;
ModeData* data = static_cast<ModeData*>(context->getUserData());
data->mData.clear();
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode avg_mode::nextValue(mcsv1Context* context, ColumnDatum* valsIn)
{
MODE_DATA& data = static_cast<ModeData*>(context->getUserData())->mData;
MODE_DATA& data = static_cast<ModeData*>(context->getUserData())->mData;
if (valsIn[0].columnData.empty())
{
return mcsv1_UDAF::SUCCESS; // Ought not happen when UDAF_IGNORE_NULLS is on.
}
if (valsIn[0].columnData.empty())
{
return mcsv1_UDAF::SUCCESS; // Ought not happen when UDAF_IGNORE_NULLS is on.
}
DATATYPE val = toDouble(valsIn[0]);
DATATYPE val = toDouble(valsIn[0]);
data[val]++;
data[val]++;
return mcsv1_UDAF::SUCCESS;
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode avg_mode::subEvaluate(mcsv1Context* context, const UserData* userDataIn)
{
if (!userDataIn)
{
return mcsv1_UDAF::SUCCESS;
}
MODE_DATA& outData = static_cast<ModeData*>(context->getUserData())->mData;
const MODE_DATA& inData = static_cast<const ModeData*>(userDataIn)->mData;
MODE_DATA::const_iterator iter = inData.begin();
for (; iter != inData.end(); ++iter)
{
outData[iter->first] += iter->second;
}
if (!userDataIn)
{
return mcsv1_UDAF::SUCCESS;
}
MODE_DATA& outData = static_cast<ModeData*>(context->getUserData())->mData;
const MODE_DATA& inData = static_cast<const ModeData*>(userDataIn)->mData;
MODE_DATA::const_iterator iter = inData.begin();
for (; iter != inData.end(); ++iter)
{
outData[iter->first] += iter->second;
}
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode avg_mode::evaluate(mcsv1Context* context, static_any::any& valOut)
{
uint64_t maxCnt = 0;
MODE_DATA& data = static_cast<ModeData*>(context->getUserData())->mData;
if (data.size() == 0)
{
valOut = (DATATYPE)0;
return mcsv1_UDAF::SUCCESS;
}
MODE_DATA::iterator iter(data.begin());
for (; iter != data.end(); ++iter)
{
if (iter->second > maxCnt)
{
valOut = iter->first;
maxCnt = iter->second;
}
}
uint64_t maxCnt = 0;
MODE_DATA& data = static_cast<ModeData*>(context->getUserData())->mData;
if (data.size() == 0)
{
valOut = (DATATYPE)0;
return mcsv1_UDAF::SUCCESS;
}
MODE_DATA::iterator iter(data.begin());
for (; iter != data.end(); ++iter)
{
if (iter->second > maxCnt)
{
valOut = iter->first;
maxCnt = iter->second;
}
}
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode avg_mode::dropValue(mcsv1Context* context, ColumnDatum* valsDropped)
{
MODE_DATA& data = static_cast<ModeData*>(context->getUserData())->mData;
MODE_DATA& data = static_cast<ModeData*>(context->getUserData())->mData;
if (valsDropped[0].columnData.empty())
{
return mcsv1_UDAF::SUCCESS; // Ought not happen when UDAF_IGNORE_NULLS is on.
}
if (valsDropped[0].columnData.empty())
{
return mcsv1_UDAF::SUCCESS; // Ought not happen when UDAF_IGNORE_NULLS is on.
}
DATATYPE val = toDouble(valsDropped[0]);
DATATYPE val = toDouble(valsDropped[0]);
data[val]--;
data[val]--;
return mcsv1_UDAF::SUCCESS;
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode avg_mode::createUserData(UserData*& userData, int32_t& length)
{
userData = new ModeData;
length = sizeof(ModeData);
return mcsv1_UDAF::SUCCESS;
userData = new ModeData;
length = sizeof(ModeData);
return mcsv1_UDAF::SUCCESS;
}
void ModeData::serialize(messageqcpp::ByteStream& bs) const
{
MODE_DATA::const_iterator iter = mData.begin();
DATATYPE num;
uint32_t cnt;
bs << (int32_t)mData.size();
MODE_DATA::const_iterator iter = mData.begin();
DATATYPE num;
uint32_t cnt;
bs << (int32_t)mData.size();
for (; iter != mData.end(); ++iter)
{
num = iter->first;
bs << num;
cnt = iter->second;
bs << cnt;
}
for (; iter != mData.end(); ++iter)
{
num = iter->first;
bs << num;
cnt = iter->second;
bs << cnt;
}
}
void ModeData::unserialize(messageqcpp::ByteStream& bs)
{
mData.clear();
int32_t sz;
DATATYPE num;
uint32_t cnt;
bs >> sz;
mData.clear();
int32_t sz;
DATATYPE num;
uint32_t cnt;
bs >> sz;
for (int i = 0; i < sz; ++i)
{
bs >> num;
bs >> cnt;
mData[num] = cnt;
}
for (int i = 0; i < sz; ++i)
{
bs >> num;
bs >> cnt;
mData[num] = cnt;
}
}