55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
from __future__ import with_statement, unicode_literals
|
|
|
|
from fabric.api import sudo, env, hosts
|
|
from fabric.api import task, parallel
|
|
from fabric.contrib.files import sed
|
|
from fabric.operations import run, put
|
|
from fabric.context_managers import settings
|
|
|
|
from hostlist import public_dns_names
|
|
from ssh_key import ssh_key_path
|
|
|
|
# Ignore known_hosts
|
|
# http://docs.fabfile.org/en/1.10/usage/env.html#disable-known-hosts
|
|
env.disable_known_hosts = True
|
|
|
|
# What remote servers should Fabric connect to? With what usernames?
|
|
env.user = 'ubuntu'
|
|
env.hosts = public_dns_names
|
|
|
|
# SSH key files to try when connecting:
|
|
# http://docs.fabfile.org/en/1.10/usage/env.html#key-filename
|
|
env.key_filename = ssh_key_path
|
|
|
|
|
|
@task
|
|
@parallel
|
|
def put_benchmark_utils():
|
|
put('benchmark_utils.py')
|
|
|
|
|
|
@task
|
|
@parallel
|
|
def set_statsd_host(statsd_host='localhost'):
|
|
run('python3 benchmark_utils.py set-statsd-host {}'.format(statsd_host))
|
|
print('update configuration')
|
|
run('bigchaindb show-config')
|
|
|
|
|
|
@task
|
|
@parallel
|
|
def prepare_backlog(num_transactions=10000):
|
|
run('python3 benchmark_utils.py add-backlog {}'.format(num_transactions))
|
|
|
|
|
|
@task
|
|
@parallel
|
|
def start_bigchaindb():
|
|
run('screen -d -m bigchaindb start &', pty=False)
|
|
|
|
|
|
@task
|
|
@parallel
|
|
def kill_bigchaindb():
|
|
run('killall bigchaindb')
|