yardstick/benchmark/runners/arithmetic.py

   1 ##############################################################################
   2 # Copyright (c) 2015 Ericsson AB and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10 '''A runner that every run arithmetically steps a specified input value to
  11 the scenario. This just means a step value is added to the previous value.
  12 '''
  13
  14 import os
  15 import multiprocessing
  16 import logging
  17 import traceback
  18 import time
  19
  20 from yardstick.benchmark.runners import base
  21
  22 LOG = logging.getLogger(__name__)
  23
  24
  25 def _worker_process(queue, cls, method_name, context, scenario_args):
  26
  27     sequence = 1
  28
  29     interval = context.get("interval", 1)
  30     arg_name = context.get('name')
  31     stop = context.get('stop')
  32     step = context.get('step')
  33     options = scenario_args['options']
  34     start = options.get(arg_name, 0)
  35
  36     context['runner'] = os.getpid()
  37
  38     LOG.info("worker START, step(%s, %d, %d, %d), class %s",
  39              arg_name, start, stop, step, cls)
  40
  41     benchmark = cls(context)
  42     benchmark.setup()
  43     method = getattr(benchmark, method_name)
  44
  45     record_context = {"runner": context["runner"],
  46                       "host": context["host"]}
  47
  48     sla_action = None
  49     if "sla" in scenario_args:
  50         sla_action = scenario_args["sla"].get("action", "assert")
  51
  52     for value in range(start, stop+step, step):
  53
  54         options[arg_name] = value
  55
  56         LOG.debug("runner=%(runner)s seq=%(sequence)s START" %
  57                   {"runner": context["runner"], "sequence": sequence})
  58
  59         data = {}
  60         errors = ""
  61
  62         try:
  63             data = method(scenario_args)
  64         except AssertionError as assertion:
  65             # SLA validation failed in scenario, determine what to do now
  66             if sla_action == "assert":
  67                 raise
  68             elif sla_action == "monitor":
  69                 LOG.warning("SLA validation failed: %s" % assertion.args)
  70                 errors = assertion.args
  71         except Exception as e:
  72             errors = traceback.format_exc()
  73             LOG.exception(e)
  74
  75         time.sleep(interval)
  76
  77         benchmark_output = {
  78             'timestamp': time.time(),
  79             'sequence': sequence,
  80             'data': data,
  81             'errors': errors
  82         }
  83
  84         queue.put({'context': record_context, 'sargs:': scenario_args,
  85                    'benchmark': benchmark_output})
  86
  87         LOG.debug("runner=%(runner)s seq=%(sequence)s END" %
  88                   {"runner": context["runner"], "sequence": sequence})
  89
  90         sequence += 1
  91
  92         if errors:
  93             break
  94
  95     benchmark.teardown()
  96     LOG.info("worker END")
  97
  98
  99 class ArithmeticRunner(base.Runner):
 100     '''Run a scenario arithmetically stepping an input value
 101
 102   Parameters
 103     interval - time to wait between each scenario invocation
 104         type:    int
 105         unit:    seconds
 106         default: 1 sec
 107     name - name of scenario option that will be increased for each invocation
 108         type:    string
 109         unit:    na
 110         default: none
 111     start - value to use in first invocation of scenario
 112         type:    int
 113         unit:    na
 114         default: none
 115     step - value added to start value in next invocation of scenario
 116         type:    int
 117         unit:    na
 118         default: none
 119     stop - value indicating end of invocation
 120         type:    int
 121         unit:    na
 122         default: none
 123     '''
 124
 125     __execution_type__ = 'Arithmetic'
 126
 127     def _run_benchmark(self, cls, method, scenario_args):
 128         self.process = multiprocessing.Process(
 129             target=_worker_process,
 130             args=(self.result_queue, cls, method, self.config, scenario_args))
 131         self.process.start()