import subprocess

from .base import BaseService
from ..hands import *

__all__ = ['SchedulerService']


class SchedulerService(BaseService):

    def __init__(self, **kwargs):
        self.worker = 1
        super().__init__(**kwargs)

    @property
    def cmd(self):
        print("\n- Start Scheduler Server")

        log_format = '%(h)s %(t)s %(L)ss "%(r)s" %(s)s %(b)s '
        bind = f'0.0.0.0:{CONFIG.get("SCHEDULER_HTTP_PORT", 6060)}'
        max_requests = 10240 if int(self.worker) > 1 else 0
        cmd = [
            'gunicorn', 'nebula.wsgi:application',
            '-b', bind,
            '-k', 'gthread',
            '--threads', '200',
            '-w', str(self.worker),
            '--max-requests', str(max_requests),
            '--max-requests-jitter', '2048',
            '--timeout', '30',
            '--graceful-timeout', '300',
            '--access-logformat', log_format,
            '--access-logfile', '/dev/null',
            '--error-logfile', '-'
        ]
        if DEBUG:
            cmd.append('--reload')
        return cmd

    @property
    def cwd(self):
        return APPS_DIR

    def open_subprocess(self):
        # 复制当前环境变量，并设置 ENABLE_SCHEDULER=1
        env = os.environ.copy()
        env['ENABLE_SCHEDULER'] = '1'
        env['SERVER_NAME'] = 'scheduler'
        kwargs = {
            'cwd': self.cwd,
            'stderr': self.log_file,
            'stdout': self.log_file,
            'env': env
        }
        self._process = subprocess.Popen(self.cmd, **kwargs)
