109 lines
2.5 KiB
C
109 lines
2.5 KiB
C
/*
|
|
* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
|
|
*
|
|
* This program is free software: you can use, redistribute, and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3
|
|
* or later ("AGPL"), as published by the Free Software Foundation.
|
|
*
|
|
* 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.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef TDENGINE_TUDF_INT_H
|
|
#define TDENGINE_TUDF_INT_H
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
//TODO replaces them with fnDebug
|
|
//#define debugPrint(...) taosPrintLog("Function", DEBUG_INFO, 135, __VA_ARGS__)
|
|
#define debugPrint(...) {fprintf(stderr, __VA_ARGS__);fprintf(stderr, "\n");}
|
|
enum {
|
|
UDF_TASK_SETUP = 0,
|
|
UDF_TASK_CALL = 1,
|
|
UDF_TASK_TEARDOWN = 2
|
|
|
|
};
|
|
|
|
enum {
|
|
TSDB_UDF_CALL_AGG_PROC = 0,
|
|
TSDb_UDF_CALL_AGG_FIN,
|
|
TSDB_UDF_CALL_SCALA_PROC,
|
|
};
|
|
|
|
typedef struct SUdfSetupRequest {
|
|
char udfName[16]; //
|
|
int8_t scriptType; // 0:c, 1: lua, 2:js
|
|
int8_t udfType; //udaf, udf
|
|
int16_t pathSize;
|
|
char *path;
|
|
} SUdfSetupRequest;
|
|
|
|
typedef struct SUdfSetupResponse {
|
|
int64_t udfHandle;
|
|
} SUdfSetupResponse;
|
|
|
|
typedef struct SUdfCallRequest {
|
|
int64_t udfHandle;
|
|
int8_t callType;
|
|
|
|
SUdfDataBlock block;
|
|
SUdfInterBuf interBuf;
|
|
bool initFirst;
|
|
} SUdfCallRequest;
|
|
|
|
typedef struct SUdfCallResponse {
|
|
SUdfColumnData resultData;
|
|
SUdfInterBuf interBuf;
|
|
} SUdfCallResponse;
|
|
|
|
|
|
typedef struct SUdfTeardownRequest {
|
|
int64_t udfHandle;
|
|
} SUdfTeardownRequest;
|
|
|
|
|
|
typedef struct SUdfTeardownResponse {
|
|
} SUdfTeardownResponse;
|
|
|
|
typedef struct SUdfRequest {
|
|
int32_t msgLen;
|
|
int64_t seqNum;
|
|
|
|
int8_t type;
|
|
union {
|
|
SUdfSetupRequest setup;
|
|
SUdfCallRequest call;
|
|
SUdfTeardownRequest teardown;
|
|
};
|
|
} SUdfRequest;
|
|
|
|
typedef struct SUdfResponse {
|
|
int32_t msgLen;
|
|
int64_t seqNum;
|
|
|
|
int8_t type;
|
|
int32_t code;
|
|
union {
|
|
SUdfSetupResponse setupRsp;
|
|
SUdfCallResponse callRsp;
|
|
SUdfTeardownResponse teardownRsp;
|
|
};
|
|
} SUdfResponse;
|
|
|
|
int32_t decodeRequest(char *buf, int32_t bufLen, SUdfRequest *pRequest);
|
|
int32_t encodeRequest(char **buf, int32_t *bufLen, SUdfRequest *request);
|
|
int32_t decodeResponse(char *buf, int32_t bufLen, SUdfResponse *pResponse);
|
|
int32_t encodeResponse(char **buf, int32_t *bufLen, SUdfResponse *response);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif // TDENGINE_TUDF_INT_H
|