def main(algo, count):
p = Producer()
c = Client()
interval = traffic_frequency
c.describe_topic(topic_name)
c.describe_consumer_group(group_name)
for i in range(interval):
transaction_list = p.read_transaction_list()
count1 = count % len(transaction_list)
transaction = int(transaction_list[count1])
if workload_type == "fix":
transaction = transaction_value
message_count = traffic_ratio * int(transaction/interval)
start_time = time.time()
producer_info = p.producer_per_test(topic_name, message_count)
end_time = time.time()
time_diff = end_time - start_time
print "producers take %s time " % time_diff
p.write_latency(algo, producer_info)
# output = c.describe_consumer_group(group_name)
# print output
duration = int(60/interval)
for i in range(int(duration)):
p.wait_time(1)
inter_end_time = time.time()
if inter_end_time - start_time >= duration:
break
def read_transactions():
message_list = []
p = Producer()
transaction_list = p.read_transaction_list()
for i in range(traffic_interval):
for j in range(traffic_frequency):
transaction = int(transaction_list[i]/traffic_frequency)
message = transaction * traffic_ratio
message_list.append(message)
return message_list
