homework-jianmu/tests/script/sh/gpd.c

47 lines
1.2 KiB
C

#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#ifdef LINUX
#include <unistd.h>
#endif
#ifdef WINDOWS
#include <windows.h>
#endif
#include "taosudf.h"
TAOS* taos = NULL;
DLL_EXPORT int32_t gpd_init() {
return 0;
}
DLL_EXPORT int32_t gpd_destroy() {
return 0;
}
DLL_EXPORT int32_t gpd(SUdfDataBlock* block, SUdfColumn *resultCol) {
SUdfColumnMeta *meta = &resultCol->colMeta;
meta->bytes = 4;
meta->type = TSDB_DATA_TYPE_INT;
meta->scale = 0;
meta->precision = 0;
SUdfColumnData *resultData = &resultCol->colData;
resultData->numOfRows = block->numOfRows;
for (int32_t i = 0; i < resultData->numOfRows; ++i) {
int64_t* calc_ts = (int64_t*)udfColDataGetData(block->udfCols[0], i);
char* varTbname = udfColDataGetData(block->udfCols[1], i);
char* varDbname = udfColDataGetData(block->udfCols[2], i);
char dbName[256] = {0};
char tblName[256] = {0};
memcpy(dbName, varDataVal(varDbname), varDataLen(varDbname));
memcpy(tblName, varDataVal(varTbname), varDataLen(varTbname));
printf("%s, %s\n", dbName, tblName);
int32_t result = 0;
udfColDataSet(resultCol, i, (char*)&result, false);
}
return 0;
}