Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
lzh_me
TDengine
提交
b1c6a7d4
TDengine
项目概览
lzh_me
/
TDengine
与 Fork 源项目一致
Fork自
taosdata / TDengine
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
TDengine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
b1c6a7d4
编写于
2月 06, 2021
作者:
sangshuduo
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
[TD-2771] <feature>: python taosdemo. add query func.
上级
32f57e5b
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
42 addition
and
52 deletion
+42
-52
tests/examples/python/taosdemo/taosdemo.py
tests/examples/python/taosdemo/taosdemo.py
+42
-52
未找到文件。
tests/examples/python/taosdemo/taosdemo.py
浏览文件 @
b1c6a7d4
...
...
@@ -133,62 +133,52 @@ def query_func(process: int, thread: int, cmd: str):
host
,
port
,
user
,
password
,
cmd
)
def
query_data_process
(
q_lock
,
i
:
int
,
cmd
:
str
):
time
.
sleep
(
0.01
)
v_print
(
"Process:%d threads: %d cmd: %s"
,
i
,
threads
,
cmd
)
q_lock
.
aquire
()
cursor
.
execute
(
cmd
)
q_lock
.
release
()
return
i
def
query_data
(
cmd
:
str
):
v_print
(
"query_data processes: %d, cmd: %s"
,
processes
,
cmd
)
q_lock
=
Lock
()
pool
=
Pool
(
processes
)
for
i
in
range
(
processes
):
pool
.
apply_async
(
query_data_process
,
args
=
(
q_lock
,
i
,
cmd
))
# time.sleep(1)
pool
.
close
()
pool
.
join
()
def
insert_data
(
processes
:
int
):
i_lock
=
Lock
()
pool
=
Pool
(
processes
)
begin
=
0
end
=
0
quotient
=
numOfTb
//
processes
if
quotient
<
1
:
processes
=
numOfTb
quotient
=
1
def
query_data_process
(
cmd
:
str
):
# establish connection if native
if
native
:
v_print
(
"host:%s, user:%s passwd:%s configDir:%s "
,
host
,
user
,
password
,
configDir
)
try
:
conn
=
taos
.
connect
(
host
=
host
,
user
=
user
,
password
=
password
,
config
=
configDir
)
print
(
"conn: %s"
%
str
(
conn
.
__class__
))
except
Exception
as
e
:
print
(
"Error: %s"
%
e
.
args
[
0
])
sys
.
exit
(
1
)
remainder
=
numOfTb
%
processes
v_print
(
"insert_data num of tables: %d, quotient: %d, remainder: %d"
,
numOfTb
,
quotient
,
remainder
)
try
:
cursor
=
conn
.
cursor
()
print
(
"cursor:%d %s"
%
(
id
(
cursor
),
str
(
cursor
.
__class__
)))
except
Exception
as
e
:
print
(
"Error: %s"
%
e
.
args
[
0
])
sys
.
exit
(
1
)
for
i
in
range
(
processes
):
begin
=
end
if
native
:
try
:
cursor
.
execute
(
cmd
)
cols
=
cursor
.
description
data
=
cursor
.
fetchall
()
if
i
<
remainder
:
end
=
begin
+
quotient
+
1
else
:
end
=
begin
+
quotient
for
col
in
data
:
print
(
col
)
except
Exception
as
e
:
conn
.
close
()
print
(
"Error: %s"
%
e
.
args
[
0
])
sys
.
exit
(
1
)
v_print
(
"insert_data Process %d from %d to %d"
,
i
,
begin
,
end
)
pool
.
apply_async
(
insert_data_process
,
args
=
(
i_lock
,
i
,
begin
,
end
))
else
:
restful_execute
(
host
,
port
,
user
,
password
,
cmd
)
pool
.
close
()
pool
.
join
()
if
native
:
cursor
.
close
()
conn
.
close
()
def
create_stb
():
...
...
@@ -795,7 +785,7 @@ if __name__ == "__main__":
# query data
if
queryCmd
!=
"NO"
:
print
(
"queryCmd: %s"
%
queryCmd
)
query_data
(
queryCmd
)
query_data
_process
(
queryCmd
)
if
measure
:
end_time
=
time
.
time
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录