xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

test(stream): update the test script.

This commit is contained in:
Haojun Liao 2025-01-03 11:01:32 +08:00
parent 055441d8fb
commit 74aead7be3
1 changed files with 22 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
tests/perf-test/stream.py
View File

@ -21,25 +21,29 @@ class MonitorSystemLoad:
def get_proc_status(self): def get_proc_status(self):
process = psutil.Process(self.pid) process = psutil.Process(self.pid)
while True: with open('/tmp/pref.txt', 'w+') as f:
cpu_percent = process.cpu_percent(interval=1) while True:
cpu_percent = process.cpu_percent(interval=1)
memory_info = process.memory_info() memory_info = process.memory_info()
memory_percent = process.memory_percent() memory_percent = process.memory_percent()
io_counters = process.io_counters() io_counters = process.io_counters()
sys_load = psutil.getloadavg() sys_load = psutil.getloadavg()
print("load: %s, CPU:%s, Mem:%.2f MiB(%.2f%%), Read: %.2fMiB(%d), Write: %.2fMib (%d)" % ( s = "load: %.2f, CPU:%s, Mem:%.2f MiB(%.2f%%), Read: %.2fMiB(%d), Write: %.2fMib (%d)" % (
sys_load, cpu_percent, memory_info.rss / 1048576.0, sys_load[0], cpu_percent, memory_info.rss / 1048576.0,
memory_percent, io_counters.read_bytes / 1048576.0, io_counters.read_count, memory_percent, io_counters.read_bytes / 1048576.0, io_counters.read_count,
io_counters.write_bytes / 1048576.0, io_counters.write_count)) io_counters.write_bytes / 1048576.0, io_counters.write_count)
time.sleep(1) print(s)
self.count -= 1 f.write(s + '\n')
if self.count <= 0: time.sleep(1)
break
self.count -= 1
if self.count <= 0:
break
class StreamStarter: class StreamStarter:
@ -60,11 +64,11 @@ class StreamStarter:
"rest_port": 6041, "rest_port": 6041,
"user": "root", "user": "root",
"password": "taosdata", "password": "taosdata",
"thread_count": 20, "thread_count": 5,
"create_table_thread_count": 40, "create_table_thread_count": 10,
"result_file": "/tmp/taosBenchmark_result.log", "result_file": "/tmp/taosBenchmark_result.log",
"confirm_parameter_prompt": "no", "confirm_parameter_prompt": "no",
"insert_interval": 0, "insert_interval": 1000,
"num_of_records_per_req": 10000, "num_of_records_per_req": 10000,
"max_sql_len": 1024000, "max_sql_len": 1024000,
"databases": [ "databases": [
@ -87,7 +91,7 @@ class StreamStarter:
{ {
"name": "stb", "name": "stb",
"child_table_exists": "yes", "child_table_exists": "yes",
"childtable_count": 500, "childtable_count": 50000,
"childtable_prefix": "ctb0_", "childtable_prefix": "ctb0_",
"escape_character": "no", "escape_character": "no",
"auto_create_table": "yes", "auto_create_table": "yes",