Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
taosdata
TDengine
提交
8e5dae39
T
TDengine
项目概览
taosdata
/
TDengine
1 年多 前同步成功
通知
1185
Star
22016
Fork
4786
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
TDengine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
8e5dae39
编写于
2月 21, 2023
作者:
S
slzhou
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix: change script udf info structure
上级
89db7bf6
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
19 addition
and
10 deletion
+19
-10
include/libs/function/taosudf.h
include/libs/function/taosudf.h
+9
-4
source/libs/function/src/udfd.c
source/libs/function/src/udfd.c
+10
-6
未找到文件。
include/libs/function/taosudf.h
浏览文件 @
8e5dae39
...
@@ -262,17 +262,22 @@ typedef int32_t (*TUdfAggFinishFunc)(SUdfInterBuf *buf, SUdfInterBuf *resultData
...
@@ -262,17 +262,22 @@ typedef int32_t (*TUdfAggFinishFunc)(SUdfInterBuf *buf, SUdfInterBuf *resultData
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
typedef
struct
SUdfInfo
{
typedef
enum
EUdfFuncType
{
UDF_FUNC_TYPE_SCALAR
=
1
,
UDF_FUNC_TYPE_AGG
=
2
}
EUdfFuncType
;
typedef
struct
SScriptUdfInfo
{
char
*
name
;
char
*
name
;
int8_t
funcType
;
EUdfFuncType
funcType
;
int8_t
scriptType
;
int8_t
scriptType
;
int8_t
outputType
;
int8_t
outputType
;
int32_t
outputLen
;
int32_t
outputLen
;
int32_t
bufSize
;
int32_t
bufSize
;
char
*
path
;
char
*
path
;
}
SUdfInfo
;
}
S
Script
UdfInfo
;
typedef
int32_t
(
*
TScriptUdfScalarProcFunc
)(
SUdfDataBlock
*
block
,
SUdfColumn
*
resultCol
,
void
*
udfCtx
);
typedef
int32_t
(
*
TScriptUdfScalarProcFunc
)(
SUdfDataBlock
*
block
,
SUdfColumn
*
resultCol
,
void
*
udfCtx
);
...
@@ -282,7 +287,7 @@ typedef int32_t (*TScriptUdfAggProcessFunc)(SUdfDataBlock *block, SUdfInterBuf *
...
@@ -282,7 +287,7 @@ typedef int32_t (*TScriptUdfAggProcessFunc)(SUdfDataBlock *block, SUdfInterBuf *
typedef
int32_t
(
*
TScriptUdfAggMergeFunc
)(
SUdfInterBuf
*
inputBuf1
,
SUdfInterBuf
*
inputBuf2
,
SUdfInterBuf
*
outputBuf
,
typedef
int32_t
(
*
TScriptUdfAggMergeFunc
)(
SUdfInterBuf
*
inputBuf1
,
SUdfInterBuf
*
inputBuf2
,
SUdfInterBuf
*
outputBuf
,
void
*
udfCtx
);
void
*
udfCtx
);
typedef
int32_t
(
*
TScriptUdfAggFinishFunc
)(
SUdfInterBuf
*
buf
,
SUdfInterBuf
*
resultData
,
void
*
udfCtx
);
typedef
int32_t
(
*
TScriptUdfAggFinishFunc
)(
SUdfInterBuf
*
buf
,
SUdfInterBuf
*
resultData
,
void
*
udfCtx
);
typedef
int32_t
(
*
TScriptUdfInitFunc
)(
SUdfInfo
*
info
,
void
**
pUdfCtx
);
typedef
int32_t
(
*
TScriptUdfInitFunc
)(
S
Script
UdfInfo
*
info
,
void
**
pUdfCtx
);
typedef
int32_t
(
*
TScriptUdfDestoryFunc
)(
void
*
udfCtx
);
typedef
int32_t
(
*
TScriptUdfDestoryFunc
)(
void
*
udfCtx
);
// the following function is for open/close script plugin.
// the following function is for open/close script plugin.
...
...
source/libs/function/src/udfd.c
浏览文件 @
8e5dae39
...
@@ -52,7 +52,7 @@ int32_t udfdCPluginOpen(void *scriptCtx) { return 0; }
...
@@ -52,7 +52,7 @@ int32_t udfdCPluginOpen(void *scriptCtx) { return 0; }
int32_t
udfdCPluginClose
(
void
*
scriptCtx
)
{
return
0
;
}
int32_t
udfdCPluginClose
(
void
*
scriptCtx
)
{
return
0
;
}
int32_t
udfdCPluginUdfInit
(
SUdfInfo
*
udf
,
void
**
pUdfCtx
)
{
int32_t
udfdCPluginUdfInit
(
S
Script
UdfInfo
*
udf
,
void
**
pUdfCtx
)
{
int32_t
err
=
0
;
int32_t
err
=
0
;
SUdfCPluginCtx
*
udfCtx
=
taosMemoryCalloc
(
1
,
sizeof
(
SUdfCPluginCtx
));
SUdfCPluginCtx
*
udfCtx
=
taosMemoryCalloc
(
1
,
sizeof
(
SUdfCPluginCtx
));
err
=
uv_dlopen
(
udf
->
path
,
&
udfCtx
->
lib
);
err
=
uv_dlopen
(
udf
->
path
,
&
udfCtx
->
lib
);
...
@@ -73,11 +73,11 @@ int32_t udfdCPluginUdfInit(SUdfInfo *udf, void **pUdfCtx) {
...
@@ -73,11 +73,11 @@ int32_t udfdCPluginUdfInit(SUdfInfo *udf, void **pUdfCtx) {
strncat
(
destroyFuncName
,
destroySuffix
,
strlen
(
destroySuffix
));
strncat
(
destroyFuncName
,
destroySuffix
,
strlen
(
destroySuffix
));
uv_dlsym
(
&
udfCtx
->
lib
,
destroyFuncName
,
(
void
**
)(
&
udfCtx
->
destroyFunc
));
uv_dlsym
(
&
udfCtx
->
lib
,
destroyFuncName
,
(
void
**
)(
&
udfCtx
->
destroyFunc
));
if
(
udf
->
funcType
==
TSDB
_FUNC_TYPE_SCALAR
)
{
if
(
udf
->
funcType
==
UDF
_FUNC_TYPE_SCALAR
)
{
char
processFuncName
[
TSDB_FUNC_NAME_LEN
]
=
{
0
};
char
processFuncName
[
TSDB_FUNC_NAME_LEN
]
=
{
0
};
strcpy
(
processFuncName
,
udfName
);
strcpy
(
processFuncName
,
udfName
);
uv_dlsym
(
&
udfCtx
->
lib
,
processFuncName
,
(
void
**
)(
&
udfCtx
->
scalarProcFunc
));
uv_dlsym
(
&
udfCtx
->
lib
,
processFuncName
,
(
void
**
)(
&
udfCtx
->
scalarProcFunc
));
}
else
if
(
udf
->
funcType
==
TSDB_FUNC_TYPE_AGGREGATE
)
{
}
else
if
(
udf
->
funcType
==
UDF_FUNC_TYPE_AGG
)
{
char
processFuncName
[
TSDB_FUNC_NAME_LEN
]
=
{
0
};
char
processFuncName
[
TSDB_FUNC_NAME_LEN
]
=
{
0
};
strcpy
(
processFuncName
,
udfName
);
strcpy
(
processFuncName
,
udfName
);
uv_dlsym
(
&
udfCtx
->
lib
,
processFuncName
,
(
void
**
)(
&
udfCtx
->
aggProcFunc
));
uv_dlsym
(
&
udfCtx
->
lib
,
processFuncName
,
(
void
**
)(
&
udfCtx
->
aggProcFunc
));
...
@@ -411,9 +411,13 @@ void udfdProcessRequest(uv_work_t *req) {
...
@@ -411,9 +411,13 @@ void udfdProcessRequest(uv_work_t *req) {
}
}
}
}
void
convertUdf2UdfInfo
(
SUdf
*
udf
,
SUdfInfo
*
udfInfo
)
{
void
convertUdf2UdfInfo
(
SUdf
*
udf
,
S
Script
UdfInfo
*
udfInfo
)
{
udfInfo
->
bufSize
=
udf
->
bufSize
;
udfInfo
->
bufSize
=
udf
->
bufSize
;
udfInfo
->
funcType
=
udf
->
funcType
;
if
(
udf
->
funcType
==
TSDB_FUNC_TYPE_AGGREGATE
)
{
udfInfo
->
funcType
=
UDF_FUNC_TYPE_AGG
;
}
else
if
(
udf
->
funcType
==
TSDB_FUNC_TYPE_SCALAR
)
{
udfInfo
->
funcType
=
UDF_FUNC_TYPE_SCALAR
;
}
udfInfo
->
name
=
udf
->
name
;
udfInfo
->
name
=
udf
->
name
;
udfInfo
->
outputLen
=
udf
->
outputLen
;
udfInfo
->
outputLen
=
udf
->
outputLen
;
udfInfo
->
outputType
=
udf
->
outputType
;
udfInfo
->
outputType
=
udf
->
outputType
;
...
@@ -438,7 +442,7 @@ int32_t udfdInitUdf(char *udfName, SUdf *udf) {
...
@@ -438,7 +442,7 @@ int32_t udfdInitUdf(char *udfName, SUdf *udf) {
}
}
uv_mutex_unlock
(
&
global
.
scriptPluginsMutex
);
uv_mutex_unlock
(
&
global
.
scriptPluginsMutex
);
udf
->
scriptPlugin
=
scriptPlugin
;
udf
->
scriptPlugin
=
scriptPlugin
;
SUdfInfo
info
=
{
0
};
S
Script
UdfInfo
info
=
{
0
};
convertUdf2UdfInfo
(
udf
,
&
info
);
convertUdf2UdfInfo
(
udf
,
&
info
);
udf
->
scriptPlugin
->
udfInitFunc
(
&
info
,
&
udf
->
scriptUdfCtx
);
udf
->
scriptPlugin
->
udfInitFunc
(
&
info
,
&
udf
->
scriptUdfCtx
);
return
0
;
return
0
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录