Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
小楼昨夜-听风雨
TDengine
提交
ea3cde91
T
TDengine
项目概览
小楼昨夜-听风雨
/
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
ea3cde91
编写于
3月 25, 2021
作者:
S
Shengliang Guan
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add shell tools
上级
bee74151
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
213 addition
and
2 deletion
+213
-2
src/kit/shell/CMakeLists.txt
src/kit/shell/CMakeLists.txt
+1
-0
src/kit/shell/inc/shell.h
src/kit/shell/inc/shell.h
+2
-0
src/kit/shell/src/shellCheck.c
src/kit/shell/src/shellCheck.c
+199
-0
src/kit/shell/src/shellEngine.c
src/kit/shell/src/shellEngine.c
+7
-2
src/kit/shell/src/shellLinux.c
src/kit/shell/src/shellLinux.c
+4
-0
未找到文件。
src/kit/shell/CMakeLists.txt
浏览文件 @
ea3cde91
...
...
@@ -36,6 +36,7 @@ ELSEIF (TD_DARWIN)
LIST
(
APPEND SRC ./src/shellDarwin.c
)
LIST
(
APPEND SRC ./src/shellCommand.c
)
LIST
(
APPEND SRC ./src/shellImport.c
)
LIST
(
APPEND SRC ./src/shellCheck.c
)
ADD_EXECUTABLE
(
shell
${
SRC
}
)
# linking with dylib
TARGET_LINK_LIBRARIES
(
shell taos
)
...
...
src/kit/shell/inc/shell.h
浏览文件 @
ea3cde91
...
...
@@ -52,6 +52,7 @@ typedef struct SShellArguments {
char
dir
[
TSDB_FILENAME_LEN
];
int
threadNum
;
char
*
commands
;
int
check
;
int
abort
;
int
port
;
int
pktLen
;
...
...
@@ -72,6 +73,7 @@ void write_history();
void
source_file
(
TAOS
*
con
,
char
*
fptr
);
void
source_dir
(
TAOS
*
con
,
SShellArguments
*
args
);
void
get_history_path
(
char
*
history
);
void
shellCheck
(
TAOS
*
con
,
SShellArguments
*
args
);
void
cleanup_handler
(
void
*
arg
);
void
exitShell
();
int
shellDumpResult
(
TAOS_RES
*
con
,
char
*
fname
,
int
*
error_no
,
bool
printMode
);
...
...
src/kit/shell/src/shellCheck.c
0 → 100644
浏览文件 @
ea3cde91
/*
* 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/>.
*/
#define _GNU_SOURCE
#define _XOPEN_SOURCE
#define _DEFAULT_SOURCE
#include "os.h"
#include "shell.h"
#include "shellCommand.h"
#include "tglobal.h"
#include "tutil.h"
#define SHELL_SQL_LEN 1024
static
int32_t
tbNum
=
0
;
static
int32_t
tbMallocNum
=
0
;
static
char
**
tbNames
=
NULL
;
static
int32_t
checkedNum
=
0
;
static
int32_t
errorNum
=
0
;
typedef
struct
{
pthread_t
threadID
;
int
threadIndex
;
int
totalThreads
;
void
*
taos
;
char
*
db
;
}
ShellThreadObj
;
static
int32_t
shellUseDb
(
TAOS
*
con
,
char
*
db
)
{
if
(
db
==
NULL
)
{
fprintf
(
stdout
,
"no dbname input
\n
"
);
return
-
1
;
}
char
sql
[
SHELL_SQL_LEN
]
=
{
0
};
snprintf
(
sql
,
SHELL_SQL_LEN
,
"use %s"
,
db
);
TAOS_RES
*
pSql
=
taos_query
(
con
,
sql
);
int32_t
code
=
taos_errno
(
pSql
);
if
(
code
!=
0
)
{
fprintf
(
stdout
,
"failed to execute sql:%s since %s"
,
sql
,
taos_errstr
(
pSql
));
}
taos_free_result
(
pSql
);
return
code
;
}
static
int32_t
shellShowTables
(
TAOS
*
con
,
char
*
db
)
{
char
sql
[
SHELL_SQL_LEN
]
=
{
0
};
snprintf
(
sql
,
SHELL_SQL_LEN
,
"show %s.tables"
,
db
);
TAOS_RES
*
pSql
=
taos_query
(
con
,
sql
);
int32_t
code
=
taos_errno
(
pSql
);
if
(
code
!=
0
)
{
fprintf
(
stdout
,
"failed to execute sql:%s since %s
\n
"
,
sql
,
taos_errstr
(
pSql
));
}
else
{
TAOS_ROW
row
;
while
((
row
=
taos_fetch_row
(
pSql
)))
{
int32_t
tbIndex
=
tbNum
++
;
if
(
tbMallocNum
<
tbNum
)
{
tbMallocNum
=
(
tbMallocNum
*
2
+
1
);
tbNames
=
realloc
(
tbNames
,
tbMallocNum
*
sizeof
(
char
*
));
if
(
tbNames
==
NULL
)
{
fprintf
(
stdout
,
"failed to malloc tablenames, num:%d
\n
"
,
tbMallocNum
);
code
=
TSDB_CODE_TSC_OUT_OF_MEMORY
;
break
;
}
}
tbNames
[
tbIndex
]
=
malloc
(
TSDB_TABLE_NAME_LEN
);
strncpy
(
tbNames
[
tbIndex
],
(
const
char
*
)
row
[
0
],
TSDB_TABLE_NAME_LEN
);
if
(
tbIndex
%
100000
==
0
&&
tbIndex
!=
0
)
{
fprintf
(
stdout
,
"%d tablenames fetched
\n
"
,
tbIndex
);
}
}
}
taos_free_result
(
pSql
);
fprintf
(
stdout
,
"total %d tablenames fetched, over
\n
"
,
tbNum
);
return
code
;
}
static
void
shellFreeTbnames
()
{
for
(
int32_t
i
=
0
;
i
<
tbNum
;
++
i
)
{
free
(
tbNames
[
i
]);
}
free
(
tbNames
);
}
static
void
*
shellCheckThreadFp
(
void
*
arg
)
{
ShellThreadObj
*
pThread
=
(
ShellThreadObj
*
)
arg
;
int32_t
interval
=
tbNum
/
pThread
->
totalThreads
+
1
;
int32_t
start
=
pThread
->
threadIndex
*
interval
;
int32_t
end
=
(
pThread
->
threadIndex
+
1
)
*
interval
;
if
(
end
>
tbNum
)
end
=
tbNum
+
1
;
char
file
[
32
]
=
{
0
};
snprintf
(
file
,
32
,
"tb%d.txt"
,
pThread
->
threadIndex
);
FILE
*
fp
=
fopen
(
file
,
"w"
);
if
(
!
fp
)
{
fprintf
(
stdout
,
"failed to open %s, reason:%s"
,
file
,
strerror
(
errno
));
return
NULL
;
}
char
sql
[
SHELL_SQL_LEN
];
for
(
int32_t
t
=
start
;
t
<
end
;
++
t
)
{
char
*
tbname
=
tbNames
[
t
];
if
(
tbname
==
NULL
)
break
;
snprintf
(
sql
,
SHELL_SQL_LEN
,
"select * from %s limit 1"
,
tbname
);
TAOS_RES
*
pSql
=
taos_query
(
pThread
->
taos
,
sql
);
int32_t
code
=
taos_errno
(
pSql
);
if
(
code
!=
0
)
{
int32_t
len
=
snprintf
(
sql
,
SHELL_SQL_LEN
,
"drop table %s.%s;
\n
"
,
pThread
->
db
,
tbname
);
fwrite
(
sql
,
1
,
len
,
fp
);
atomic_add_fetch_32
(
&
errorNum
,
1
);
}
int32_t
cnum
=
atomic_add_fetch_32
(
&
checkedNum
,
1
);
if
(
cnum
%
5000
==
0
&&
cnum
!=
0
)
{
fprintf
(
stdout
,
"%d tables checked
\n
"
,
cnum
);
}
taos_free_result
(
pSql
);
}
fsync
(
fileno
(
fp
));
fclose
(
fp
);
return
NULL
;
}
static
void
shellRunCheckThreads
(
TAOS
*
con
,
SShellArguments
*
args
)
{
pthread_attr_t
thattr
;
ShellThreadObj
*
threadObj
=
(
ShellThreadObj
*
)
calloc
(
args
->
threadNum
,
sizeof
(
ShellThreadObj
));
for
(
int
t
=
0
;
t
<
args
->
threadNum
;
++
t
)
{
ShellThreadObj
*
pThread
=
threadObj
+
t
;
pThread
->
threadIndex
=
t
;
pThread
->
totalThreads
=
args
->
threadNum
;
pThread
->
taos
=
con
;
pThread
->
db
=
args
->
database
;
pthread_attr_init
(
&
thattr
);
pthread_attr_setdetachstate
(
&
thattr
,
PTHREAD_CREATE_JOINABLE
);
if
(
pthread_create
(
&
(
pThread
->
threadID
),
&
thattr
,
shellCheckThreadFp
,
(
void
*
)
pThread
)
!=
0
)
{
fprintf
(
stderr
,
"ERROR: thread:%d failed to start
\n
"
,
pThread
->
threadIndex
);
exit
(
0
);
}
}
for
(
int
t
=
0
;
t
<
args
->
threadNum
;
++
t
)
{
pthread_join
(
threadObj
[
t
].
threadID
,
NULL
);
}
for
(
int
t
=
0
;
t
<
args
->
threadNum
;
++
t
)
{
taos_close
(
threadObj
[
t
].
taos
);
}
free
(
threadObj
);
}
void
shellCheck
(
TAOS
*
con
,
SShellArguments
*
args
)
{
int64_t
start
=
taosGetTimestampMs
();
if
(
shellUseDb
(
con
,
args
->
database
)
!=
0
)
{
shellFreeTbnames
();
return
;
}
if
(
shellShowTables
(
con
,
args
->
database
)
!=
0
)
{
shellFreeTbnames
();
return
;
}
fprintf
(
stdout
,
"total %d tables will be checked by %d threads
\n
"
,
tbNum
,
args
->
threadNum
);
shellRunCheckThreads
(
con
,
args
);
int64_t
end
=
taosGetTimestampMs
();
fprintf
(
stdout
,
"total %d tables checked, failed:%d, time spent %.2f seconds
\n
"
,
checkedNum
,
errorNum
,
(
end
-
start
)
/
1000
.
0
);
}
\ No newline at end of file
src/kit/shell/src/shellEngine.c
浏览文件 @
ea3cde91
...
...
@@ -121,12 +121,17 @@ TAOS *shellInit(SShellArguments *args) {
taos_close
(
con
);
exit
(
EXIT_SUCCESS
);
}
if
(
args
->
check
!=
0
)
{
shellCheck
(
con
,
args
);
taos_close
(
con
);
exit
(
EXIT_SUCCESS
);
}
#endif
return
con
;
}
static
bool
isEmptyCommand
(
const
char
*
cmd
)
{
for
(
char
c
=
*
cmd
++
;
c
!=
0
;
c
=
*
cmd
++
)
{
if
(
c
!=
' '
&&
c
!=
'\t'
&&
c
!=
';'
)
{
...
...
@@ -412,7 +417,7 @@ static char* formatTimestamp(char* buf, int64_t val, int precision) {
#ifdef WINDOWS
if
(
tt
<
0
)
tt
=
0
;
#endif
if
(
tt
<
0
&&
ms
!=
0
)
{
if
(
tt
<
=
0
&&
ms
!=
0
)
{
tt
--
;
if
(
precision
==
TSDB_TIME_PRECISION_MICRO
)
{
ms
+=
1000000
;
...
...
src/kit/shell/src/shellLinux.c
浏览文件 @
ea3cde91
...
...
@@ -45,6 +45,7 @@ static struct argp_option options[] = {
{
"file"
,
'f'
,
"FILE"
,
0
,
"Script to run without enter the shell."
},
{
"directory"
,
'D'
,
"DIRECTORY"
,
0
,
"Use multi-thread to import all SQL files in the directory separately."
},
{
"thread"
,
'T'
,
"THREADNUM"
,
0
,
"Number of threads when using multi-thread to import data."
},
{
"check"
,
'k'
,
"CHECK"
,
0
,
"Check tables."
},
{
"database"
,
'd'
,
"DATABASE"
,
0
,
"Database to use when connecting to the server."
},
{
"timezone"
,
't'
,
"TIMEZONE"
,
0
,
"Time zone of the shell, default is local."
},
{
"netrole"
,
'n'
,
"NETROLE"
,
0
,
"Net role when network connectivity test, default is startup, options: client|server|rpc|startup|sync."
},
...
...
@@ -130,6 +131,9 @@ static error_t parse_opt(int key, char *arg, struct argp_state *state) {
return
-
1
;
}
break
;
case
'k'
:
arguments
->
check
=
atoi
(
arg
);
break
;
case
'd'
:
arguments
->
database
=
arg
;
break
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录