import zmq
import random
import time
context = zmq.Context()
sender = context.socket(zmq.PUSH)
sender.bind("tcp://*:5557")
sink = context.socket(zmq.PUSH)
sink.connect("tcp://localhost:5558")
print("Press Enter when the workers are ready: ")
_ = input()
print("Sending tasks to workers…")
sink.send_string('0')
random.seed()
total_msec = 0
for task_nbr in range(100):
workload = random.randint(1, 100)
total_msec += workload
sender.send_string(str(workload))
print("Total expected cost: %s msec" % total_msec)
time.sleep(1)