Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
慢慢CG
TDengine
提交
0d738dc6
T
TDengine
项目概览
慢慢CG
/
TDengine
与 Fork 源项目一致
Fork自
taosdata / TDengine
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
TDengine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
0d738dc6
编写于
5月 24, 2021
作者:
M
Minglei Jin
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
[TD-4270]<feature>: the first round rest udf implementation
上级
5fd5e775
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
87 addition
and
1 deletion
+87
-1
src/plugins/http/inc/httpParser.h
src/plugins/http/inc/httpParser.h
+1
-1
src/plugins/http/src/httpRestHandle.c
src/plugins/http/src/httpRestHandle.c
+86
-0
未找到文件。
src/plugins/http/inc/httpParser.h
浏览文件 @
0d738dc6
...
...
@@ -17,7 +17,7 @@
#define HTTP_PARSER_H
#include "httpGzip.h"
#define HTTP_MAX_URL
5
// http url stack size
#define HTTP_MAX_URL
6
// http url stack size
typedef
enum
HTTP_PARSER_STATE
{
HTTP_PARSER_BEGIN
,
...
...
src/plugins/http/src/httpRestHandle.c
浏览文件 @
0d738dc6
...
...
@@ -119,6 +119,90 @@ bool restProcessSqlRequest(HttpContext* pContext, int32_t timestampFmt) {
return
true
;
}
#define REST_FUNC_URL_POS 2
#define REST_OUTP_URL_POS 3
#define REST_AGGR_URL_POS 4
#define REST_BUFF_URL_POS 5
#define HTTP_FUNC_LEN 32
#define HTTP_OUTP_LEN 16
#define HTTP_AGGR_LEN 2
#define HTTP_BUFF_LEN 32
static
int
udfSaveFile
(
const
char
*
fname
,
const
char
*
content
,
long
len
)
{
int
fd
=
open
(
fname
,
O_WRONLY
|
O_CREAT
|
O_EXCL
|
O_BINARY
,
0755
);
if
(
fd
<
0
)
return
-
1
;
if
(
taosWrite
(
fd
,
(
void
*
)
content
,
len
)
<
0
)
return
-
1
;
return
0
;
}
static
bool
restProcessUdfRequest
(
HttpContext
*
pContext
)
{
HttpParser
*
pParser
=
pContext
->
parser
;
if
(
pParser
->
path
[
REST_FUNC_URL_POS
].
pos
>=
HTTP_FUNC_LEN
||
pParser
->
path
[
REST_FUNC_URL_POS
].
pos
<=
0
)
{
return
false
;
}
if
(
pParser
->
path
[
REST_OUTP_URL_POS
].
pos
>=
HTTP_OUTP_LEN
||
pParser
->
path
[
REST_OUTP_URL_POS
].
pos
<=
0
)
{
return
false
;
}
if
(
pParser
->
path
[
REST_AGGR_URL_POS
].
pos
>=
HTTP_AGGR_LEN
||
pParser
->
path
[
REST_AGGR_URL_POS
].
pos
<=
0
)
{
return
false
;
}
if
(
pParser
->
path
[
REST_BUFF_URL_POS
].
pos
>=
HTTP_BUFF_LEN
||
pParser
->
path
[
REST_BUFF_URL_POS
].
pos
<=
0
)
{
return
false
;
}
char
*
sql
=
pContext
->
parser
->
body
.
str
;
int
len
=
pContext
->
parser
->
body
.
size
;
if
(
sql
==
NULL
)
{
httpSendErrorResp
(
pContext
,
TSDB_CODE_HTTP_NO_SQL_INPUT
);
return
false
;
}
char
udfDir
[
256
]
=
{
0
};
char
buf
[
64
]
=
"udf-"
;
char
funcName
[
64
]
=
{
0
};
int
aggr
=
0
;
char
outputType
[
16
]
=
{
0
};
char
buffSize
[
32
]
=
{
0
};
tstrncpy
(
funcName
,
pParser
->
path
[
REST_FUNC_URL_POS
].
str
,
HTTP_FUNC_LEN
);
tstrncpy
(
buf
+
4
,
funcName
,
HTTP_FUNC_LEN
);
if
(
pParser
->
path
[
REST_AGGR_URL_POS
].
str
[
0
]
!=
'0'
)
{
aggr
=
1
;
}
tstrncpy
(
outputType
,
pParser
->
path
[
REST_OUTP_URL_POS
].
str
,
HTTP_OUTP_LEN
);
tstrncpy
(
buffSize
,
pParser
->
path
[
REST_BUFF_URL_POS
].
str
,
HTTP_BUFF_LEN
);
taosGetTmpfilePath
(
funcName
,
udfDir
);
udfSaveFile
(
udfDir
,
sql
,
len
);
tfree
(
sql
);
pContext
->
parser
->
body
.
str
=
malloc
(
1024
);
sql
=
pContext
->
parser
->
body
.
str
;
int
sqlLen
=
sprintf
(
sql
,
"create %s function %s as
\"
%s
\"
outputtype %s bufsize %s"
,
aggr
==
1
?
"aggregate"
:
" "
,
funcName
,
udfDir
,
outputType
,
buffSize
);
pContext
->
parser
->
body
.
pos
=
sqlLen
;
pContext
->
parser
->
body
.
size
=
sqlLen
+
1
;
HttpSqlCmd
*
cmd
=
&
(
pContext
->
singleCmd
);
cmd
->
nativSql
=
sql
;
pContext
->
reqType
=
HTTP_REQTYPE_SINGLE_SQL
;
pContext
->
encodeMethod
=
&
restEncodeSqlLocalTimeStringMethod
;
return
true
;
}
bool
restProcessRequest
(
struct
HttpContext
*
pContext
)
{
if
(
httpUrlMatch
(
pContext
,
REST_ACTION_URL_POS
,
"login"
))
{
restGetUserFromUrl
(
pContext
);
...
...
@@ -138,6 +222,8 @@ bool restProcessRequest(struct HttpContext* pContext) {
return
restProcessSqlRequest
(
pContext
,
REST_TIMESTAMP_FMT_UTC_STRING
);
}
else
if
(
httpUrlMatch
(
pContext
,
REST_ACTION_URL_POS
,
"login"
))
{
return
restProcessLoginRequest
(
pContext
);
}
else
if
(
httpUrlMatch
(
pContext
,
REST_ACTION_URL_POS
,
"udf"
))
{
return
restProcessUdfRequest
(
pContext
);
}
else
{
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录