kafka producer fix - don't wait indefinately for flush()

edenhill · edenhill · commit 260faebb65bb · 2017-10-19T14:18:24.000+02:00
diff --git a/confluent_kafka/kafkatest/verifiable_producer.py b/confluent_kafka/kafkatest/verifiable_producer.py
@@ -131,10 +131,10 @@ def dr_cb(self, err, msg):
     # Flush remaining messages to broker.
     vp.dbg('Flushing')
     try:
-        vp.producer.flush()
+        vp.producer.flush(5)
     except KeyboardInterrupt:
         pass
 
-    vp.send({'name': 'shutdown_complete'})
+    vp.send({'name': 'shutdown_complete', '_qlen': len(vp.producer)})
 
     vp.dbg('All done')
