import os
import sys
import time
import setup_util
import subprocess
import multiprocessing

def get_env_for_database(args):
  if args.database == 'MySQL':
     driver = 'mysql'
     dbname = 'hello_world'
  else:
     driver = 'sqlite'
     dbname = os.environ['ULIB_ROOT'] + '/db/%.*s'
  return {
      'UMEMPOOL': '135,0,0,34,8465,129,-17,-22,41',
      'ORM_DRIVER': driver,
      'ORM_OPTION': "host=" + args.database_host + " user=benchmarkdbuser password=benchmarkdbpass character-set=utf8 dbname=" + dbname
    }

def start(args, logfile, errfile):
  try:
    ulib_root = os.environ['ULIB_ROOT']
    fcfg = ulib_root + "/benchmark.cfg"
    fprg = ulib_root + "/bin/userver_tcp"
    # 1. Change ULib Server configuration
    setup_util.replace_text(fcfg, "PREFORK_CHILD .*", "PREFORK_CHILD " + str(multiprocessing.cpu_count()))
    # 2. Start ULib Server (userver_tcp)
    subprocess.Popen( "%s -c %s" % (fprg, fcfg), shell=True, stdout=logfile, stderr=errfile, env = get_env_for_database(args))
    return 0
  except subprocess.CalledProcessError:
    return 1

def stop(logfile, errfile):
  try:
    # Stop ULib Server (userver_tcp)
    subprocess.check_call("kill -TERM $( cat $ULIB_ROOT/userver_tcp.pid )", shell=True, stderr=errfile, stdout=logfile)
    time.sleep(3);
    p = subprocess.Popen(['pgrep', 'userver_tcp'], stdout=subprocess.PIPE)
    out, err = p.communicate()
    for line in out.splitlines():
       if 'userver_tcp' in line:
         pid = int(line.split(None, 2)[1])
         os.kill(pid, 9)
    subprocess.call("rm -f $ULIB_ROOT/userver_tcp.pid", shell=True, stderr=errfile, stdout=logfile)
    return 0
  except subprocess.CalledProcessError:
    return 1