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

runudf call scalar

This commit is contained in:
shenglian zhou 2022-04-17 17:37:07 +08:00
parent 18943125f4
commit d9580cff53
1 changed files with 20 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -5,6 +5,7 @@
#include "uv.h" #include "uv.h"
#include "os.h" #include "os.h"
#include "tudf.h" #include "tudf.h"
#include "tdatablock.h"
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
startUdfService(); startUdfService();
@ -23,17 +24,25 @@ int main(int argc, char *argv[]) {
SEpSet epSet; SEpSet epSet;
setupUdf("udf1", &epSet, &handle); setupUdf("udf1", &epSet, &handle);
//char state[5000000] = "state"; SSDataBlock block = {0};
//char input[5000000] = "input"; SSDataBlock* pBlock = █
int dataSize = 500; pBlock->pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData));
int callCount = 2; pBlock->info.numOfCols = 1;
if (argc > 1) dataSize = atoi(argv[1]); pBlock->info.rows = 4;
if (argc > 2) callCount = atoi(argv[2]); for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
char *state = taosMemoryMalloc(dataSize); SColumnInfoData colInfo = {0};
char *input = taosMemoryMalloc(dataSize); colInfo.info.type = TSDB_DATA_TYPE_INT;
//todo: call udf colInfo.info.bytes = sizeof(int32_t);
taosMemoryFree(state); colInfo.info.colId = 1;
taosMemoryFree(input); for (int32_t j = 0; j < pBlock->info.rows; ++j) {
colDataAppendInt32(&colInfo, j, &j);
}
taosArrayPush(pBlock->pDataBlock, &colInfo);
}
SSDataBlock output = {0};
callUdfScalaProcess(handle, pBlock, &output);
teardownUdf(handle); teardownUdf(handle);
stopUdfService(); stopUdfService();