blob: a57faffb7e3bb2d4017e01aa6979c13a0aac1f4c [file] [log] [blame]
koder aka kdanilovbc2c8982015-06-13 02:50:43 +03001import re
2import time
3import json
koder aka kdanilovf236b9c2015-06-24 18:17:22 +03004import stat
koder aka kdanilov6ab4d432015-06-22 00:26:28 +03005import random
koder aka kdanilovbc2c8982015-06-13 02:50:43 +03006import os.path
7import logging
8import datetime
9import functools
10import subprocess
11import collections
12
13import yaml
14import paramiko
15import texttable
16from paramiko.ssh_exception import SSHException
17from concurrent.futures import ThreadPoolExecutor
18
koder aka kdanilov6ab4d432015-06-22 00:26:28 +030019import wally
koder aka kdanilovbc2c8982015-06-13 02:50:43 +030020from wally.pretty_yaml import dumps
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +030021from wally.statistic import round_3_digit, data_property, average
koder aka kdanilovbc2c8982015-06-13 02:50:43 +030022from wally.utils import ssize2b, sec_to_str, StopTestError, Barrier, get_os
23from wally.ssh_utils import (save_to_remote, read_from_remote, BGSSHTask, reconnect)
24
25from .fio_task_parser import (execution_time, fio_cfg_compile,
koder aka kdanilovf236b9c2015-06-24 18:17:22 +030026 get_test_summary, get_test_summary_tuple,
27 get_test_sync_mode, FioJobSection)
28
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +030029from ..itest import (TimeSeriesValue, PerfTest, TestResults,
30 run_on_node, TestConfig, MeasurementMatrix)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +030031
32logger = logging.getLogger("wally")
33
34
35# Results folder structure
36# results/
37# {loadtype}_{num}/
38# config.yaml
39# ......
40
41
42class NoData(object):
43 pass
44
45
46def cached_prop(func):
47 @property
48 @functools.wraps(func)
49 def closure(self):
50 val = getattr(self, "_" + func.__name__)
51 if val is NoData:
52 val = func(self)
53 setattr(self, "_" + func.__name__, val)
54 return val
55 return closure
56
57
58def load_fio_log_file(fname):
59 with open(fname) as fd:
60 it = [ln.split(',')[:2] for ln in fd]
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +030061
62 vals = [(float(off) / 1000, # convert us to ms
63 float(val.strip()) + 0.5) # add 0.5 to compemsate average value
64 # as fio trimm all values in log to integer
65 for off, val in it]
66
koder aka kdanilovbc2c8982015-06-13 02:50:43 +030067 return TimeSeriesValue(vals)
68
69
70def load_test_results(cls, folder, run_num):
71 res = {}
72 params = None
73
74 fn = os.path.join(folder, str(run_num) + '_params.yaml')
75 params = yaml.load(open(fn).read())
76
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +030077 conn_ids_set = set()
78 rr = r"{0}_(?P<conn_id>.*?)_(?P<type>[^_.]*)\.\d+\.log$".format(run_num)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +030079 for fname in os.listdir(folder):
koder aka kdanilovbc2c8982015-06-13 02:50:43 +030080 rm = re.match(rr, fname)
81 if rm is None:
82 continue
83
84 conn_id_s = rm.group('conn_id')
85 conn_id = conn_id_s.replace('_', ':')
86 ftype = rm.group('type')
87
88 if ftype not in ('iops', 'bw', 'lat'):
89 continue
90
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +030091 ts = load_fio_log_file(os.path.join(folder, fname))
92 res.setdefault(ftype, {}).setdefault(conn_id, []).append(ts)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +030093
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +030094 conn_ids_set.add(conn_id)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +030095
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +030096 mm_res = {}
97
98 for key, data in res.items():
99 conn_ids = sorted(conn_ids_set)
100 matr = [data[conn_id] for conn_id in conn_ids]
101
102 mm_res[key] = MeasurementMatrix(matr, conn_ids)
103
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300104 # iops_from_lat_matr = []
105 # for node_ts in mm_res['lat'].data:
106 # iops_from_lat_matr.append([])
107 # for thread_ts in node_ts:
108 # ndt = [(start + ln, 1000000. / val)
109 # for (start, ln, val) in thread_ts.data]
110 # new_ts = TimeSeriesValue(ndt)
111 # iops_from_lat_matr[-1].append(new_ts)
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300112
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300113 # mm_res['iops_from_lat'] = MeasurementMatrix(iops_from_lat_matr, conn_ids)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300114
115 raw_res = {}
116 for conn_id in conn_ids:
117 fn = os.path.join(folder, "{0}_{1}_rawres.json".format(run_num, conn_id_s))
118
119 # remove message hack
120 fc = "{" + open(fn).read().split('{', 1)[1]
121 raw_res[conn_id] = json.loads(fc)
122
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300123 fio_task = FioJobSection(params['name'])
124 fio_task.vals.update(params['vals'])
125
126 config = TestConfig('io', params, None, params['nodes'], folder, None)
127 return cls(config, fio_task, mm_res, raw_res, params['intervals'])
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300128
129
130class Attrmapper(object):
131 def __init__(self, dct):
132 self.__dct = dct
133
134 def __getattr__(self, name):
135 try:
136 return self.__dct[name]
137 except KeyError:
138 raise AttributeError(name)
139
140
141class DiskPerfInfo(object):
142 def __init__(self, name, summary, params, testnodes_count):
143 self.name = name
144 self.bw = None
145 self.iops = None
146 self.lat = None
147 self.lat_50 = None
148 self.lat_95 = None
149
150 self.raw_bw = []
151 self.raw_iops = []
152 self.raw_lat = []
153
154 self.params = params
155 self.testnodes_count = testnodes_count
156 self.summary = summary
157 self.p = Attrmapper(self.params['vals'])
158
159 self.sync_mode = get_test_sync_mode(self.params['vals'])
160 self.concurence = self.params['vals'].get('numjobs', 1)
161
162
163def get_lat_perc_50_95(lat_mks):
164 curr_perc = 0
165 perc_50 = None
166 perc_95 = None
167 pkey = None
168 for key, val in sorted(lat_mks.items()):
169 if curr_perc + val >= 50 and perc_50 is None:
170 if pkey is None or val < 1.:
171 perc_50 = key
172 else:
173 perc_50 = (50. - curr_perc) / val * (key - pkey) + pkey
174
175 if curr_perc + val >= 95:
176 if pkey is None or val < 1.:
177 perc_95 = key
178 else:
179 perc_95 = (95. - curr_perc) / val * (key - pkey) + pkey
180 break
181
182 pkey = key
183 curr_perc += val
184
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300185 # for k, v in sorted(lat_mks.items()):
186 # if k / 1000 > 0:
187 # print "{0:>4}".format(k / 1000), v
188
189 # print perc_50 / 1000., perc_95 / 1000.
190 # exit(1)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300191 return perc_50 / 1000., perc_95 / 1000.
192
193
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300194class IOTestResult(TestResults):
195 """
196 Fio run results
197 config: TestConfig
198 fio_task: FioJobSection
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300199 ts_results: {str: MeasurementMatrix[TimeSeriesValue]}
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300200 raw_result: ????
201 run_interval:(float, float) - test tun time, used for sensors
202 """
203 def __init__(self, config, fio_task, ts_results, raw_result, run_interval):
204
205 self.name = fio_task.name.split("_")[0]
206 self.fio_task = fio_task
207
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300208 self.bw = ts_results.get('bw')
209 self.lat = ts_results.get('lat')
210 self.iops = ts_results.get('iops')
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300211 # self.iops_from_lat = ts_results.get('iops_from_lat')
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300212
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300213 # self.slat = drop_warmup(res.get('clat', None), self.params)
214 # self.clat = drop_warmup(res.get('slat', None), self.params)
215
216 res = {"bw": self.bw, "lat": self.lat, "iops": self.iops}
217
218 self.sensors_data = None
219 self._pinfo = None
220 TestResults.__init__(self, config, res, raw_result, run_interval)
221
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300222 def get_params_from_fio_report(self):
223 nodes = self.bw.connections_ids
224
225 iops = [self.raw_result[node]['jobs'][0]['mixed']['iops'] for node in nodes]
226 total_ios = [self.raw_result[node]['jobs'][0]['mixed']['total_ios'] for node in nodes]
227 runtime = [self.raw_result[node]['jobs'][0]['mixed']['runtime'] / 1000 for node in nodes]
228 flt_iops = [float(ios) / rtime for ios, rtime in zip(total_ios, runtime)]
229
230 bw = [self.raw_result[node]['jobs'][0]['mixed']['bw'] for node in nodes]
231 total_bytes = [self.raw_result[node]['jobs'][0]['mixed']['io_bytes'] for node in nodes]
232 flt_bw = [float(tbytes) / rtime for tbytes, rtime in zip(total_bytes, runtime)]
233
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300234 return {'iops': iops,
235 'flt_iops': flt_iops,
236 'bw': bw,
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300237 'flt_bw': flt_bw}
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300238
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300239 def summary(self):
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300240 return get_test_summary(self.fio_task, len(self.config.nodes))
241
242 def summary_tpl(self):
243 return get_test_summary_tuple(self.fio_task, len(self.config.nodes))
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300244
245 def get_yamable(self):
246 return self.summary()
247
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300248 def get_lat_perc_50_95_multy(self):
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300249 lat_mks = collections.defaultdict(lambda: 0)
250 num_res = 0
251
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300252 for result in self.raw_result.values():
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300253 num_res += len(result['jobs'])
254 for job_info in result['jobs']:
255 for k, v in job_info['latency_ms'].items():
256 if isinstance(k, basestring) and k.startswith('>='):
257 lat_mks[int(k[2:]) * 1000] += v
258 else:
259 lat_mks[int(k) * 1000] += v
260
261 for k, v in job_info['latency_us'].items():
262 lat_mks[int(k)] += v
263
264 for k, v in lat_mks.items():
265 lat_mks[k] = float(v) / num_res
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300266 return get_lat_perc_50_95(lat_mks)
267
268 def disk_perf_info(self, avg_interval=2.0):
269
270 if self._pinfo is not None:
271 return self._pinfo
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300272
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300273 testnodes_count = len(self.config.nodes)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300274
275 pinfo = DiskPerfInfo(self.name,
276 self.summary(),
277 self.params,
278 testnodes_count)
279
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300280 # ramp_time = self.fio_task.vals.get('ramp_time', 0)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300281
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300282 def prepare(data, drop=1):
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300283 if data is None:
284 return data
285
286 res = []
287 for ts_data in data:
288 # if ramp_time > 0:
289 # ts_data = ts_data.skip(ramp_time)
290
291 if ts_data.average_interval() < avg_interval:
292 ts_data = ts_data.derived(avg_interval)
293
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300294 # drop last value on bounds
295 # as they may contains ranges without activities
296 assert len(ts_data.values) >= drop + 1
297
298 if drop > 0:
299 res.append(ts_data.values[:-drop])
300 else:
301 res.append(ts_data.values)
302
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300303 return res
304
305 def agg_data(matr):
306 arr = sum(matr, [])
307 min_len = min(map(len, arr))
308 res = []
309 for idx in range(min_len):
310 res.append(sum(dt[idx] for dt in arr))
311 return res
312
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300313 # pinfo.raw_lat = map(prepare, self.lat.per_vm())
314 # num_th = sum(map(len, pinfo.raw_lat))
315 # avg_lat = [val / num_th for val in agg_data(pinfo.raw_lat)]
316 # pinfo.lat = data_property(avg_lat)
317 pinfo.lat_50, pinfo.lat_95 = self.get_lat_perc_50_95_multy()
318 pinfo.lat = pinfo.lat_50
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300319
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300320 pinfo.raw_bw = map(prepare, self.bw.per_vm())
321 pinfo.raw_iops = map(prepare, self.iops.per_vm())
322
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300323 fparams = self.get_params_from_fio_report()
324 fio_report_bw = sum(fparams['flt_bw'])
325 fio_report_iops = sum(fparams['flt_iops'])
326
327 agg_bw = agg_data(pinfo.raw_bw)
328 agg_iops = agg_data(pinfo.raw_iops)
329
330 log_bw_avg = average(agg_bw)
331 log_iops_avg = average(agg_iops)
332
333 # update values to match average from fio report
334 coef_iops = fio_report_iops / float(log_iops_avg)
335 coef_bw = fio_report_bw / float(log_bw_avg)
336
337 bw_log = data_property([val * coef_bw for val in agg_bw])
338 iops_log = data_property([val * coef_iops for val in agg_iops])
339
340 bw_report = data_property([fio_report_bw])
341 iops_report = data_property([fio_report_iops])
342
343 # When IOPS/BW per thread is too low
344 # data from logs is rounded to match
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300345 iops_per_th = sum(sum(pinfo.raw_iops, []), [])
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300346 if average(iops_per_th) > 10:
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300347 pinfo.iops = iops_log
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300348 pinfo.iops2 = iops_report
349 else:
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300350 pinfo.iops = iops_report
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300351 pinfo.iops2 = iops_log
352
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300353 bw_per_th = sum(sum(pinfo.raw_bw, []), [])
354 if average(bw_per_th) > 10:
355 pinfo.bw = bw_log
356 else:
357 pinfo.bw2 = bw_report
358
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300359 self._pinfo = pinfo
360
361 return pinfo
362
363
364class IOPerfTest(PerfTest):
365 tcp_conn_timeout = 30
366 max_pig_timeout = 5
367 soft_runcycle = 5 * 60
368
369 def __init__(self, config):
370 PerfTest.__init__(self, config)
371
372 get = self.config.params.get
373 do_get = self.config.params.__getitem__
374
375 self.config_fname = do_get('cfg')
376
377 if '/' not in self.config_fname and '.' not in self.config_fname:
378 cfgs_dir = os.path.dirname(__file__)
379 self.config_fname = os.path.join(cfgs_dir,
380 self.config_fname + '.cfg')
381
382 self.alive_check_interval = get('alive_check_interval')
383 self.use_system_fio = get('use_system_fio', False)
384
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300385 if get('prefill_files') is not None:
386 logger.warning("prefill_files option is depricated. Use force_prefill instead")
387
388 self.force_prefill = get('force_prefill', False)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300389 self.config_params = get('params', {}).copy()
390
391 self.io_py_remote = self.join_remote("agent.py")
392 self.results_file = self.join_remote("results.json")
393 self.pid_file = self.join_remote("pid")
394 self.task_file = self.join_remote("task.cfg")
395 self.sh_file = self.join_remote("cmd.sh")
396 self.err_out_file = self.join_remote("fio_err_out")
397 self.exit_code_file = self.join_remote("exit_code")
398
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300399 self.max_latency = get("max_lat", None)
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300400 self.min_bw_per_thread = get("min_bw", None)
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300401
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300402 self.use_sudo = get("use_sudo", True)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300403
404 self.raw_cfg = open(self.config_fname).read()
405 self.fio_configs = fio_cfg_compile(self.raw_cfg,
406 self.config_fname,
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300407 self.config_params)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300408 self.fio_configs = list(self.fio_configs)
409
410 @classmethod
411 def load(cls, folder):
412 for fname in os.listdir(folder):
413 if re.match("\d+_params.yaml$", fname):
414 num = int(fname.split('_')[0])
415 yield load_test_results(IOTestResult, folder, num)
416
417 def cleanup(self):
418 # delete_file(conn, self.io_py_remote)
419 # Need to remove tempo files, used for testing
420 pass
421
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300422 # size is megabytes
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300423 def check_prefill_required(self, rossh, fname, size, num_blocks=16):
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300424 with rossh.connection.open_sftp() as sftp:
425 fstats = sftp.stat(fname)
426
427 if stat.S_ISREG(fstats) and fstats.st_size < size * 1024 ** 2:
koder aka kdanilovf95cfc12015-06-23 03:33:19 +0300428 return True
429
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300430 cmd = 'python -c "' + \
431 "import sys;" + \
432 "fd = open('{0}', 'rb');" + \
433 "fd.seek({1});" + \
434 "data = fd.read(1024); " + \
435 "sys.stdout.write(data + ' ' * ( 1024 - len(data)))\" | md5sum"
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300436
437 if self.use_sudo:
438 cmd = "sudo " + cmd
439
koder aka kdanilovf95cfc12015-06-23 03:33:19 +0300440 zero_md5 = '0f343b0931126a20f133d67c2b018a3b'
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300441 offsets = [random.randrange(size - 1024) for _ in range(num_blocks)]
442 offsets.append(size - 1024)
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300443
koder aka kdanilovf95cfc12015-06-23 03:33:19 +0300444 for offset in offsets:
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300445 data = rossh(cmd.format(fname, offset), nolog=True)
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300446
447 md = ""
448 for line in data.split("\n"):
449 if "unable to resolve" not in line:
450 md = line.split()[0].strip()
451 break
koder aka kdanilovf95cfc12015-06-23 03:33:19 +0300452
453 if len(md) != 32:
454 logger.error("File data check is failed - " + data)
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300455 return True
koder aka kdanilovf95cfc12015-06-23 03:33:19 +0300456
457 if zero_md5 == md:
458 return True
459
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300460 return False
461
462 def prefill_test_files(self, rossh, files, force=False):
463 if self.use_system_fio:
464 cmd_templ = "fio "
465 else:
466 cmd_templ = "{0}/fio ".format(self.config.remote_dir)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300467
468 if self.use_sudo:
469 cmd_templ = "sudo " + cmd_templ
470
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300471 cmd_templ += "--name=xxx --filename={0} --direct=1" + \
472 " --bs=4m --size={1}m --rw=write"
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300473
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300474 ssize = 0
475
476 if force:
477 logger.info("File prefilling is forced")
478
479 ddtime = 0
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300480 for fname, curr_sz in files.items():
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300481 if not force:
482 if not self.check_prefill_required(rossh, fname, curr_sz):
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300483 logger.debug("prefill is skipped")
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300484 continue
485
486 logger.info("Prefilling file {0}".format(fname))
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300487 cmd = cmd_templ.format(fname, curr_sz)
488 ssize += curr_sz
489
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300490 stime = time.time()
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300491 rossh(cmd, timeout=curr_sz)
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300492 ddtime += time.time() - stime
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300493
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300494 if ddtime > 1.0:
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300495 fill_bw = int(ssize / ddtime)
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300496 mess = "Initiall fio fill bw is {0} MiBps for this vm"
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300497 logger.info(mess.format(fill_bw))
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300498
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300499 def install_utils(self, node, rossh, max_retry=3, timeout=5):
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300500 need_install = []
501 packs = [('screen', 'screen')]
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300502 os_info = get_os(rossh)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300503
504 if self.use_system_fio:
505 packs.append(('fio', 'fio'))
506 else:
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300507 packs.append(('bzip2', 'bzip2'))
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300508
509 for bin_name, package in packs:
510 if bin_name is None:
511 need_install.append(package)
512 continue
513
514 try:
515 rossh('which ' + bin_name, nolog=True)
516 except OSError:
517 need_install.append(package)
518
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300519 if len(need_install) != 0:
520 if 'redhat' == os_info.distro:
521 cmd = "sudo yum -y install " + " ".join(need_install)
522 else:
523 cmd = "sudo apt-get -y install " + " ".join(need_install)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300524
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300525 for _ in range(max_retry):
526 try:
527 rossh(cmd)
528 break
529 except OSError as err:
530 time.sleep(timeout)
531 else:
532 raise OSError("Can't install - " + str(err))
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300533
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300534 if not self.use_system_fio:
535 fio_dir = os.path.dirname(os.path.dirname(wally.__file__))
536 fio_dir = os.path.join(os.getcwd(), fio_dir)
537 fio_dir = os.path.join(fio_dir, 'fio_binaries')
538 fname = 'fio_{0.release}_{0.arch}.bz2'.format(os_info)
539 fio_path = os.path.join(fio_dir, fname)
540
541 if not os.path.exists(fio_path):
542 raise RuntimeError("No prebuild fio available for {0}".format(os_info))
543
544 bz_dest = self.join_remote('fio.bz2')
545 with node.connection.open_sftp() as sftp:
546 sftp.put(fio_path, bz_dest)
547
548 rossh("bzip2 --decompress " + bz_dest, nolog=True)
549 rossh("chmod a+x " + self.join_remote("fio"), nolog=True)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300550
551 def pre_run(self):
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300552 files = {}
553 for section in self.fio_configs:
554 sz = ssize2b(section.vals['size'])
555 msz = sz / (1024 ** 2)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300556
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300557 if sz % (1024 ** 2) != 0:
558 msz += 1
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300559
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300560 fname = section.vals['filename']
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300561
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300562 # if already has other test with the same file name
563 # take largest size
564 files[fname] = max(files.get(fname, 0), msz)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300565
566 with ThreadPoolExecutor(len(self.config.nodes)) as pool:
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300567 fc = functools.partial(self.pre_run_th,
568 files=files,
569 force=self.force_prefill)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300570 list(pool.map(fc, self.config.nodes))
571
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300572 def pre_run_th(self, node, files, force):
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300573 # fill files with pseudo-random data
574 rossh = run_on_node(node)
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300575 rossh.connection = node.connection
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300576
577 try:
578 cmd = 'mkdir -p "{0}"'.format(self.config.remote_dir)
579 if self.use_sudo:
580 cmd = "sudo " + cmd
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300581 cmd += " ; sudo chown {0} {1}".format(node.get_user(),
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300582 self.config.remote_dir)
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300583 rossh(cmd, nolog=True)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300584
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300585 assert self.config.remote_dir != "" and self.config.remote_dir != "/"
586 rossh("rm -rf {0}/*".format(self.config.remote_dir), nolog=True)
587
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300588 except Exception as exc:
589 msg = "Failed to create folder {0} on remote {1}. Error: {2!s}"
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300590 msg = msg.format(self.config.remote_dir, node.get_conn_id(), exc)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300591 logger.exception(msg)
592 raise StopTestError(msg, exc)
593
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300594 self.install_utils(node, rossh)
595 self.prefill_test_files(rossh, files, force)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300596
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300597 def show_test_execution_time(self):
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300598 if len(self.fio_configs) > 1:
599 # +10% - is a rough estimation for additional operations
600 # like sftp, etc
601 exec_time = int(sum(map(execution_time, self.fio_configs)) * 1.1)
602 exec_time_s = sec_to_str(exec_time)
603 now_dt = datetime.datetime.now()
604 end_dt = now_dt + datetime.timedelta(0, exec_time)
605 msg = "Entire test should takes aroud: {0} and finished at {1}"
606 logger.info(msg.format(exec_time_s,
607 end_dt.strftime("%H:%M:%S")))
608
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300609 def run(self):
610 logger.debug("Run preparation")
611 self.pre_run()
612 self.show_test_execution_time()
613
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300614 tname = os.path.basename(self.config_fname)
615 if tname.endswith('.cfg'):
616 tname = tname[:-4]
617
618 barrier = Barrier(len(self.config.nodes))
619 results = []
620
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300621 # set of Operation_Mode_BlockSize str's
622 # which should not be tested anymore, as
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300623 # they already too slow with previous thread count
624 lat_bw_limit_reached = set()
625
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300626 with ThreadPoolExecutor(len(self.config.nodes)) as pool:
koder aka kdanilov6b872662015-06-23 01:58:36 +0300627 self.fio_configs.sort(key=lambda x: int(x.vals.get('numjobs', 1)))
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300628
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300629 for pos, fio_cfg in enumerate(self.fio_configs):
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300630 test_descr = get_test_summary(fio_cfg.vals).split("th")[0]
631 if test_descr in lat_bw_limit_reached:
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300632 continue
633 else:
634 logger.info("Will run {0} test".format(fio_cfg.name))
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300635
636 templ = "Test should takes about {0}." + \
637 " Should finish at {1}," + \
638 " will wait at most till {2}"
639 exec_time = execution_time(fio_cfg)
640 exec_time_str = sec_to_str(exec_time)
641 timeout = int(exec_time + max(300, exec_time))
642
643 now_dt = datetime.datetime.now()
644 end_dt = now_dt + datetime.timedelta(0, exec_time)
645 wait_till = now_dt + datetime.timedelta(0, timeout)
646
647 logger.info(templ.format(exec_time_str,
648 end_dt.strftime("%H:%M:%S"),
649 wait_till.strftime("%H:%M:%S")))
650
651 func = functools.partial(self.do_run,
652 barrier=barrier,
653 fio_cfg=fio_cfg,
654 pos=pos)
655
656 max_retr = 3
657 for idx in range(max_retr):
658 try:
659 intervals = list(pool.map(func, self.config.nodes))
660 break
661 except (EnvironmentError, SSHException) as exc:
662 logger.exception("During fio run")
663 if idx == max_retr - 1:
664 raise StopTestError("Fio failed", exc)
665
666 logger.info("Sleeping 30s and retrying")
667 time.sleep(30)
668
669 fname = "{0}_task.fio".format(pos)
670 with open(os.path.join(self.config.log_directory, fname), "w") as fd:
671 fd.write(str(fio_cfg))
672
673 params = {'vm_count': len(self.config.nodes)}
674 params['name'] = fio_cfg.name
675 params['vals'] = dict(fio_cfg.vals.items())
676 params['intervals'] = intervals
677 params['nodes'] = [node.get_conn_id() for node in self.config.nodes]
678
679 fname = "{0}_params.yaml".format(pos)
680 with open(os.path.join(self.config.log_directory, fname), "w") as fd:
681 fd.write(dumps(params))
682
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300683 res = load_test_results(IOTestResult, self.config.log_directory, pos)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300684 results.append(res)
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300685
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300686 if self.max_latency is not None:
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300687 lat_50, _ = res.get_lat_perc_50_95_multy()
688
689 # conver us to ms
690 if self.max_latency < lat_50:
691 logger.info(("Will skip all subsequent tests of {0} " +
692 "due to lat/bw limits").format(fio_cfg.name))
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300693 lat_bw_limit_reached.add(test_descr)
694
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300695 test_res = res.get_params_from_fio_report()
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300696 if self.min_bw_per_thread is not None:
697 if self.min_bw_per_thread > average(test_res['bw']):
698 lat_bw_limit_reached.add(test_descr)
699
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300700 return results
701
702 def do_run(self, node, barrier, fio_cfg, pos, nolog=False):
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300703 if self.use_sudo:
704 sudo = "sudo "
705 else:
706 sudo = ""
707
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300708 bash_file = "#!/bin/bash\n" + \
709 "cd {exec_folder}\n" + \
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300710 "{fio_path}fio --output-format=json --output={out_file} " + \
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300711 "--alloc-size=262144 {job_file} " + \
712 " >{err_out_file} 2>&1 \n" + \
713 "echo $? >{res_code_file}\n"
714
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300715 exec_folder = self.config.remote_dir
716
717 if self.use_system_fio:
718 fio_path = ""
719 else:
720 if not exec_folder.endswith("/"):
721 fio_path = exec_folder + "/"
722 else:
723 fio_path = exec_folder
724
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300725 bash_file = bash_file.format(out_file=self.results_file,
726 job_file=self.task_file,
727 err_out_file=self.err_out_file,
728 res_code_file=self.exit_code_file,
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300729 exec_folder=exec_folder,
730 fio_path=fio_path)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300731
732 with node.connection.open_sftp() as sftp:
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300733 save_to_remote(sftp, self.task_file, str(fio_cfg))
734 save_to_remote(sftp, self.sh_file, bash_file)
735
736 exec_time = execution_time(fio_cfg)
737
738 timeout = int(exec_time + max(300, exec_time))
739 soft_tout = exec_time
740
741 begin = time.time()
742
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300743 fnames_before = run_on_node(node)("ls -1 " + exec_folder, nolog=True)
744
745 barrier.wait()
746
koder aka kdanilov5414a992015-06-13 03:07:25 +0300747 task = BGSSHTask(node, self.use_sudo)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300748 task.start(sudo + "bash " + self.sh_file)
749
750 while True:
751 try:
752 task.wait(soft_tout, timeout)
753 break
754 except paramiko.SSHException:
755 pass
756
757 try:
758 node.connection.close()
759 except:
760 pass
761
762 reconnect(node.connection, node.conn_url)
763
764 end = time.time()
765 rossh = run_on_node(node)
766 fnames_after = rossh("ls -1 " + exec_folder, nolog=True)
767
768 conn_id = node.get_conn_id().replace(":", "_")
769 if not nolog:
770 logger.debug("Test on node {0} is finished".format(conn_id))
771
772 log_files_pref = []
773 if 'write_lat_log' in fio_cfg.vals:
774 fname = fio_cfg.vals['write_lat_log']
775 log_files_pref.append(fname + '_clat')
776 log_files_pref.append(fname + '_lat')
777 log_files_pref.append(fname + '_slat')
778
779 if 'write_iops_log' in fio_cfg.vals:
780 fname = fio_cfg.vals['write_iops_log']
781 log_files_pref.append(fname + '_iops')
782
783 if 'write_bw_log' in fio_cfg.vals:
784 fname = fio_cfg.vals['write_bw_log']
785 log_files_pref.append(fname + '_bw')
786
787 files = collections.defaultdict(lambda: [])
788 all_files = [os.path.basename(self.results_file)]
789 new_files = set(fnames_after.split()) - set(fnames_before.split())
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300790
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300791 for fname in new_files:
792 if fname.endswith('.log') and fname.split('.')[0] in log_files_pref:
793 name, _ = os.path.splitext(fname)
794 if fname.count('.') == 1:
795 tp = name.split("_")[-1]
796 cnt = 0
797 else:
798 tp_cnt = name.split("_")[-1]
799 tp, cnt = tp_cnt.split('.')
800 files[tp].append((int(cnt), fname))
801 all_files.append(fname)
802
803 arch_name = self.join_remote('wally_result.tar.gz')
804 tmp_dir = os.path.join(self.config.log_directory, 'tmp_' + conn_id)
805 os.mkdir(tmp_dir)
806 loc_arch_name = os.path.join(tmp_dir, 'wally_result.{0}.tar.gz'.format(conn_id))
807 file_full_names = " ".join(all_files)
808
809 try:
810 os.unlink(loc_arch_name)
811 except:
812 pass
813
814 with node.connection.open_sftp() as sftp:
815 exit_code = read_from_remote(sftp, self.exit_code_file)
816 err_out = read_from_remote(sftp, self.err_out_file)
817 exit_code = exit_code.strip()
818
819 if exit_code != '0':
820 msg = "fio exit with code {0}: {1}".format(exit_code, err_out)
821 logger.critical(msg.strip())
822 raise StopTestError("fio failed")
823
824 rossh("rm -f {0}".format(arch_name), nolog=True)
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300825 pack_files_cmd = "cd {0} ; tar zcvf {1} {2}".format(exec_folder, arch_name, file_full_names)
826 rossh(pack_files_cmd, nolog=True)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300827 sftp.get(arch_name, loc_arch_name)
828
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300829 unpack_files_cmd = "cd {0} ; tar xvzf {1} >/dev/null".format(tmp_dir, loc_arch_name)
830 subprocess.check_call(unpack_files_cmd, shell=True)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300831 os.unlink(loc_arch_name)
832
833 for ftype, fls in files.items():
834 for idx, fname in fls:
835 cname = os.path.join(tmp_dir, fname)
836 loc_fname = "{0}_{1}_{2}.{3}.log".format(pos, conn_id, ftype, idx)
837 loc_path = os.path.join(self.config.log_directory, loc_fname)
838 os.rename(cname, loc_path)
839
840 cname = os.path.join(tmp_dir,
841 os.path.basename(self.results_file))
842 loc_fname = "{0}_{1}_rawres.json".format(pos, conn_id)
843 loc_path = os.path.join(self.config.log_directory, loc_fname)
844 os.rename(cname, loc_path)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300845 os.rmdir(tmp_dir)
koder aka kdanilov6ab4d432015-06-22 00:26:28 +0300846
847 remove_remote_res_files_cmd = "cd {0} ; rm -f {1} {2}".format(exec_folder,
848 arch_name,
849 file_full_names)
850 rossh(remove_remote_res_files_cmd, nolog=True)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300851 return begin, end
852
853 @classmethod
koder aka kdanilov6b872662015-06-23 01:58:36 +0300854 def prepare_data(cls, results):
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300855 """
856 create a table with io performance report
857 for console
858 """
859
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300860 def key_func(data):
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300861 tpl = data.summary_tpl()
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300862 return (data.name.rsplit("_", 1)[0],
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300863 tpl.oper,
864 tpl.mode,
865 tpl.bsize,
866 int(tpl.th_count) * int(tpl.vm_count))
koder aka kdanilov6b872662015-06-23 01:58:36 +0300867 res = []
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300868
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300869 for item in sorted(results, key=key_func):
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300870 test_dinfo = item.disk_perf_info()
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300871
872 iops, _ = test_dinfo.iops.rounded_average_conf()
873
874 bw, bw_conf = test_dinfo.bw.rounded_average_conf()
875 _, bw_dev = test_dinfo.bw.rounded_average_dev()
876 conf_perc = int(round(bw_conf * 100 / bw))
877 dev_perc = int(round(bw_dev * 100 / bw))
878
koder aka kdanilov6b872662015-06-23 01:58:36 +0300879 lat_50 = round_3_digit(int(test_dinfo.lat_50))
880 lat_95 = round_3_digit(int(test_dinfo.lat_95))
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300881
koder aka kdanilovbb6d6cd2015-06-20 02:55:07 +0300882 testnodes_count = len(item.config.nodes)
883 iops_per_vm = round_3_digit(iops / testnodes_count)
884 bw_per_vm = round_3_digit(bw / testnodes_count)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300885
886 iops = round_3_digit(iops)
887 bw = round_3_digit(bw)
888
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300889 summ = "{0.oper}{0.mode} {0.bsize:>4} {0.th_count:>3}th {0.vm_count:>2}vm".format(item.summary_tpl())
890
891 res.append({"name": key_func(item)[0],
892 "key": key_func(item)[:4],
893 "summ": summ,
koder aka kdanilov6b872662015-06-23 01:58:36 +0300894 "iops": int(iops),
895 "bw": int(bw),
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300896 "conf": str(conf_perc),
897 "dev": str(dev_perc),
koder aka kdanilov6b872662015-06-23 01:58:36 +0300898 "iops_per_vm": int(iops_per_vm),
899 "bw_per_vm": int(bw_per_vm),
900 "lat_50": lat_50,
901 "lat_95": lat_95})
koder aka kdanilovbc2c8982015-06-13 02:50:43 +0300902
koder aka kdanilov6b872662015-06-23 01:58:36 +0300903 return res
904
905 Field = collections.namedtuple("Field", ("header", "attr", "allign", "size"))
906 fiels_and_header = [
907 Field("Name", "name", "l", 7),
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300908 Field("Description", "summ", "l", 19),
koder aka kdanilov6b872662015-06-23 01:58:36 +0300909 Field("IOPS\ncum", "iops", "r", 3),
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300910 Field("KiBps\ncum", "bw", "r", 6),
911 Field("Cnf %\n95%", "conf", "r", 3),
912 Field("Dev%", "dev", "r", 3),
913 Field("iops\n/vm", "iops_per_vm", "r", 3),
914 Field("KiBps\n/vm", "bw_per_vm", "r", 6),
koder aka kdanilov6b872662015-06-23 01:58:36 +0300915 Field("lat ms\nmedian", "lat_50", "r", 3),
916 Field("lat ms\n95%", "lat_95", "r", 3)
917 ]
918
919 fiels_and_header_dct = dict((item.attr, item) for item in fiels_and_header)
920
921 @classmethod
922 def format_for_console(cls, results):
923 """
924 create a table with io performance report
925 for console
926 """
927
928 tab = texttable.Texttable(max_width=120)
929 tab.set_deco(tab.HEADER | tab.VLINES | tab.BORDER)
930 tab.set_cols_align([f.allign for f in cls.fiels_and_header])
931 sep = ["-" * f.size for f in cls.fiels_and_header]
932 tab.header([f.header for f in cls.fiels_and_header])
933
934 prev_k = None
935 for item in cls.prepare_data(results):
koder aka kdanilov6b872662015-06-23 01:58:36 +0300936 if prev_k is not None:
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300937 if prev_k != item["key"]:
koder aka kdanilov6b872662015-06-23 01:58:36 +0300938 tab.add_row(sep)
939
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300940 prev_k = item["key"]
koder aka kdanilov6b872662015-06-23 01:58:36 +0300941 tab.add_row([item[f.attr] for f in cls.fiels_and_header])
942
943 return tab.draw()
944
945 @classmethod
946 def format_diff_for_console(cls, list_of_results):
947 """
948 create a table with io performance report
949 for console
950 """
951
952 tab = texttable.Texttable(max_width=200)
953 tab.set_deco(tab.HEADER | tab.VLINES | tab.BORDER)
954
955 header = [
956 cls.fiels_and_header_dct["name"].header,
957 cls.fiels_and_header_dct["summ"].header,
958 ]
959 allign = ["l", "l"]
960
961 header.append("IOPS ~ Cnf% ~ Dev%")
962 allign.extend(["r"] * len(list_of_results))
963 header.extend(
964 "IOPS_{0} %".format(i + 2) for i in range(len(list_of_results[1:]))
965 )
966
967 header.append("BW")
968 allign.extend(["r"] * len(list_of_results))
969 header.extend(
970 "BW_{0} %".format(i + 2) for i in range(len(list_of_results[1:]))
971 )
972
973 header.append("LAT")
974 allign.extend(["r"] * len(list_of_results))
975 header.extend(
976 "LAT_{0}".format(i + 2) for i in range(len(list_of_results[1:]))
977 )
978
979 tab.header(header)
980 sep = ["-" * 3] * len(header)
981 processed_results = map(cls.prepare_data, list_of_results)
982
983 key2results = []
984 for res in processed_results:
985 key2results.append(dict(
986 ((item["name"], item["summ"]), item) for item in res
987 ))
988
989 prev_k = None
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300990 iops_frmt = "{0[iops]} ~ {0[conf]:>2} ~ {0[dev]:>2}"
koder aka kdanilov6b872662015-06-23 01:58:36 +0300991 for item in processed_results[0]:
koder aka kdanilov6b872662015-06-23 01:58:36 +0300992 if prev_k is not None:
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300993 if prev_k != item["key"]:
koder aka kdanilov6b872662015-06-23 01:58:36 +0300994 tab.add_row(sep)
995
koder aka kdanilovf236b9c2015-06-24 18:17:22 +0300996 prev_k = item["key"]
koder aka kdanilov6b872662015-06-23 01:58:36 +0300997
998 key = (item['name'], item['summ'])
999 line = list(key)
1000 base = key2results[0][key]
1001
1002 line.append(iops_frmt.format(base))
1003
1004 for test_results in key2results[1:]:
1005 val = test_results.get(key)
1006 if val is None:
1007 line.append("-")
1008 elif base['iops'] == 0:
1009 line.append("Nan")
1010 else:
koder aka kdanilovf236b9c2015-06-24 18:17:22 +03001011 prc_val = {'dev': val['dev'], 'conf': val['conf']}
koder aka kdanilov6b872662015-06-23 01:58:36 +03001012 prc_val['iops'] = int(100 * val['iops'] / base['iops'])
1013 line.append(iops_frmt.format(prc_val))
1014
1015 line.append(base['bw'])
1016
1017 for test_results in key2results[1:]:
1018 val = test_results.get(key)
1019 if val is None:
1020 line.append("-")
1021 elif base['bw'] == 0:
1022 line.append("Nan")
1023 else:
1024 line.append(int(100 * val['bw'] / base['bw']))
1025
1026 for test_results in key2results:
1027 val = test_results.get(key)
1028 if val is None:
1029 line.append("-")
1030 else:
1031 line.append("{0[lat_50]} - {0[lat_95]}".format(val))
1032
1033 tab.add_row(line)
1034
1035 tab.set_cols_align(allign)
koder aka kdanilovbc2c8982015-06-13 02:50:43 +03001036 return tab.draw()