diff --git a/swh/journal/writer/kafka.py b/swh/journal/writer/kafka.py --- a/swh/journal/writer/kafka.py +++ b/swh/journal/writer/kafka.py @@ -38,11 +38,12 @@ value=value_to_kafka(value), ) - def flush(self): - self.producer.flush() # Need to service the callbacks regularly by calling poll self.producer.poll(0) + def flush(self): + self.producer.flush() + def _get_key(self, object_type, object_): if object_type in ('revision', 'release', 'directory', 'snapshot'): return object_['id']