1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
##############################################################################
# Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
from __future__ import absolute_import
from __future__ import print_function
import os
import sys
import time
from six.moves import range
from yardstick.common import constants as consts
from yardstick.common.httpClient import HttpClient
class EnvCommand(object):
"""
Set of commands to prepare environment
"""
def do_influxdb(self, args):
data = {'action': 'createInfluxDBContainer'}
task_id = self._start_async_task(data)
start = '* creating influxDB'
self._check_status(task_id, start)
def do_grafana(self, args):
data = {'action': 'createGrafanaContainer'}
task_id = self._start_async_task(data)
start = '* creating grafana'
self._check_status(task_id, start)
def do_prepare(self, args):
data = {'action': 'prepareYardstickEnv'}
task_id = self._start_async_task(data)
start = '* preparing yardstick environment'
self._check_status(task_id, start)
def _start_async_task(self, data):
url = consts.ENV_ACTION_API
return HttpClient().post(url, data)['result']['task_id']
def _check_status(self, task_id, start):
self._print_status(start, '[]\r')
url = '{}?task_id={}'.format(consts.ASYNC_TASK_API, task_id)
CHECK_STATUS_RETRY = 20
CHECK_STATUS_DELAY = 5
for retry in range(CHECK_STATUS_RETRY):
response = HttpClient().get(url)
status = response['status']
if status:
break
# wait until the async task finished
time.sleep(CHECK_STATUS_DELAY * (retry + 1))
switcher = {
0: 'Timeout',
1: 'Finished',
2: 'Error'
}
self._print_status(start, '[{}]'.format(switcher[status]))
if status == 2:
print(response['result'])
sys.stdout.flush()
return status
def _print_status(self, s, e):
try:
columns = int(os.popen('stty size', 'r').read().split()[1])
word = '{}{}{}'.format(s, ' ' * (columns - len(s) - len(e)), e)
sys.stdout.write(word)
sys.stdout.flush()
except IndexError:
pass
|