The script 'bench_write_req.py' allows comparing performances of write request for two qemu-img binary files.
Suggested-by: Denis V. Lunev <d...@openvz.org> Suggested-by: Vladimir Sementsov-Ogievskiy <vsement...@virtuozzo.com> Signed-off-by: Andrey Shinkevich <andrey.shinkev...@virtuozzo.com> --- scripts/simplebench/bench_write_req.py | 150 +++++++++++++++++++++++++++++++++ 1 file changed, 150 insertions(+) create mode 100755 scripts/simplebench/bench_write_req.py diff --git a/scripts/simplebench/bench_write_req.py b/scripts/simplebench/bench_write_req.py new file mode 100755 index 0000000..e127dcc --- /dev/null +++ b/scripts/simplebench/bench_write_req.py @@ -0,0 +1,150 @@ +#!/usr/bin/env python3 +# +# Test to compare performance of write requests for two qemu-img binary files. +# +# Copyright (c) 2020 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 subprocess +import simplebench + + +def bench_func(env, case): + """ Handle one "cell" of benchmarking table. """ + return bench_write_req(env['qemu_img'], env['image_name'], + case['block_size'], case['block_offset'], + case['requests']) + + +def qemu_img_pipe(*args): + '''Run qemu-img and return its output''' + subp = subprocess.Popen(list(args), + stdout=subprocess.PIPE, + stderr=subprocess.STDOUT, + universal_newlines=True) + exitcode = subp.wait() + if exitcode < 0: + sys.stderr.write('qemu-img received signal %i: %s\n' + % (-exitcode, ' '.join(list(args)))) + return subp.communicate()[0] + + +def bench_write_req(qemu_img, image_name, block_size, block_offset, requests): + """Benchmark write requests + + qemu_img -- path to qemu_img executable file + image_name -- QCOW2 image name to create + block_size -- size of a block to write to clusters + block_offset -- offset of the block in clusters + + Returns {'seconds': int} on success and {'error': str} on failure. + Return value is compatible with simplebench lib. + """ + + if not os.path.isfile(qemu_img): + print('File not found: {}'.format(qemu_img)) + sys.exit(1) + + image_dir = os.path.dirname(os.path.abspath(image_name)) + if not os.path.isdir(image_dir): + print('Path not found: {}'.format(image_name)) + sys.exit(1) + + cluster_size = 1024 * 1024 + image_size = 1024 * cluster_size + count = requests * int(image_size / cluster_size) + image_size = str(image_size) + + args_create = [qemu_img, 'create', '-f', 'qcow2', '-o', + 'cluster_size={}'.format(cluster_size), + image_name, image_size] + + cluster_size = str(cluster_size) + offset = str(block_offset) + size = str(block_size) + cnt = str(count) + args_bench = [qemu_img, 'bench', '-c', cnt, '-w', '-t', 'none', '-s', size, + '-S', cluster_size, '-o', offset, '-f', 'qcow2', image_name] + + try: + qemu_img_pipe(*args_create) + except OSError as e: + return {'error': 'qemu_img create failed: ' + str(e)} + + try: + ret = qemu_img_pipe(*args_bench) + except OSError as e: + return {'error': 'qemu_img bench failed: ' + str(e)} + finally: + os.remove(image_name) + if not ret: + return {'error': 'qemu_img bench failed'} + if 'seconds' in ret: + ret = ret.split() + index = ret.index('seconds.') + return {'seconds': float(ret[index-1])} + else: + return {'error': 'qemu_img bench failed: ' + ret} + + +if __name__ == '__main__': + + if len(sys.argv) < 4: + print('USAGE: {} <path to qemu-img binary file> ' + '<path to another qemu-img to compare performance with> ' + '<full or relative name for 10G QCOW2 image to create>' + ''.format(os.path.basename(sys.argv[0]))) + exit(1) + + # Test-cases are "rows" in benchmark resulting table, 'id' is a caption + # for the row, other fields are handled by bench_func. + test_cases = [ + { + 'id': '<cluster middle>', + 'block_size': 4096, + 'block_offset': 524288, + 'requests': 100 + }, + { + 'id': '<cluster overlap>', + 'block_size': 524288, + 'block_offset': 4096, + 'requests': 10 + }, + ] + + # Test-envs are "columns" in benchmark resulting table, 'id is a caption + # for the column, other fields are handled by bench_func. + # Set the paths below to desired values + test_envs = [ + { + 'id': '<qemu-img binary 1>', + 'qemu_img': '{}'.format(sys.argv[1]), + 'image_name': '{}'.format(sys.argv[3]) + }, + { + 'id': '<qemu-img binary 2>', + 'qemu_img': '{}'.format(sys.argv[2]), + 'image_name': '{}'.format(sys.argv[3]) + }, + ] + + result = simplebench.bench(bench_func, test_envs, test_cases, count=3, + initial_run=False) + print(simplebench.ascii(result)) -- 1.8.3.1