mirror of
https://github.com/valitydev/yandex-tank.git
synced 2024-11-07 02:45:21 +00:00
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
from Tank.Core import AbstractPlugin
|
|
from Tank.Plugins.Aggregator import AggregatorPlugin
|
|
from Tank import Utils
|
|
import subprocess
|
|
import tempfile
|
|
|
|
|
|
class ApacheBenchmarkPlugin(AbstractPlugin):
|
|
|
|
SECTION = 'ab'
|
|
|
|
def __init__(self, core):
|
|
AbstractPlugin.__init__(self, core)
|
|
self.end = None
|
|
self.out_file = None
|
|
self.process = None
|
|
|
|
@staticmethod
|
|
def get_key():
|
|
return __file__;
|
|
|
|
def configure(self):
|
|
self.options = self.get_option("options", '')
|
|
self.url = self.get_option("url", 'http://localhost/')
|
|
self.requests = self.get_option("requests", '100')
|
|
self.concurrency = self.get_option("concurrency", '1')
|
|
self.out_file = tempfile.mkstemp('.log', 'ab_')[1]
|
|
self.core.add_artifact_file(self.out_file)
|
|
|
|
def prepare_test(self):
|
|
aggregator = None
|
|
try:
|
|
aggregator = self.core.get_plugin_of_type(AggregatorPlugin)
|
|
except Exception, ex:
|
|
self.log.warning("No aggregator found: %s", ex)
|
|
|
|
if aggregator:
|
|
aggregator.set_source_files(self.out_file, None)
|
|
|
|
def start_test(self):
|
|
args = ['ab', '-r', '-g', self.out_file,
|
|
'-n', self.requests,
|
|
'-c', self.concurrency, self.url]
|
|
self.log.debug("Starting ab with arguments: %s", args)
|
|
self.process = subprocess.Popen(args, stderr=subprocess.PIPE, stdout=subprocess.PIPE)
|
|
|
|
def is_test_finished(self):
|
|
rc = self.process.poll()
|
|
if rc != None:
|
|
self.log.debug("%s exit code: %s", self.SECTION, rc)
|
|
return rc
|
|
else:
|
|
return -1
|
|
|
|
if self.process:
|
|
Utils.log_stdout_stderr(self.log, self.process.stdout, self.process.stderr, self.SECTION)
|
|
|
|
|
|
def end_test(self, retcode):
|
|
if self.process and self.process.poll() == None:
|
|
self.log.warn("Terminating ab process with PID %s", self.process.pid)
|
|
self.process.terminate()
|
|
else:
|
|
self.log.debug("Seems ab finished OK")
|
|
|
|
if self.process:
|
|
Utils.log_stdout_stderr(self.log, self.process.stdout, self.process.stderr, self.SECTION)
|
|
return retcode
|
|
|
|
|
|
|