database/mariadb-columnstore-engine
1
0
Fork 0
mirror of https://github.com/mariadb-corporation/mariadb-columnstore-engine.git synced 2025-07-29 08:21:15 +03:00
Code Activity

MCOL-523 Add UDAF and UDAnF SDK

Browse Source
This commit is contained in:
David Hall
2017-07-26 11:53:08 -05:00
parent 630b113565
commit bc2a4e7795
75 changed files with 10250 additions and 4523 deletions
Download Patch File Download Diff File Expand all files Collapse all files

96
utils/udfsdk/allnull.cpp Executable file
View File

@ -0,0 +1,96 @@
/* Copyright (C) 2017 MariaDB Corporaton
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. */
#include <sstream>
#include <cstring>
#include "allnull.h"
#include "bytestream.h"
#include "objectreader.h"
using namespace mcsv1sdk;
struct allnull_data
{
uint64_t totalQuantity;
uint64_t totalNulls;
};
#define OUT_TYPE int64_t
mcsv1_UDAF::ReturnCode allnull::init(mcsv1Context* context,
COL_TYPES& colTypes)
{
context->setUserDataSize(sizeof(allnull_data));
if (colTypes.size() < 1)
{
// The error message will be prepended with
// "The storage engine for the table doesn't support "
context->setErrorMessage("allnull() with 0 arguments");
return mcsv1_UDAF::ERROR;
}
context->setResultType(CalpontSystemCatalog::TINYINT);
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode allnull::finish(mcsv1Context* context)
{
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode allnull::reset(mcsv1Context* context)
{
struct allnull_data* data = (struct allnull_data*)context->getUserData()->data;
data->totalQuantity = 0;
data->totalNulls = 0;
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode allnull::nextValue(mcsv1Context* context,
std::vector<ColumnDatum>& valsIn)
{
struct allnull_data* data = (struct allnull_data*)context->getUserData()->data;
for (size_t i = 0; i < context->getParameterCount(); i++)
{
data->totalQuantity++;
if (context->isParamNull(0))
{
data->totalNulls++;
}
}
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode allnull::subEvaluate(mcsv1Context* context, const UserData* userDataIn)
{
struct allnull_data* outData = (struct allnull_data*)context->getUserData()->data;
struct allnull_data* inData = (struct allnull_data*)userDataIn->data;
outData->totalQuantity += inData->totalQuantity;
outData->totalNulls += inData->totalNulls;
return mcsv1_UDAF::SUCCESS;
}
mcsv1_UDAF::ReturnCode allnull::evaluate(mcsv1Context* context, static_any::any& valOut)
{
OUT_TYPE allNull;
struct allnull_data* data = (struct allnull_data*)context->getUserData()->data;
allNull = data->totalQuantity > 0 && data->totalNulls == data->totalQuantity;
valOut = allNull;
return mcsv1_UDAF::SUCCESS;
}