summaryrefslogtreecommitdiff
path: root/oslo_messaging/_drivers/impl_kafka.py
diff options
context:
space:
mode:
authorGuillaume Espanel <guillaume.espanel.ext@ovhcloud.com>2022-07-11 10:58:32 +0200
committerGuillaume Espanel <guillaume.espanel.ext@ovhcloud.com>2022-08-03 17:35:16 +0200
commit43f2224aacb668aa51de3d1274ff8939d8aa73ae (patch)
tree5a941a264e73c1a8b0a3b858a5b337cef875c3ae /oslo_messaging/_drivers/impl_kafka.py
parent41863867489c8a84823f2b3d73cab5585f1fb4ae (diff)
downloadoslo-messaging-43f2224aacb668aa51de3d1274ff8939d8aa73ae.tar.gz
Remove logging from ProducerConnection._produce_message
In impl_kafka, _produce_message is run in a tpool.execute context but it was also calling logging functions. This could cause subsequent calls to logging functions to deadlock. This patch moves the logging calls out of the tpool.execute scope. Change-Id: I81167eea0a6b1a43a88baa3bc383af684f4b1345 Closes-bug: #1981093
Diffstat (limited to 'oslo_messaging/_drivers/impl_kafka.py')
-rw-r--r--oslo_messaging/_drivers/impl_kafka.py42
1 files changed, 27 insertions, 15 deletions
diff --git a/oslo_messaging/_drivers/impl_kafka.py b/oslo_messaging/_drivers/impl_kafka.py
index 6729f87..09abfc5 100644
--- a/oslo_messaging/_drivers/impl_kafka.py
+++ b/oslo_messaging/_drivers/impl_kafka.py
@@ -265,18 +265,17 @@ class ProducerConnection(Connection):
self.producer = None
self.producer_lock = threading.Lock()
- def _produce_message(self, topic, message):
- while True:
- try:
- self.producer.produce(topic, message)
- except KafkaException as e:
- LOG.error("Produce message failed: %s" % str(e))
- except BufferError:
- LOG.debug("Produce message queue full, waiting for deliveries")
- self.producer.poll(0.5)
- continue
- break
-
+ def _produce_message(self, topic, message, poll):
+ if poll:
+ self.producer.poll(poll)
+ try:
+ self.producer.produce(topic, message)
+ except KafkaException as e:
+ self.producer.poll(0)
+ raise e
+ except BufferError as e:
+ # We'll have to poll next time
+ raise e
self.producer.poll(0)
def notify_send(self, topic, ctxt, msg, retry):
@@ -293,9 +292,22 @@ class ProducerConnection(Connection):
try:
self._ensure_producer()
- if eventletutils.is_monkey_patched('thread'):
- return tpool.execute(self._produce_message, topic, message)
- return self._produce_message(topic, message)
+ poll = 0
+ while True:
+ try:
+ if eventletutils.is_monkey_patched('thread'):
+ return tpool.execute(self._produce_message, topic,
+ message, poll)
+ return self._produce_message(topic, message, poll)
+ except KafkaException as e:
+ LOG.error("Produce message failed: %s" % str(e))
+ break
+ except BufferError:
+ LOG.debug("Produce message queue full, "
+ "waiting for deliveries")
+ # We'll retry with .5s polling
+ poll = 0.5
+
except Exception:
# NOTE(sileht): if something goes wrong close the producer
# connection