installer.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635
  1. import subprocess
  2. import os
  3. import time
  4. import traceback
  5. import sys
  6. class Installer:
  7. ############################################################
  8. # install_software
  9. ############################################################
  10. def install_software(self):
  11. if self.benchmarker.install == 'all' or self.benchmarker.install == 'server':
  12. self.__install_server_software()
  13. if self.benchmarker.install == 'all' or self.benchmarker.install == 'database':
  14. self.__install_database_software()
  15. if self.benchmarker.install == 'all' or self.benchmarker.install == 'client':
  16. self.__install_client_software()
  17. ############################################################
  18. # End install_software
  19. ############################################################
  20. ############################################################
  21. # __install_server_software
  22. ############################################################
  23. def __install_server_software(self):
  24. print("\nINSTALL: Installing server software\n")
  25. #######################################
  26. # Prerequisites
  27. #######################################
  28. self.__run_command("sudo apt-get update", True)
  29. self.__run_command("sudo apt-get upgrade", True)
  30. self.__run_command("sudo apt-get install build-essential libpcre3 libpcre3-dev libpcrecpp0 libssl-dev zlib1g-dev python-software-properties unzip git-core libcurl4-openssl-dev libbz2-dev libmysqlclient-dev mongodb-clients libreadline6-dev libyaml-dev libsqlite3-dev sqlite3 libxml2-dev libxslt-dev libgdbm-dev ncurses-dev automake libffi-dev htop libtool bison libevent-dev libgstreamer-plugins-base0.10-0 libgstreamer0.10-0 liborc-0.4-0 libwxbase2.8-0 libwxgtk2.8-0 libgnutls-dev libjson0-dev libmcrypt-dev libicu-dev cmake gettext curl libpq-dev mercurial mlton", True)
  31. self.__run_command("sudo add-apt-repository ppa:ubuntu-toolchain-r/test", True)
  32. self.__run_command("sudo apt-get update", True)
  33. self.__run_command("sudo apt-get install gcc-4.8 g++-4.8", True)
  34. self.__run_command("cp ../config/benchmark_profile ../../.bash_profile")
  35. self.__run_command("cat ../config/benchmark_profile >> ../../.profile")
  36. self.__run_command("cat ../config/benchmark_profile >> ../../.bashrc")
  37. self.__run_command(". ../../.profile")
  38. self.__run_command("sudo sh -c \"echo '* - nofile 65535' >> /etc/security/limits.conf\"")
  39. ##############################################################
  40. # System Tools
  41. ##############################################################
  42. #
  43. # Leiningen
  44. #
  45. self.__run_command("mkdir -p bin")
  46. self.__download("https://raw.github.com/technomancy/leiningen/stable/bin/lein")
  47. self.__run_command("mv lein bin/lein")
  48. self.__run_command("chmod +x bin/lein")
  49. #
  50. # Maven
  51. #
  52. self.__run_command("sudo apt-get install maven -qq")
  53. self.__run_command("mvn -version")
  54. #######################################
  55. # Languages
  56. #######################################
  57. self._install_python()
  58. #
  59. # Dart
  60. #
  61. self.__download("http://storage.googleapis.com/dart-archive/channels/stable/release/latest/sdk/dartsdk-linux-x64-release.zip")
  62. self.__run_command("unzip -qqo dartsdk-linux-x64-release.zip")
  63. #
  64. # Erlang
  65. #
  66. self.__run_command("sudo cp ../config/erlang.list /etc/apt/sources.list.d/erlang.list")
  67. self.__download("http://binaries.erlang-solutions.com/debian/erlang_solutions.asc")
  68. self.__run_command("sudo apt-key add erlang_solutions.asc")
  69. self.__run_command("sudo apt-get update")
  70. self.__run_command("sudo apt-get install esl-erlang", True)
  71. #
  72. # nodejs
  73. #
  74. self.__download("http://nodejs.org/dist/v0.10.8/node-v0.10.8-linux-x64.tar.gz")
  75. self.__run_command("tar xzf node-v0.10.8-linux-x64.tar.gz")
  76. #
  77. # Java
  78. #
  79. self.__run_command("sudo apt-get install openjdk-7-jdk", True)
  80. self.__run_command("sudo apt-get remove --purge openjdk-6-jre openjdk-6-jre-headless", True)
  81. #
  82. # Ruby/JRuby
  83. #
  84. self.__run_command("curl -L get.rvm.io | bash -s head --auto-dotfiles")
  85. self.__run_command("echo rvm_auto_reload_flag=2 >> ~/.rvmrc")
  86. self.__bash_from_string("source ~/.rvm/scripts/'rvm' && rvm install 2.0.0-p0")
  87. self.__bash_from_string("source ~/.rvm/scripts/'rvm' && rvm 2.0.0-p0 do gem install bundler")
  88. self.__bash_from_string("source ~/.rvm/scripts/'rvm' && rvm install jruby-1.7.8")
  89. self.__bash_from_string("source ~/.rvm/scripts/'rvm' && rvm jruby-1.7.8 do gem install bundler")
  90. #
  91. # go
  92. #
  93. self.__download("http://go.googlecode.com/files/go1.2.linux-amd64.tar.gz");
  94. self.__run_command("tar xzf go1.2.linux-amd64.tar.gz")
  95. #
  96. # Perl
  97. #
  98. self.__download("http://downloads.activestate.com/ActivePerl/releases/5.16.3.1603/ActivePerl-5.16.3.1603-x86_64-linux-glibc-2.3.5-296746.tar.gz");
  99. self.__run_command("tar xzf ActivePerl-5.16.3.1603-x86_64-linux-glibc-2.3.5-296746.tar.gz");
  100. self.__run_command("sudo ./install.sh --license-accepted --prefix /opt/ActivePerl-5.16 --no-install-html", cwd="ActivePerl-5.16.3.1603-x86_64-linux-glibc-2.3.5-296746", send_yes=True, retry=True)
  101. self.__download("http://cpanmin.us", "cpanminus.pl")
  102. self.__run_command("perl cpanminus.pl --sudo App::cpanminus", retry=True)
  103. self.__run_command("cpanm -f -S DBI DBD::mysql Kelp Dancer Mojolicious Kelp::Module::JSON::XS Dancer::Plugin::Database Starman Plack JSON Web::Simple DBD::Pg JSON::XS EV HTTP::Parser::XS Monoceros EV IO::Socket::IP IO::Socket::SSL", retry=True)
  104. #
  105. # php
  106. #
  107. self.__download("http://museum.php.net/php5/php-5.4.13.tar.gz")
  108. self.__run_command("tar xzf php-5.4.13.tar.gz")
  109. self.__run_command("./configure --with-pdo-mysql --with-mysql --with-mcrypt --enable-intl --enable-mbstring --enable-fpm --with-fpm-user=www-data --with-fpm-group=www-data --with-openssl", cwd="php-5.4.13")
  110. self.__run_command("make", cwd="php-5.4.13")
  111. self.__run_command("sudo make install", cwd="php-5.4.13")
  112. self.__run_command("printf \"\\n\" | sudo pecl install -f apc-beta", cwd="php-5.4.13", retry=True)
  113. self.__run_command("sudo cp ../config/php.ini /usr/local/lib/php.ini")
  114. self.__run_command("sudo cp ../config/php-fpm.conf /usr/local/lib/php-fpm.conf")
  115. # Composer
  116. self.__download("https://getcomposer.org/installer", "composer-installer.php")
  117. self.__run_command("php composer-installer.php --install-dir=bin")
  118. # Phalcon
  119. self.__run_command("git clone git://github.com/phalcon/cphalcon.git", retry=True)
  120. self.__run_command("sudo ./install", cwd="cphalcon/build")
  121. # YAF
  122. self.__run_command("sudo pecl install -f yaf")
  123. #
  124. # Haskell
  125. #
  126. self.__run_command("sudo apt-get install ghc cabal-install", True)
  127. #
  128. # RingoJs
  129. #
  130. self.__download("http://www.ringojs.org/downloads/ringojs_0.9-1_all.deb")
  131. self.__run_command("sudo apt-get install jsvc", True)
  132. self.__run_command("sudo dpkg -i ringojs_0.9-1_all.deb", True)
  133. self.__run_command("rm ringojs_0.9-1_all.deb")
  134. #
  135. # Mono
  136. #
  137. self.__run_command("git clone git://github.com/mono/mono", retry=True)
  138. self.__run_command("git checkout mono-3.2.8-branch", cwd="mono")
  139. self.__run_command("./autogen.sh --prefix=/usr/local", cwd="mono")
  140. self.__run_command("make get-monolite-latest", cwd="mono")
  141. self.__run_command("make EXTERNAL_MCS=${PWD}/mcs/class/lib/monolite/basic.exe", cwd="mono")
  142. self.__run_command("sudo make install", cwd="mono")
  143. self.__run_command("mozroots --import --sync", retry=True)
  144. self.__run_command("git clone git://github.com/mono/xsp", retry=True)
  145. self.__run_command("./autogen.sh --prefix=/usr/local", cwd="xsp")
  146. self.__run_command("make", cwd="xsp")
  147. self.__run_command("sudo make install", cwd="xsp")
  148. #
  149. # Nimrod
  150. #
  151. self.__run_command("git clone git://github.com/Araq/Nimrod.git nimrod", retry=True)
  152. self.__run_command("git checkout 987ac2439a87d74838233a7b188e4db340495ee5", cwd="nimrod")
  153. self.__run_command("git clone git://github.com/nimrod-code/csources.git", cwd="nimrod", retry=True)
  154. self.__run_command("git checkout 704015887981932c78a033dd5ede623b2ad6ae27", cwd="nimrod/csources")
  155. self.__run_command("chmod +x build.sh", cwd="nimrod/csources")
  156. self.__run_command("./build.sh", cwd="nimrod/csources")
  157. self.__run_command("bin/nimrod c koch", cwd="nimrod")
  158. self.__run_command("./koch boot -d:release", cwd="nimrod")
  159. #
  160. # Racket
  161. #
  162. self.__download("https://github.com/plt/racket/archive/v5.3.6.tar.gz", "racket-5.3.6.tar.gz")
  163. self.__run_command("tar xzf racket-5.3.6.tar.gz")
  164. self.__run_command("./configure", cwd="racket-5.3.6/src")
  165. self.__run_command("make", cwd="racket-5.3.6/src")
  166. self.__run_command("sudo make install", cwd="racket-5.3.6/src")
  167. #
  168. # Ur/Web
  169. #
  170. self.__run_command("hg clone http://hg.impredicative.com/urweb")
  171. self.__run_command("./autogen.sh", cwd="urweb")
  172. self.__run_command("./configure", cwd="urweb")
  173. self.__run_command("make", cwd="urweb")
  174. self.__run_command("sudo make install", cwd="urweb")
  175. #
  176. # HHVM
  177. #
  178. self.__run_command("wget -O - http://dl.hhvm.com/conf/hhvm.gpg.key | sudo apt-key add -")
  179. self.__run_command("echo deb http://dl.hhvm.com/ubuntu trusty main | sudo tee /etc/apt/sources.list.d/hhvm.list")
  180. self.__run_command("sudo apt-get update")
  181. self.__run_command("sudo apt-get install hhvm", True)
  182. #######################################
  183. # Webservers
  184. #######################################
  185. #
  186. # Nginx
  187. #
  188. self.__download("http://nginx.org/download/nginx-1.4.1.tar.gz")
  189. self.__run_command("tar xzf nginx-1.4.1.tar.gz")
  190. self.__run_command("./configure", cwd="nginx-1.4.1")
  191. self.__run_command("make", cwd="nginx-1.4.1")
  192. self.__run_command("sudo make install", cwd="nginx-1.4.1")
  193. #
  194. # Openresty (nginx with lua stuff)
  195. #
  196. self.__download("http://openresty.org/download/ngx_openresty-1.5.8.1.tar.gz")
  197. self.__run_command("tar xzf ngx_openresty-1.5.8.1.tar.gz")
  198. self.__run_command("./configure --with-luajit --with-http_postgres_module", cwd="ngx_openresty-1.5.8.1")
  199. self.__run_command("make", cwd="ngx_openresty-1.5.8.1")
  200. self.__run_command("sudo make install", cwd="ngx_openresty-1.5.8.1")
  201. #
  202. # Lapis
  203. #
  204. self.__run_command("sudo apt-get install luarocks", True)
  205. self.__run_command("sudo luarocks install http://github.com/leafo/lapis/raw/master/lapis-dev-1.rockspec")
  206. #
  207. # Resin
  208. #
  209. self.__run_command("sudo cp -r /usr/lib/jvm/java-1.7.0-openjdk-amd64/include /usr/lib/jvm/java-1.7.0-openjdk-amd64/jre/bin/")
  210. self.__download("http://www.caucho.com/download/resin-4.0.36.tar.gz")
  211. self.__run_command("tar xzf resin-4.0.36.tar.gz")
  212. self.__run_command("./configure --prefix=`pwd`", cwd="resin-4.0.36")
  213. self.__run_command("make", cwd="resin-4.0.36")
  214. self.__run_command("make install", cwd="resin-4.0.36")
  215. self.__run_command("mv conf/resin.properties conf/resin.properties.orig", cwd="resin-4.0.36")
  216. self.__run_command("cat ../config/resin.properties > resin-4.0.36/conf/resin.properties")
  217. self.__run_command("mv conf/resin.xml conf/resin.xml.orig", cwd="resin-4.0.36")
  218. self.__run_command("cat ../config/resin.xml > resin-4.0.36/conf/resin.xml")
  219. #
  220. # Mongrel2
  221. #
  222. self.__download("http://download.zeromq.org/zeromq-4.0.3.tar.gz")
  223. self.__run_command("tar xzf zeromq-4.0.3.tar.gz")
  224. self.__run_command("./configure", cwd="zeromq-4.0.3")
  225. self.__run_command("make", cwd="zeromq-4.0.3")
  226. self.__run_command("sudo make install", cwd="zeromq-4.0.3")
  227. self.__run_command("sudo apt-get install sqlite3 libsqlite3-dev uuid uuid-runtime uuid-dev", True)
  228. self.__run_command("sudo ldconfig -v")
  229. self.__download("https://github.com/zedshaw/mongrel2/tarball/v1.8.1", "mongrel2.tar.gz")
  230. self.__run_command("tar xvf mongrel2.tar.gz")
  231. self.__run_command("mv zedshaw-mongrel2-aa2ecf8 mongrel2")
  232. # for zmq4, we update the following file manually (not in v1.8.1)
  233. self.__download("https://raw.github.com/zedshaw/mongrel2/9b565eeea003783c47502c2d350b99c9684ce97c/src/zmq_compat.h")
  234. self.__run_command("mv -f zmq_compat.h mongrel2/src/")
  235. self.__run_command("make clean all && sudo make install", cwd="mongrel2")
  236. ##############################################################
  237. # Frameworks
  238. ##############################################################
  239. #
  240. # Grails
  241. #
  242. self.__download("http://dist.springframework.org.s3.amazonaws.com/release/GRAILS/grails-2.3.3.zip")
  243. self.__run_command("unzip -o grails-2.3.3.zip")
  244. #
  245. # Play 2
  246. #
  247. self.__download("http://downloads.typesafe.com/play/2.2.0/play-2.2.0.zip")
  248. self.__run_command("unzip -o play-2.2.0.zip")
  249. #
  250. # Play 1
  251. #
  252. self.__download("http://downloads.typesafe.com/releases/play-1.2.5.zip")
  253. self.__run_command("unzip -o play-1.2.5.zip")
  254. self.__run_command("mv play-1.2.5/play play-1.2.5/play1")
  255. # siena
  256. self.__run_command("yes | play-1.2.5/play1 install siena")
  257. #
  258. # TreeFrog Framework
  259. #
  260. self.__run_command("sudo apt-get install qt4-qmake libqt4-dev libqt4-sql-mysql libqt4-sql-psql g++", True)
  261. self.__download("http://downloads.sourceforge.net/project/treefrog/src/treefrog-1.7.5.tar.gz")
  262. self.__run_command("tar xzf treefrog-1.7.5.tar.gz")
  263. self.__run_command("rm treefrog-1.7.5.tar.gz")
  264. self.__run_command("./configure", cwd="treefrog-1.7.5")
  265. self.__run_command("make", cwd="treefrog-1.7.5/src")
  266. self.__run_command("sudo make install", cwd="treefrog-1.7.5/src")
  267. self.__run_command("make", cwd="treefrog-1.7.5/tools")
  268. self.__run_command("sudo make install", cwd="treefrog-1.7.5/tools")
  269. #
  270. # Vert.x
  271. #
  272. self.__download("http://dl.bintray.com/vertx/downloads/vert.x-2.1M3.tar.gz?direct=true", "vert.x-2.1M3.tar.gz")
  273. self.__run_command("tar xzf vert.x-2.1M3.tar.gz")
  274. #
  275. # Yesod
  276. #
  277. self.__run_command("cabal update", retry=True)
  278. self.__run_command("cabal install yesod persistent-mysql", retry=True)
  279. #
  280. # Jester
  281. #
  282. self.__run_command("git clone git://github.com/dom96/jester.git jester/jester", retry=True)
  283. #
  284. # Onion
  285. #
  286. self.__run_command("git clone https://github.com/davidmoreno/onion.git")
  287. self.__run_command("mkdir build", cwd="onion")
  288. self.__run_command("cmake ..", cwd="onion/build")
  289. self.__run_command("make", cwd="onion/build")
  290. # nawak
  291. #
  292. self.__run_command("git clone git://github.com/idlewan/nawak.git nawak/nawak", retry=True)
  293. print("\nINSTALL: Finished installing server software\n")
  294. ############################################################
  295. # End __install_server_software
  296. ############################################################
  297. def _install_python(self):
  298. """Install Python runtime, frameworks and libraries"""
  299. # PyPy 2.3.1
  300. f = "pypy-2.3.1-linux64.tar.bz2"
  301. if not os.path.exists(f):
  302. self.__download("https://bitbucket.org/pypy/pypy/downloads/" + f, f)
  303. self.__run_command("tar xjf " + f)
  304. self.__run_command('ln -sf pypy-2.3.1-linux64 pypy')
  305. # CPython 2.7.7
  306. f = "Python-2.7.7.tgz"
  307. if not os.path.exists(f):
  308. self.__download("http://www.python.org/ftp/python/2.7.7/" + f, f)
  309. self.__run_command("tar xf " + f)
  310. self.__run_command("./configure --prefix=$HOME/FrameworkBenchmarks/installs/py2 --disable-shared", cwd="Python-2.7.7")
  311. self.__run_command("make -j4", cwd="Python-2.7.7")
  312. self.__run_command("make install", cwd="Python-2.7.7")
  313. # CPython 3.4.1
  314. f = "Python-3.4.1.tar.xz"
  315. if not os.path.exists(f):
  316. self.__download("https://www.python.org/ftp/python/3.4.1/" + f, f)
  317. self.__run_command("tar xf " + f)
  318. self.__run_command("./configure --prefix=$HOME/FrameworkBenchmarks/installs/py3 --disable-shared", cwd="Python-3.4.1")
  319. self.__run_command("make -j4", cwd="Python-3.4.1")
  320. self.__run_command("make install", cwd="Python-3.4.1")
  321. if not os.path.exists("get-pip.py"):
  322. self.__download("https://bootstrap.pypa.io/get-pip.py", "get-pip.py")
  323. self.__run_command("py2/bin/python get-pip.py")
  324. self.__run_command("pypy/bin/pypy get-pip.py")
  325. # Python 3.4.1 installs pip by default.
  326. self.__run_command('py2/bin/pip install -r ../config/requirements.txt')
  327. self.__run_command('py3/bin/pip install -r ../config/requirements.txt')
  328. self.__run_command('pypy/bin/pip install -r ../config/requirements-pypy.txt')
  329. ############################################################
  330. # __install_error
  331. ############################################################
  332. def __install_error(self, message):
  333. print("\nINSTALL ERROR: %s\n" % message)
  334. if self.benchmarker.install_error_action == 'abort':
  335. sys.exit("Installation aborted.")
  336. ############################################################
  337. # End __install_error
  338. ############################################################
  339. ############################################################
  340. # __install_database_software
  341. ############################################################
  342. def __install_database_software(self):
  343. print("\nINSTALL: Installing database software\n")
  344. self.__run_command("cd .. && " + self.benchmarker.database_sftp_string(batch_file="../config/database_sftp_batch"), True)
  345. remote_script = """
  346. ##############################
  347. # Prerequisites
  348. ##############################
  349. yes | sudo apt-get update
  350. yes | sudo apt-get install build-essential git libev-dev libpq-dev libreadline6-dev postgresql
  351. sudo sh -c "echo '* - nofile 65535' >> /etc/security/limits.conf"
  352. sudo mkdir -p /ssd
  353. sudo mkdir -p /ssd/log
  354. ##############################
  355. # MySQL
  356. ##############################
  357. sudo sh -c "echo mysql-server mysql-server/root_password_again select secret | debconf-set-selections"
  358. sudo sh -c "echo mysql-server mysql-server/root_password select secret | debconf-set-selections"
  359. yes | sudo apt-get install mysql-server-5.6
  360. sudo stop mysql
  361. # disable checking of disk size
  362. sudo cp mysql /etc/init.d/mysql
  363. sudo chmod +x /etc/init.d/mysql
  364. sudo cp mysql.conf /etc/init/mysql.conf
  365. # use the my.cnf file to overwrite /etc/mysql/my.cnf
  366. sudo mv /etc/mysql/my.cnf /etc/mysql/my.cnf.orig
  367. sudo mv my.cnf /etc/mysql/my.cnf
  368. sudo cp -R -p /var/lib/mysql /ssd/
  369. sudo cp -R -p /var/log/mysql /ssd/log
  370. sudo cp usr.sbin.mysqld /etc/apparmor.d/
  371. sudo /etc/init.d/apparmor reload
  372. sudo start mysql
  373. # Insert data
  374. mysql -uroot -psecret < create.sql
  375. ##############################
  376. # Postgres
  377. ##############################
  378. sudo useradd benchmarkdbuser -p benchmarkdbpass
  379. sudo -u postgres psql template1 < create-postgres-database.sql
  380. sudo -u benchmarkdbuser psql hello_world < create-postgres.sql
  381. sudo -u postgres -H /etc/init.d/postgresql stop
  382. sudo mv postgresql.conf /etc/postgresql/9.3/main/postgresql.conf
  383. sudo mv pg_hba.conf /etc/postgresql/9.3/main/pg_hba.conf
  384. sudo cp -R -p /var/lib/postgresql/9.3/main /ssd/postgresql
  385. sudo -u postgres -H /etc/init.d/postgresql start
  386. sudo mv 60-postgresql-shm.conf /etc/sysctl.d/60-postgresql-shm.conf
  387. ##############################
  388. # MongoDB
  389. ##############################
  390. sudo apt-key adv --keyserver hkp://keyserver.ubuntu.com:80 --recv 7F0CEB10
  391. sudo cp 10gen.list /etc/apt/sources.list.d/10gen.list
  392. sudo apt-get update
  393. yes | sudo apt-get remove mongodb-clients
  394. yes | sudo apt-get install mongodb-10gen
  395. sudo stop mongodb
  396. sudo mv /etc/mongodb.conf /etc/mongodb.conf.orig
  397. sudo mv mongodb.conf /etc/mongodb.conf
  398. sudo cp -R -p /var/lib/mongodb /ssd/
  399. sudo cp -R -p /var/log/mongodb /ssd/log/
  400. sudo start mongodb
  401. """
  402. print("\nINSTALL: %s" % self.benchmarker.database_ssh_string)
  403. p = subprocess.Popen(self.benchmarker.database_ssh_string.split(" "), stdin=subprocess.PIPE)
  404. p.communicate(remote_script)
  405. returncode = p.returncode
  406. if returncode != 0:
  407. self.__install_error("status code %s running subprocess '%s'." % (returncode, self.benchmarker.database_ssh_string))
  408. print("\nINSTALL: Finished installing database software\n")
  409. ############################################################
  410. # End __install_database_software
  411. ############################################################
  412. ############################################################
  413. # __install_client_software
  414. ############################################################
  415. def __install_client_software(self):
  416. print("\nINSTALL: Installing client software\n")
  417. remote_script = """
  418. ##############################
  419. # Prerequisites
  420. ##############################
  421. yes | sudo apt-get update
  422. yes | sudo apt-get install build-essential git libev-dev libpq-dev libreadline6-dev
  423. sudo sh -c "echo '* - nofile 65535' >> /etc/security/limits.conf"
  424. ##############################
  425. # wrk
  426. ##############################
  427. git clone https://github.com/wg/wrk.git
  428. cd wrk
  429. make
  430. sudo cp wrk /usr/local/bin
  431. cd ~
  432. #############################
  433. # pipeline.lua
  434. #############################
  435. cat << EOF | tee pipeline.lua
  436. init = function(args)
  437. wrk.init(args)
  438. local r = {}
  439. local depth = tonumber(args[1]) or 1
  440. for i=1,depth do
  441. r[i] = wrk.format()
  442. end
  443. req = table.concat(r)
  444. end
  445. request = function()
  446. return req
  447. end
  448. EOF
  449. """
  450. print("\nINSTALL: %s" % self.benchmarker.client_ssh_string)
  451. p = subprocess.Popen(self.benchmarker.client_ssh_string.split(" "), stdin=subprocess.PIPE)
  452. p.communicate(remote_script)
  453. returncode = p.returncode
  454. if returncode != 0:
  455. self.__install_error("status code %s running subprocess '%s'." % (returncode, self.benchmarker.client_ssh_string))
  456. print("\nINSTALL: Finished installing client software\n")
  457. ############################################################
  458. # End __install_client_software
  459. ############################################################
  460. ############################################################
  461. # __run_command
  462. ############################################################
  463. def __run_command(self, command, send_yes=False, cwd=None, retry=False):
  464. try:
  465. cwd = os.path.join(self.install_dir, cwd)
  466. except AttributeError:
  467. cwd = self.install_dir
  468. if retry:
  469. max_attempts = 5
  470. else:
  471. max_attempts = 1
  472. attempt = 1
  473. delay = 0
  474. if send_yes:
  475. command = "yes yes | " + command
  476. print("\nINSTALL: %s (cwd=%s)" % (command, cwd))
  477. while True:
  478. error_message = ""
  479. try:
  480. # Execute command.
  481. subprocess.check_call(command, shell=True, cwd=cwd)
  482. break # Exit loop if successful.
  483. except:
  484. exceptionType, exceptionValue, exceptionTraceBack = sys.exc_info()
  485. error_message = "".join(traceback.format_exception_only(exceptionType, exceptionValue))
  486. print error_message
  487. # Exit if there are no more attempts left.
  488. attempt += 1
  489. if attempt > max_attempts:
  490. break
  491. # Delay before next attempt.
  492. if delay == 0:
  493. delay = 5
  494. else:
  495. delay = delay * 2
  496. print("Attempt %s/%s starting in %s seconds." % (attempt, max_attempts, delay))
  497. time.sleep(delay)
  498. if error_message:
  499. self.__install_error(error_message)
  500. ############################################################
  501. # End __run_command
  502. ############################################################
  503. ############################################################
  504. # __bash_from_string
  505. # Runs bash -c "command" in install_dir.
  506. ############################################################
  507. def __bash_from_string(self, command):
  508. self.__run_command('bash -c "%s"' % command)
  509. ############################################################
  510. # End __bash_from_string
  511. ############################################################
  512. ############################################################
  513. # __download
  514. # Downloads a file from a URI.
  515. ############################################################
  516. def __download(self, uri, filename=""):
  517. if filename:
  518. filename_option = "-O %s " % filename
  519. else:
  520. filename_option = ""
  521. command = "wget -nv --no-check-certificate --trust-server-names %s%s" % (filename_option, uri)
  522. self.__run_command(command, retry=True)
  523. ############################################################
  524. # End __download
  525. ############################################################
  526. ############################################################
  527. # __init__(benchmarker)
  528. ############################################################
  529. def __init__(self, benchmarker):
  530. self.benchmarker = benchmarker
  531. self.install_dir = "installs"
  532. try:
  533. os.mkdir(self.install_dir)
  534. except OSError:
  535. pass
  536. ############################################################
  537. # End __init__
  538. ############################################################
  539. # vim: sw=2