110 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			110 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			C
		
	
	
	
| //
 | |
| // Created by shenglian on 28/02/22.
 | |
| //
 | |
| 
 | |
| #ifndef UDF_UDF_H
 | |
| #define UDF_UDF_H
 | |
| 
 | |
| #include <stdlib.h>
 | |
| #define DEBUG
 | |
| #ifdef DEBUG
 | |
| #define debugPrint(...) fprintf(__VA_ARGS__)
 | |
| #else
 | |
| #define debugPrint(...) /**/
 | |
| #endif
 | |
| 
 | |
| enum {
 | |
|   UDF_TASK_SETUP = 0,
 | |
|   UDF_TASK_CALL = 1,
 | |
|   UDF_TASK_TEARDOWN = 2
 | |
| 
 | |
| };
 | |
| 
 | |
| typedef struct SSDataBlock {
 | |
|   char   *data;
 | |
|   int32_t size;
 | |
| } SSDataBlock;
 | |
| 
 | |
| typedef struct SUdfInfo {
 | |
|   char *udfName;
 | |
|   char *path;
 | |
| } SUdfInfo;
 | |
| 
 | |
| typedef void *UdfcFuncHandle;
 | |
| 
 | |
| int32_t createUdfdProxy();
 | |
| 
 | |
| int32_t destroyUdfdProxy();
 | |
| 
 | |
| // int32_t setupUdf(SUdfInfo *udf, int32_t numOfUdfs, UdfcFuncHandle *handles);
 | |
| 
 | |
| int32_t setupUdf(SUdfInfo *udf, UdfcFuncHandle *handle);
 | |
| 
 | |
| int32_t callUdf(UdfcFuncHandle handle, int8_t step, char *state, int32_t stateSize, SSDataBlock input, char **newstate,
 | |
|                 int32_t *newStateSize, SSDataBlock *output);
 | |
| 
 | |
| int32_t doTeardownUdf(UdfcFuncHandle handle);
 | |
| 
 | |
| typedef struct SUdfSetupRequest {
 | |
|   char    udfName[16];  //
 | |
|   int8_t  scriptType;   // 0:c, 1: lua, 2:js
 | |
|   int8_t  udfType;      // udaf, udf, udtf
 | |
|   int16_t pathSize;
 | |
|   char   *path;
 | |
| } SUdfSetupRequest;
 | |
| 
 | |
| typedef struct SUdfSetupResponse {
 | |
|   int64_t udfHandle;
 | |
| } SUdfSetupResponse;
 | |
| 
 | |
| typedef struct SUdfCallRequest {
 | |
|   int64_t udfHandle;
 | |
|   int8_t  step;
 | |
| 
 | |
|   int32_t inputBytes;
 | |
|   char   *input;
 | |
| 
 | |
|   int32_t stateBytes;
 | |
|   char   *state;
 | |
| } SUdfCallRequest;
 | |
| 
 | |
| typedef struct SUdfCallResponse {
 | |
|   int32_t outputBytes;
 | |
|   char   *output;
 | |
|   int32_t newStateBytes;
 | |
|   char   *newState;
 | |
| } SUdfCallResponse;
 | |
| 
 | |
| typedef struct SUdfTeardownRequest {
 | |
|   int64_t udfHandle;
 | |
| } SUdfTeardownRequest;
 | |
| 
 | |
| typedef struct SUdfTeardownResponse {
 | |
| #ifdef WINDOWS
 | |
|   size_t avoidCompilationErrors;
 | |
| #endif
 | |
| } SUdfTeardownResponse;
 | |
| 
 | |
| typedef struct SUdfRequest {
 | |
|   int32_t msgLen;
 | |
|   int64_t seqNum;
 | |
| 
 | |
|   int8_t type;
 | |
|   void  *subReq;
 | |
| } SUdfRequest;
 | |
| 
 | |
| typedef struct SUdfResponse {
 | |
|   int32_t msgLen;
 | |
|   int64_t seqNum;
 | |
| 
 | |
|   int8_t  type;
 | |
|   int32_t code;
 | |
|   void   *subRsp;
 | |
| } SUdfResponse;
 | |
| 
 | |
| int32_t decodeRequest(char *buf, int32_t bufLen, SUdfRequest **pRequest);
 | |
| int32_t encodeResponse(char **buf, int32_t *bufLen, SUdfResponse *response);
 | |
| int32_t encodeRequest(char **buf, int32_t *bufLen, SUdfRequest *request);
 | |
| int32_t decodeResponse(char *buf, int32_t bufLen, SUdfResponse **pResponse);
 | |
| #endif  // UDF_UDF_H
 |