Mirroristas/bigchaindb
2
0
Fork 0
mirror of https://github.com/bigchaindb/bigchaindb.git synced 2024-10-13 13:34:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added --payload-size option to benchmark_utils.py add-backlog

Browse Source
This commit is contained in:
troymc 2016-05-11 15:38:42 +02:00
parent ced473aaa6
commit dd93e03750
1 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
benchmarking-tests/benchmark_utils.py
View File

@ -14,16 +14,27 @@ from bigchaindb.util import ProcessGroup
from bigchaindb.commands import utils from bigchaindb.commands import utils
SIZE_OF_FILLER = {'minimal': 0,
'small': 10**3,
'medium': 10**4,
'large': 10**5}
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def create_write_transaction(tx_left): def create_write_transaction(tx_left, payload_filler):
b = Bigchain() b = Bigchain()
payload_dict = {}
if payload_filler:
payload_dict['filler'] = payload_filler
while tx_left > 0: while tx_left > 0:
# use uuid to prevent duplicate transactions (transactions with the same hash) # Include a random uuid string in the payload to prevent duplicate
# transactions (i.e. transactions with the same hash)
payload_dict['msg'] = str(uuid.uuid4())
tx = b.create_transaction(b.me, b.me, None, 'CREATE', tx = b.create_transaction(b.me, b.me, None, 'CREATE',
payload={'msg': str(uuid.uuid4())}) payload=payload_dict)
tx_signed = b.sign_transaction(tx, b.me_private) tx_signed = b.sign_transaction(tx, b.me_private)
b.write_transaction(tx_signed) b.write_transaction(tx_signed)
tx_left -= 1 tx_left -= 1
@ -31,7 +42,9 @@ def create_write_transaction(tx_left):
def run_add_backlog(args): def run_add_backlog(args):
tx_left = args.num_transactions // mp.cpu_count() tx_left = args.num_transactions // mp.cpu_count()
workers = ProcessGroup(target=create_write_transaction, args=(tx_left,)) payload_filler = 'x' * SIZE_OF_FILLER[args.payload_size]
workers = ProcessGroup(target=create_write_transaction,
args=(tx_left, payload_filler))
workers.start() workers.start()
@ -102,8 +115,13 @@ def main():
# add transactions to backlog # add transactions to backlog
backlog_parser = subparsers.add_parser('add-backlog', backlog_parser = subparsers.add_parser('add-backlog',
help='Add transactions to the backlog') help='Add transactions to the backlog')
backlog_parser.add_argument('num_transactions', metavar='num_transactions', type=int, default=0, backlog_parser.add_argument('num_transactions', metavar='num_transactions',
type=int, default=0,
help='Number of transactions to add to the backlog') help='Number of transactions to add to the backlog')
backlog_parser.add_argument('-s', '--payload-size',
choices=['minimal', 'small', 'medium', 'large'],
default='minimal',
help='Payload size')
# set statsd host # set statsd host
statsd_parser = subparsers.add_parser('set-statsd-host', statsd_parser = subparsers.add_parser('set-statsd-host',