# Copyright 2018-2019 Florian Fischer <florian.fl.fischer@fau.de>
#
# This file is part of allocbench.
#
# allocbench is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# allocbench is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with allocbench.  If not, see <http://www.gnu.org/licenses/>.
"""Definition of the httpd benchmark"""

import re

from src.benchmark import Benchmark
import src.facter


class BenchmarkHTTPD(Benchmark):
    """TODO"""
    def __init__(self):
        name = "httpd"

        self.args = {
            "nthreads": Benchmark.scale_threads_for_cpus(2),
            "site": ["index.html", "index.php"]
        }
        self.cmd = "ab -n 10000 -c {nthreads} localhost:8080/{site}"
        self.measure_cmd = ""
        self.servers = [{"name": "nginx",
                         "cmd": "nginx -c {builddir}/benchmarks/httpd/etc/nginx/nginx.conf"},
                        {"name": "php-fpm",
                         "cmd": "php-fpm -c {builddir}/benchmarks/httpd/etc/php/php.ini "\
                                 "-y {builddir}/benchmarks/httpd/etc/php/php-fpm.conf -F"}]

        self.requirements = ["nginx", "ab"]

        super().__init__(name)

    def prepare(self):
        super().prepare()

        self.results["facts"]["versions"]["nginx"] = src.facter.exe_version(
            "nginx", "-v")
        self.results["facts"]["versions"]["ab"] = src.facter.exe_version(
            "ab", "-V")

    @staticmethod
    def process_output(result, stdout, stderr, allocator, perm):
        result["time"] = re.search(
            "Time taken for tests:\\s*(\\d*\\.\\d*) seconds", stdout).group(1)
        result["requests"] = re.search(
            "Requests per second:\\s*(\\d*\\.\\d*) .*", stdout).group(1)

    def summary(self):
        allocators = self.results["allocators"]

        self.plot_fixed_arg("{requests}",
                            xlabel='"threads"',
                            ylabel='"requests/s"',
                            autoticks=False,
                            filepostfix="requests",
                            title='perm.site + ": requests/s"')

        self.plot_fixed_arg("{nginx_vmhwm}",
                            xlabel='"threads"',
                            ylabel='"VmHWM in KB"',
                            title='perm.site + ": nginx memory usage"',
                            filepostfix="httpd_vmhwm",
                            autoticks=False)

        self.plot_fixed_arg("{php-fpm_vmhwm}",
                            xlabel='"threads"',
                            ylabel='"VmHWM in KB"',
                            title='perm.site + ": php-fpm memory usage"',
                            filepostfix="php-fpm_vmhwm",
                            autoticks=False)


httpd = BenchmarkHTTPD()