mirror of
https://github.com/taosdata/TDengine
synced 2026-05-24 10:09:01 +00:00
f1b45f2298
* fix: add abnormal exit logic * fix: add interrupt exit mechanism * fix: add insert error case * fix: add Mac memory statistics * fix: multi threaded table building with skip logic added * fix: multi threaded table building increases task allocation * fix: add csv tag file multi-threaded table creation case * fix: add 'contrl c' to tmq and exit * fix: add 'contrl c' cases * fix: add 'contrl c' cases * fix: add tmq cancel cases * fix: add ci execute cases * fix: restore mistakenly deleted files --------- Co-authored-by: taos-support <it@taosdata.com>
108 lines
No EOL
3.2 KiB
Python
108 lines
No EOL
3.2 KiB
Python
###################################################################
|
|
# Copyright (c) 2016 by TAOS Technologies, Inc.
|
|
# All rights reserved.
|
|
#
|
|
# This file is proprietary and confidential to TAOS Technologies.
|
|
# No part of this file may be reproduced, stored, transmitted,
|
|
# disclosed or used in any form or by any means other than as
|
|
# expressly provided by the written permission from Jianhui Tao
|
|
#
|
|
###################################################################
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import sys
|
|
import time
|
|
import random
|
|
|
|
import taos
|
|
import frame
|
|
import frame.etool
|
|
import json
|
|
import threading
|
|
import signal
|
|
import psutil
|
|
|
|
from frame.log import *
|
|
from frame.cases import *
|
|
from frame.sql import *
|
|
from frame.caseBase import *
|
|
from frame import *
|
|
from frame.autogen import *
|
|
from frame.srvCtl import *
|
|
|
|
class TDTestCase(TBase):
|
|
updatecfgDict = {
|
|
'slowLogScope' : "others"
|
|
}
|
|
|
|
def init(self, conn, logSql, replicaVar=1):
|
|
tdLog.info(f"start to init {__file__}")
|
|
self.replicaVar = int(replicaVar)
|
|
tdSql.init(conn.cursor(), logSql) # output sql.txt file
|
|
self._rlist = None
|
|
|
|
def get_pids_by_name(self, process_name):
|
|
pids = []
|
|
for proc in psutil.process_iter(['name']):
|
|
if proc.info['name'] == process_name:
|
|
pids.append(proc.pid)
|
|
return pids
|
|
|
|
def stopThread(self, isForceExit):
|
|
tdLog.info("dnodeNodeStopThread start")
|
|
time.sleep(10)
|
|
pids = self.get_pids_by_name("taosBenchmark")
|
|
if pids:
|
|
tdLog.info(f"Find a process named taosBbenchmark with PID: {pids}")
|
|
else:
|
|
tdLog.exit("No process named taosBbenchmark was found.")
|
|
|
|
os.kill(pids[0], signal.SIGINT)
|
|
if isForceExit:
|
|
os.kill(pids[0], signal.SIGINT)
|
|
|
|
time.sleep(10)
|
|
|
|
if self._rlist:
|
|
tdLog.info(self._rlist)
|
|
if isForceExit:
|
|
self.checkListString(self._rlist, "Benchmark process forced exit!")
|
|
else:
|
|
self.checkListString(self._rlist, "Receive SIGINT or other signal, quit benchmark")
|
|
else:
|
|
tdLog.exit("The benchmark process has not stopped!")
|
|
|
|
|
|
def dbInsertThread(self):
|
|
tdLog.info(f"dbInsertThread start")
|
|
# taosBenchmark run
|
|
cmd = "-d db -t 10000 -n 10000 -T 8 -I stmt -y"
|
|
self._rlist = self.benchmark(cmd, checkRun=False)
|
|
|
|
# run
|
|
def run(self):
|
|
tdLog.info(f"start to excute {__file__}")
|
|
t1 = threading.Thread(target=self.dbInsertThread)
|
|
t2 = threading.Thread(target=self.stopThread, args=(False,))
|
|
t1.start()
|
|
t2.start()
|
|
tdLog.success(f"{__file__} successfully executed")
|
|
t1.join()
|
|
t2.join()
|
|
|
|
t1 = threading.Thread(target=self.dbInsertThread)
|
|
t2 = threading.Thread(target=self.stopThread, args=(True,))
|
|
t1.start()
|
|
t2.start()
|
|
tdLog.success(f"{__file__} successfully executed")
|
|
t1.join()
|
|
t2.join()
|
|
|
|
def stop(self):
|
|
tdSql.close()
|
|
tdLog.success(f"{__file__} successfully executed")
|
|
|
|
|
|
tdCases.addWindows(__file__, TDTestCase())
|
|
tdCases.addLinux(__file__, TDTestCase()) |