41 lines
768 B
Python
41 lines
768 B
Python
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))
|