database/mariadb-columnstore-engine
1
0
Fork 0
mirror of https://github.com/mariadb-corporation/mariadb-columnstore-engine.git synced 2025-08-01 06:46:55 +03:00
Code Activity

moda returned local object pointer (#2089)

Browse Source
This commit is contained in:
Leonid Fedorov
2021-08-18 19:41:07 +03:00
committed by GitHub
parent 469e5c7881
commit a2f441cd9c
1 changed files with 45 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

138
utils/regr/modamysql.cpp
View File

@ -3,7 +3,7 @@
#include <iostream> #include <iostream>
#include <sstream> #include <sstream>
#include <string.h> #include <string.h>
#include <tr1/unordered_map> #include <unordered_map>
#include <algorithm> #include <algorithm>
#include "idb_mysql.h" #include "idb_mysql.h"
@ -31,10 +31,11 @@ struct moda_data
{ {
long double fSum; long double fSum;
uint64_t fCount; uint64_t fCount;
enum Item_result fReturnType; enum Item_result fReturnType;
std::tr1::unordered_map<int64_t, uint32_t> mapINT; std::unordered_map<int64_t, uint32_t> mapINT;
std::tr1::unordered_map<double, uint32_t> mapREAL; std::unordered_map<double, uint32_t> mapREAL;
std::tr1::unordered_map<long double, uint32_t> mapDECIMAL; std::unordered_map<long double, uint32_t> mapDECIMAL;
std::string result;
void clear() void clear()
{ {
fSum = 0.0; fSum = 0.0;
@ -44,9 +45,39 @@ struct moda_data
mapDECIMAL.clear(); mapDECIMAL.clear();
} }
}; };
} }
template<class TYPE, class CONTAINER>
char * moda(CONTAINER & container, struct moda_data* data)
{
TYPE avg = (TYPE)data->fCount ? data->fSum / data->fCount : 0;
TYPE val = 0.0;
uint32_t maxCnt = 0.0;
for (auto iter = container.begin(); iter != container.end(); ++iter)
{
if (iter->second > maxCnt)
{
val = iter->first;
maxCnt = iter->second;
}
else if (iter->second == maxCnt)
{
// Tie breaker: choose the closest to avg. If still tie, choose smallest
if ((abs(val-avg) > abs(iter->first-avg))
|| ((abs(val-avg) == abs(iter->first-avg)) && (abs(val) > abs(iter->first))))
{
val = iter->first;
}
}
}
data->result = std::to_string(val);
return const_cast<char*>(data->result.c_str());
}
extern "C" extern "C"
{ {
@ -83,7 +114,7 @@ void moda_deinit(UDF_INIT* initid)
struct moda_data* data = (struct moda_data*)initid->ptr; struct moda_data* data = (struct moda_data*)initid->ptr;
data->clear(); data->clear();
delete data; delete data;
} }
#ifdef _MSC_VER #ifdef _MSC_VER
__declspec(dllexport) __declspec(dllexport)
@ -98,7 +129,7 @@ void moda_clear(UDF_INIT* initid, char* is_null __attribute__((unused)),
#ifdef _MSC_VER #ifdef _MSC_VER
__declspec(dllexport) __declspec(dllexport)
#endif #endif
void moda_add(UDF_INIT* initid, void moda_add(UDF_INIT* initid,
UDF_ARGS* args, UDF_ARGS* args,
char* is_null, char* is_null,
char* message __attribute__((unused))) char* message __attribute__((unused)))
@ -188,101 +219,22 @@ void moda_remove(UDF_INIT* initid, UDF_ARGS* args,
#ifdef _MSC_VER #ifdef _MSC_VER
__declspec(dllexport) __declspec(dllexport)
#endif #endif
char* moda(UDF_INIT* initid, UDF_ARGS* args __attribute__((unused)), char* moda(UDF_INIT* initid, UDF_ARGS* args, char* is_null, char* error __attribute__((unused)))
char* is_null, char* error __attribute__((unused)))
{ {
struct moda_data* data = (struct moda_data*)initid->ptr; struct moda_data* data = (struct moda_data*)initid->ptr;
uint32_t maxCnt = 0.0;
switch (args->arg_type[0]) switch (args->arg_type[0])
{ {
case INT_RESULT: case INT_RESULT:
{ return moda<int64_t>(data->mapINT, data);
typename std::tr1::unordered_map<int64_t, uint32_t>::iterator iter;
int64_t avg = (int64_t)data->fCount ? data->fSum / data->fCount : 0;
int64_t val = 0.0;
for (iter = data->mapINT.begin(); iter != data->mapINT.end(); ++iter)
{
if (iter->second > maxCnt)
{
val = iter->first;
maxCnt = iter->second;
}
else if (iter->second == maxCnt)
{
// Tie breaker: choose the closest to avg. If still tie, choose smallest
if ((abs(val-avg) > abs(iter->first-avg))
|| ((abs(val-avg) == abs(iter->first-avg)) && (abs(val) > abs(iter->first))))
{
val = iter->first;
}
}
}
std::ostringstream oss;
oss << val;
return const_cast<char*>(oss.str().c_str());
break;
}
case REAL_RESULT: case REAL_RESULT:
{ return moda<double>(data->mapREAL, data);
typename std::tr1::unordered_map<double, uint32_t>::iterator iter;
double avg = data->fCount ? data->fSum / data->fCount : 0;
double val = 0.0;
for (iter = data->mapREAL.begin(); iter != data->mapREAL.end(); ++iter)
{
if (iter->second > maxCnt)
{
val = iter->first;
maxCnt = iter->second;
}
else if (iter->second == maxCnt)
{
// Tie breaker: choose the closest to avg. If still tie, choose smallest
if ((abs(val-avg) > abs(iter->first-avg))
|| ((abs(val-avg) == abs(iter->first-avg)) && (abs(val) > abs(iter->first))))
{
val = iter->first;
}
}
}
std::ostringstream oss;
oss << val;
return const_cast<char*>(oss.str().c_str());
break;
}
case DECIMAL_RESULT: case DECIMAL_RESULT:
case STRING_RESULT: case STRING_RESULT:
{ return moda<long double>(data->mapDECIMAL, data);
typename std::tr1::unordered_map<long double, uint32_t>::iterator iter;
long double avg = data->fCount ? data->fSum / data->fCount : 0;
long double val = 0.0;
for (iter = data->mapDECIMAL.begin(); iter != data->mapDECIMAL.end(); ++iter)
{
if (iter->second > maxCnt)
{
val = iter->first;
maxCnt = iter->second;
}
else if (iter->second == maxCnt)
{
long double thisVal = iter->first;
// Tie breaker: choose the closest to avg. If still tie, choose smallest
if ((abs(val-avg) > abs(thisVal-avg))
|| ((abs(val-avg) == abs(thisVal-avg)) && (abs(val) > abs(thisVal))))
{
val = thisVal;
}
}
}
std::ostringstream oss;
oss << val;
return const_cast<char*>(oss.str().c_str());
break;
}
default: default:
break; return NULL;
} }
return NULL; return NULL;
} }
} // Extern "C" } // Extern "C"