Skip to content

Commit 6b08a59

Browse files
author
Carlos Gonzalez Betancort
committed
MB-34324 cbworkloadgen correct estimated messages
Backport progress bar fixes for cbworkloadgen. Change-Id: Ic914a5606cf99e5eea4978c815ab78af9849e7aa Reviewed-on: http://review.couchbase.org/109832 Well-Formed: Build Bot <build@couchbase.com> Reviewed-by: Patrick Varley <patrick@couchbase.com> Tested-by: Carlos Gonzalez <carlos.gonzalez@couchbase.com>
1 parent 0ee8f4c commit 6b08a59

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

pump_gen.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,12 @@ def total_msgs(opts, source_bucket, source_node, source_map):
164164
"""Returns max-items only if exit-after-creates was specified.
165165
Else, total msgs is unknown as GenSource does not stop generating."""
166166
if source_map['cfg']['exit-after-creates'] and source_map['cfg']['ratio-sets'] > 0:
167-
total_ops = source_map['cfg']['max-items'] / source_map['cfg']['ratio-sets']
168-
return 0, int(total_ops)
167+
ratio = source_map['cfg']['ratio-sets']
168+
ops = 0
169+
sets = 0
170+
while sets != source_map['cfg']['max-items']:
171+
if ratio >= float(sets)/float(ops or 1):
172+
sets += 1
173+
ops += 1
174+
return 0, ops
169175
return 0, None

0 commit comments

Comments
 (0)