mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-02 06:30:55 +00:00
96 lines
3.3 KiB
Python
Executable File
96 lines
3.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from multiprocessing import cpu_count
|
|
from subprocess import Popen, call, STDOUT
|
|
import os
|
|
import sys
|
|
import shutil
|
|
import argparse
|
|
import logging
|
|
import time
|
|
|
|
|
|
def get_skip_list_cmd(path):
|
|
with open(path, 'r') as f:
|
|
for line in f:
|
|
if '--use-skip-list' in line:
|
|
return '--use-skip-list'
|
|
return ''
|
|
|
|
|
|
def get_options(i):
|
|
options = ""
|
|
if 0 < i:
|
|
options += " --order=random"
|
|
|
|
if i % 2 == 1:
|
|
options += " --db-engine=Ordinary"
|
|
|
|
# If database name is not specified, new database is created for each functional test.
|
|
# Run some threads with one database for all tests.
|
|
if i % 3 == 1:
|
|
options += " --database=test_{}".format(i)
|
|
|
|
if i == 13:
|
|
options += " --client-option='memory_tracker_fault_probability=0.00001'"
|
|
|
|
return options
|
|
|
|
|
|
def run_func_test(cmd, output_prefix, num_processes, skip_tests_option, global_time_limit):
|
|
skip_list_opt = get_skip_list_cmd(cmd)
|
|
|
|
global_time_limit_option = ''
|
|
if global_time_limit:
|
|
global_time_limit_option = "--global_time_limit={}".format(global_time_limit)
|
|
|
|
output_paths = [os.path.join(output_prefix, "stress_test_run_{}.txt".format(i)) for i in range(num_processes)]
|
|
pipes = []
|
|
for i in range(0, len(output_paths)):
|
|
f = open(output_paths[i], 'w')
|
|
full_command = "{} {} {} {} {}".format(cmd, skip_list_opt, get_options(i), global_time_limit_option, skip_tests_option)
|
|
logging.info("Run func tests '%s'", full_command)
|
|
p = Popen(full_command, shell=True, stdout=f, stderr=f)
|
|
pipes.append(p)
|
|
time.sleep(0.5)
|
|
return pipes
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s %(message)s')
|
|
parser = argparse.ArgumentParser(description="ClickHouse script for running stresstest")
|
|
parser.add_argument("--test-cmd", default='/usr/bin/clickhouse-test')
|
|
parser.add_argument("--skip-func-tests", default='')
|
|
parser.add_argument("--client-cmd", default='clickhouse-client')
|
|
parser.add_argument("--server-log-folder", default='/var/log/clickhouse-server')
|
|
parser.add_argument("--output-folder")
|
|
parser.add_argument("--global-time-limit", type=int, default=3600)
|
|
parser.add_argument("--num-parallel", default=cpu_count())
|
|
parser.add_argument('--hung-check', action='store_true', default=False)
|
|
|
|
args = parser.parse_args()
|
|
func_pipes = []
|
|
func_pipes = run_func_test(args.test_cmd, args.output_folder, args.num_parallel, args.skip_func_tests, args.global_time_limit)
|
|
|
|
logging.info("Will wait functests to finish")
|
|
while True:
|
|
retcodes = []
|
|
for p in func_pipes:
|
|
if p.poll() is not None:
|
|
retcodes.append(p.returncode)
|
|
if len(retcodes) == len(func_pipes):
|
|
break
|
|
logging.info("Finished %s from %s processes", len(retcodes), len(func_pipes))
|
|
time.sleep(5)
|
|
|
|
logging.info("All processes finished")
|
|
if args.hung_check:
|
|
logging.info("Checking if some queries hung")
|
|
cmd = "{} {} {}".format(args.test_cmd, "--hung-check", "00001_select_1")
|
|
res = call(cmd, shell=True, stderr=STDOUT)
|
|
if res != 0:
|
|
logging.info("Hung check failed with exit code {}".format(res))
|
|
sys.exit(1)
|
|
|
|
logging.info("Stress test finished")
|