f3021755f9
This change adds the cloud name to the indexed data. This allows you to filter out other cloud's results if you have multiple clouds feeding browbeat result data into ElasticSearch. Change-Id: I92c764af115736380660157ad4da54f737e1db98
94 lines
3.9 KiB
Python
94 lines
3.9 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import logging
|
|
import subprocess
|
|
|
|
|
|
class Grafana:
|
|
|
|
def __init__(self, config):
|
|
self.logger = logging.getLogger('browbeat.Grafana')
|
|
self.config = config
|
|
self.cloud_name = self.config['browbeat']['cloud_name']
|
|
self.hosts_file = self.config['ansible']['hosts']
|
|
self.grafana_ip = self.config['grafana']['grafana_ip']
|
|
self.grafana_port = self.config['grafana']['grafana_port']
|
|
self.playbook = self.config['ansible']['grafana_snapshot']
|
|
self.grafana_url = {}
|
|
|
|
def extra_vars(self, from_ts, to_ts, result_dir, test_name):
|
|
extra_vars = 'grafana_ip={} '.format(
|
|
self.config['grafana']['grafana_ip'])
|
|
extra_vars += 'grafana_port={} '.format(
|
|
self.config['grafana']['grafana_port'])
|
|
extra_vars += 'from={} '.format(from_ts)
|
|
extra_vars += 'to={} '.format(to_ts)
|
|
extra_vars += 'results_dir={}/{} '.format(result_dir, test_name)
|
|
extra_vars += 'var_cloud={} '.format(self.cloud_name)
|
|
if self.config['grafana']['snapshot']['snapshot_compute']:
|
|
extra_vars += 'snapshot_compute=true '
|
|
return extra_vars
|
|
|
|
def grafana_urls(self):
|
|
return self.grafana_url
|
|
|
|
def create_grafana_urls(self, time):
|
|
if 'grafana' in self.config and self.config['grafana']['enabled']:
|
|
from_ts = time['from_ts']
|
|
to_ts = time['to_ts']
|
|
url = 'http://{}:{}/dashboard/db/'.format(
|
|
self.grafana_ip, self.grafana_port)
|
|
for dashboard in self.config['grafana']['dashboards']:
|
|
self.grafana_url[dashboard] = '{}{}?from={}&to={}&var-Cloud={}'.format(
|
|
url,
|
|
dashboard,
|
|
from_ts,
|
|
to_ts,
|
|
self.cloud_name)
|
|
|
|
def print_dashboard_url(self, test_name):
|
|
for dashboard in self.grafana_url:
|
|
self.logger.info(
|
|
'{} - Grafana Dashboard {} URL: {}'.format(
|
|
test_name,
|
|
dashboard,
|
|
self.grafana_url[dashboard]))
|
|
|
|
def log_snapshot_playbook_cmd(self, from_ts, to_ts, result_dir, test_name):
|
|
if 'grafana' in self.config and self.config['grafana']['enabled']:
|
|
extra_vars = self.extra_vars(
|
|
from_ts, to_ts, result_dir, test_name)
|
|
snapshot_cmd = 'ansible-playbook -i {} {} -e "{}"'.format(
|
|
self.hosts_file, self.playbook, extra_vars)
|
|
self.logger.info('Snapshot command: {}'.format(snapshot_cmd))
|
|
|
|
def run_playbook(self, from_ts, to_ts, result_dir, test_name):
|
|
if 'grafana' in self.config and self.config['grafana']['enabled']:
|
|
if self.config['grafana']['snapshot']['enabled']:
|
|
extra_vars = self.extra_vars(
|
|
from_ts, to_ts, result_dir, test_name)
|
|
subprocess_cmd = [
|
|
'ansible-playbook',
|
|
'-i',
|
|
self.hosts_file,
|
|
self.playbook,
|
|
'-e',
|
|
'{}'.format(extra_vars)]
|
|
snapshot_log = open('{}/snapshot.log'.format(result_dir), 'a+')
|
|
self.logger.info(
|
|
'Running ansible to create snapshots for: {}'.format(test_name))
|
|
subprocess.Popen(
|
|
subprocess_cmd,
|
|
stdout=snapshot_log,
|
|
stderr=subprocess.STDOUT)
|