setup.py 778 B

1234567891011121314151617181920212223242526
  1. import subprocess
  2. import sys
  3. import setup_util
  4. import os
  5. def start(args):
  6. setup_util.replace_text("snap/bench/cfg/db.cfg", "host=\".*\"", "host=\"" + args.database_host + "\"")
  7. subprocess.check_call("cabal configure", shell=True, cwd="snap/bench")
  8. subprocess.check_call("cabal build", shell=True, cwd="snap/bench")
  9. t = str(args.max_threads)
  10. subprocess.Popen("dist/build/snap-bench/snap-bench +RTS -A" + t + "M -N" + t + " > /dev/null", shell=True, cwd="snap/bench")
  11. return 0
  12. def stop():
  13. p = subprocess.Popen(['ps', 'aux'], stdout=subprocess.PIPE)
  14. out, err = p.communicate()
  15. for line in out.splitlines():
  16. if 'snap-bench' in line:
  17. try:
  18. pid = int(line.split(None, 2)[1])
  19. os.kill(pid, 9)
  20. except OSError:
  21. pass
  22. return 0