Message ID | 20200228071914.11746-3-vsementsov@virtuozzo.com (mailing list archive) |
---|---|
State | New, archived |
Headers | show |
Series | benchmark util | expand |
On Fri, Feb 28, 2020 at 8:19 AM Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> wrote: > > Add block-job benchmarking helper functions. > > Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> > --- Reviewed-by: Aleksandar Markovic <amarkovic@wavecomp.com> > scripts/simplebench/bench_block_job.py | 119 +++++++++++++++++++++++++ > 1 file changed, 119 insertions(+) > create mode 100755 scripts/simplebench/bench_block_job.py > > diff --git a/scripts/simplebench/bench_block_job.py b/scripts/simplebench/bench_block_job.py > new file mode 100755 > index 0000000000..9808d696cf > --- /dev/null > +++ b/scripts/simplebench/bench_block_job.py > @@ -0,0 +1,119 @@ > +#!/usr/bin/env python > +# > +# Benchmark block jobs > +# > +# Copyright (c) 2019 Virtuozzo International GmbH. > +# > +# This program is free software; you can redistribute it and/or modify > +# it under the terms of the GNU General Public License as published by > +# the Free Software Foundation; either version 2 of the License, or > +# (at your option) any later version. > +# > +# This program is distributed in the hope that it will be useful, > +# but WITHOUT ANY WARRANTY; without even the implied warranty of > +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the > +# GNU General Public License for more details. > +# > +# You should have received a copy of the GNU General Public License > +# along with this program. If not, see <http://www.gnu.org/licenses/>. > +# > + > + > +import sys > +import os > +import socket > +import json > + > +sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', 'python')) > +from qemu.machine import QEMUMachine > +from qemu.qmp import QMPConnectError > + > + > +def bench_block_job(cmd, cmd_args, qemu_args): > + """Benchmark block-job > + > + cmd -- qmp command to run block-job (like blockdev-backup) > + cmd_args -- dict of qmp command arguments > + qemu_args -- list of Qemu command line arguments, including path to Qemu > + binary > + > + Returns {'seconds': int} on success and {'error': str} on failure, dict may > + contain addional 'vm-log' field. Return value is compatible with > + simplebench lib. > + """ > + > + vm = QEMUMachine(qemu_args[0], args=qemu_args[1:]) > + > + try: > + vm.launch() > + except OSError as e: > + return {'error': 'popen failed: ' + str(e)} > + except (QMPConnectError, socket.timeout): > + return {'error': 'qemu failed: ' + str(vm.get_log())} > + > + try: > + res = vm.qmp(cmd, **cmd_args) > + if res != {'return': {}}: > + vm.shutdown() > + return {'error': '"{}" command failed: {}'.format(cmd, str(res))} > + > + e = vm.event_wait('JOB_STATUS_CHANGE') > + assert e['data']['status'] == 'created' > + start_ms = e['timestamp']['seconds'] * 1000000 + \ > + e['timestamp']['microseconds'] > + > + e = vm.events_wait((('BLOCK_JOB_READY', None), > + ('BLOCK_JOB_COMPLETED', None), > + ('BLOCK_JOB_FAILED', None)), timeout=True) > + if e['event'] not in ('BLOCK_JOB_READY', 'BLOCK_JOB_COMPLETED'): > + vm.shutdown() > + return {'error': 'block-job failed: ' + str(e), > + 'vm-log': vm.get_log()} > + end_ms = e['timestamp']['seconds'] * 1000000 + \ > + e['timestamp']['microseconds'] > + finally: > + vm.shutdown() > + > + return {'seconds': (end_ms - start_ms) / 1000000.0} > + > + > +# Bench backup or mirror > +def bench_block_copy(qemu_binary, cmd, source, target): > + """Helper to run bench_block_job() for mirror or backup""" > + assert cmd in ('blockdev-backup', 'blockdev-mirror') > + > + source['node-name'] = 'source' > + target['node-name'] = 'target' > + > + return bench_block_job(cmd, > + {'job-id': 'job0', 'device': 'source', > + 'target': 'target', 'sync': 'full'}, > + [qemu_binary, > + '-blockdev', json.dumps(source), > + '-blockdev', json.dumps(target)]) > + > + > +def drv_file(filename): > + return {'driver': 'file', 'filename': filename, > + 'cache': {'direct': True}, 'aio': 'native'} > + > + > +def drv_nbd(host, port): > + return {'driver': 'nbd', > + 'server': {'type': 'inet', 'host': host, 'port': port}} > + > + > +if __name__ == '__main__': > + import sys > + > + if len(sys.argv) < 4: > + print('USAGE: {} <qmp block-job command name> ' > + '<json string of arguments for the command> ' > + '<qemu binary path and arguments>'.format(sys.argv[0])) > + exit(1) > + > + res = bench_block_job(sys.argv[1], json.loads(sys.argv[2]), sys.argv[3:]) > + if 'seconds' in res: > + print('{:.2f}'.format(res['seconds'])) > + else: > + print(res) > -- > 2.21.0 > >
diff --git a/scripts/simplebench/bench_block_job.py b/scripts/simplebench/bench_block_job.py new file mode 100755 index 0000000000..9808d696cf --- /dev/null +++ b/scripts/simplebench/bench_block_job.py @@ -0,0 +1,119 @@ +#!/usr/bin/env python +# +# Benchmark block jobs +# +# Copyright (c) 2019 Virtuozzo International GmbH. +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. +# + + +import sys +import os +import socket +import json + +sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', 'python')) +from qemu.machine import QEMUMachine +from qemu.qmp import QMPConnectError + + +def bench_block_job(cmd, cmd_args, qemu_args): + """Benchmark block-job + + cmd -- qmp command to run block-job (like blockdev-backup) + cmd_args -- dict of qmp command arguments + qemu_args -- list of Qemu command line arguments, including path to Qemu + binary + + Returns {'seconds': int} on success and {'error': str} on failure, dict may + contain addional 'vm-log' field. Return value is compatible with + simplebench lib. + """ + + vm = QEMUMachine(qemu_args[0], args=qemu_args[1:]) + + try: + vm.launch() + except OSError as e: + return {'error': 'popen failed: ' + str(e)} + except (QMPConnectError, socket.timeout): + return {'error': 'qemu failed: ' + str(vm.get_log())} + + try: + res = vm.qmp(cmd, **cmd_args) + if res != {'return': {}}: + vm.shutdown() + return {'error': '"{}" command failed: {}'.format(cmd, str(res))} + + e = vm.event_wait('JOB_STATUS_CHANGE') + assert e['data']['status'] == 'created' + start_ms = e['timestamp']['seconds'] * 1000000 + \ + e['timestamp']['microseconds'] + + e = vm.events_wait((('BLOCK_JOB_READY', None), + ('BLOCK_JOB_COMPLETED', None), + ('BLOCK_JOB_FAILED', None)), timeout=True) + if e['event'] not in ('BLOCK_JOB_READY', 'BLOCK_JOB_COMPLETED'): + vm.shutdown() + return {'error': 'block-job failed: ' + str(e), + 'vm-log': vm.get_log()} + end_ms = e['timestamp']['seconds'] * 1000000 + \ + e['timestamp']['microseconds'] + finally: + vm.shutdown() + + return {'seconds': (end_ms - start_ms) / 1000000.0} + + +# Bench backup or mirror +def bench_block_copy(qemu_binary, cmd, source, target): + """Helper to run bench_block_job() for mirror or backup""" + assert cmd in ('blockdev-backup', 'blockdev-mirror') + + source['node-name'] = 'source' + target['node-name'] = 'target' + + return bench_block_job(cmd, + {'job-id': 'job0', 'device': 'source', + 'target': 'target', 'sync': 'full'}, + [qemu_binary, + '-blockdev', json.dumps(source), + '-blockdev', json.dumps(target)]) + + +def drv_file(filename): + return {'driver': 'file', 'filename': filename, + 'cache': {'direct': True}, 'aio': 'native'} + + +def drv_nbd(host, port): + return {'driver': 'nbd', + 'server': {'type': 'inet', 'host': host, 'port': port}} + + +if __name__ == '__main__': + import sys + + if len(sys.argv) < 4: + print('USAGE: {} <qmp block-job command name> ' + '<json string of arguments for the command> ' + '<qemu binary path and arguments>'.format(sys.argv[0])) + exit(1) + + res = bench_block_job(sys.argv[1], json.loads(sys.argv[2]), sys.argv[3:]) + if 'seconds' in res: + print('{:.2f}'.format(res['seconds'])) + else: + print(res)
Add block-job benchmarking helper functions. Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> --- scripts/simplebench/bench_block_job.py | 119 +++++++++++++++++++++++++ 1 file changed, 119 insertions(+) create mode 100755 scripts/simplebench/bench_block_job.py