xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix: support more rows with udf unit test

This commit is contained in:
slzhou 2022-05-17 19:22:58 +08:00
parent 7ab15e8f02
commit e73b1c4889
1 changed files with 35 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
source/libs/function/test/runUdf.c
View File

@ -41,41 +41,41 @@ int scalarFuncTest() {
fnError("setup udf failure"); fnError("setup udf failure");
return -1; return -1;
} }
int64_t beg = taosGetTimestampUs();
SSDataBlock block = {0}; for (int k = 0; k < 1; ++k) {
SSDataBlock *pBlock = &block; SSDataBlock block = {0};
pBlock->pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData)); SSDataBlock *pBlock = &block;
pBlock->info.numOfCols = 1; pBlock->pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData));
pBlock->info.rows = 4; pBlock->info.numOfCols = 1;
char data[16] = {0}; pBlock->info.rows = 1024;
char bitmap[4] = {0}; for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) { SColumnInfoData colInfo = {0};
SColumnInfoData colInfo = {0}; colInfo.info.type = TSDB_DATA_TYPE_INT;
colInfo.info.type = TSDB_DATA_TYPE_INT; colInfo.info.bytes = sizeof(int32_t);
colInfo.info.bytes = sizeof(int32_t); colInfo.info.colId = 1;
colInfo.info.colId = 1; colInfoDataEnsureCapacity(&colInfo, 0, pBlock->info.rows);
colInfo.pData = data; for (int32_t j = 0; j < pBlock->info.rows; ++j) {
colInfo.nullbitmap = bitmap; colDataAppendInt32(&colInfo, j, &j);
for (int32_t j = 0; j < pBlock->info.rows; ++j) { }
colDataAppendInt32(&colInfo, j, &j); taosArrayPush(pBlock->pDataBlock, &colInfo);
} }
taosArrayPush(pBlock->pDataBlock, &colInfo);
SScalarParam input = {0};
input.numOfRows = pBlock->info.rows;
input.columnData = taosArrayGet(pBlock->pDataBlock, 0);
SScalarParam output = {0};
doCallUdfScalarFunc(handle, &input, 1, &output);
taosArrayDestroy(pBlock->pDataBlock);
SColumnInfoData *col = output.columnData;
for (int32_t i = 0; i < output.numOfRows; ++i) {
if (i % 100 == 0)
fprintf(stderr, "%d\t%d\n", i, *(int32_t *)(col->pData + i * sizeof(int32_t)));
}
colDataDestroy(output.columnData);
taosMemoryFree(output.columnData);
} }
int64_t end = taosGetTimestampUs();
SScalarParam input = {0}; fprintf(stderr, "time: %f\n", (end-beg)/1000.0);
input.numOfRows = pBlock->info.rows;
input.columnData = taosArrayGet(pBlock->pDataBlock, 0);
SScalarParam output = {0};
doCallUdfScalarFunc(handle, &input, 1, &output);
taosArrayDestroy(pBlock->pDataBlock);
SColumnInfoData *col = output.columnData;
for (int32_t i = 0; i < output.numOfRows; ++i) {
fprintf(stderr, "%d\t%d\n", i, *(int32_t *)(col->pData + i * sizeof(int32_t)));
}
colDataDestroy(output.columnData);
taosMemoryFree(output.columnData);
doTeardownUdf(handle); doTeardownUdf(handle);
return 0; return 0;
@ -93,16 +93,13 @@ int aggregateFuncTest() {
SSDataBlock *pBlock = &block; SSDataBlock *pBlock = &block;
pBlock->pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData)); pBlock->pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData));
pBlock->info.numOfCols = 1; pBlock->info.numOfCols = 1;
pBlock->info.rows = 4; pBlock->info.rows = 1024;
char data[16] = {0};
char bitmap[4] = {0};
for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) { for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
SColumnInfoData colInfo = {0}; SColumnInfoData colInfo = {0};
colInfo.info.type = TSDB_DATA_TYPE_INT; colInfo.info.type = TSDB_DATA_TYPE_INT;
colInfo.info.bytes = sizeof(int32_t); colInfo.info.bytes = sizeof(int32_t);
colInfo.info.colId = 1; colInfo.info.colId = 1;
colInfo.pData = data; colInfoDataEnsureCapacity(&colInfo, 0, pBlock->info.rows);
colInfo.nullbitmap = bitmap;
for (int32_t j = 0; j < pBlock->info.rows; ++j) { for (int32_t j = 0; j < pBlock->info.rows; ++j) {
colDataAppendInt32(&colInfo, j, &j); colDataAppendInt32(&colInfo, j, &j);
} }