load-test/main.py

import argparse
from multiprocessing import Pool
from multiprocessing import cpu_count

import json 
import signal




parser = argparse.ArgumentParser()
parser.add_argument('-config',help='')
args = parser.parse_args()

with open(f'/usr/local/sdt/app/{args.config}/config.json', encoding='UTF-8') as f:
    jsonData = json.load(f)

stop_loop = 0 
 
def exit_chld(x, y):
    global stop_loop
    stop_loop = 1
 
 
def f(x):
    global stop_loop
    while not stop_loop:
        x*x
 
 
signal.signal(signal.SIGINT, exit_chld)
 
if __name__ == '__main__':
    processes = jsonData["cpu"]
    print('-' * 20)
    print('Running load on CPU(s)')
    print('Utilizing %d cores' % processes)
    print('-' * 20)
    pool = Pool(processes)
    pool.map(f, range(processes))