diff options
| author | Dana Powers <dana.powers@gmail.com> | 2019-03-30 08:25:40 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2019-03-30 08:25:40 -0700 |
| commit | de6e9d3cc31db2d513e8d8f9dde4d77d400325ce (patch) | |
| tree | 200b51dd099f32e612e970fda3885205066b8839 /benchmarks/producer_performance.py | |
| parent | 227a94663d6b0ab11c12236085f79b5b6ffd5568 (diff) | |
| download | kafka-python-benchmark_fixups.tar.gz | |
Fixups to benchmark scripts for py3 / new KafkaFixture interfacebenchmark_fixups
Diffstat (limited to 'benchmarks/producer_performance.py')
| -rwxr-xr-x | benchmarks/producer_performance.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/benchmarks/producer_performance.py b/benchmarks/producer_performance.py index 0c29cbc..c0de6fd 100755 --- a/benchmarks/producer_performance.py +++ b/benchmarks/producer_performance.py @@ -26,7 +26,7 @@ def start_brokers(n): replicas = min(n, 3) print('-> {0} Brokers [{1} partitions / {2} replicas]'.format(n, partitions, replicas)) brokers = [ - KafkaFixture.instance(i, zk.host, zk.port, zk_chroot='', + KafkaFixture.instance(i, zk, zk_chroot='', partitions=partitions, replicas=replicas) for i in range(n) ] @@ -130,7 +130,7 @@ def get_args_parser(): help='Topic name for test', default='kafka-python-benchmark-test') parser.add_argument( - '--num-records', type=long, + '--num-records', type=int, help='number of messages to produce', default=1000000) parser.add_argument( |