lib/deimos/backends/kafka_async.rb in deimos-ruby-1.0.0.pre.beta25 vs lib/deimos/backends/kafka_async.rb in deimos-ruby-1.0.0.pre.beta26
- old
+ new
@@ -4,9 +4,15 @@
module Backends
# Backend which produces to Kafka via an async producer.
class KafkaAsync < Deimos::PublishBackend
include Phobos::Producer
+ # Shut down the producer cleanly.
+ def self.shutdown_producer
+ producer.async_producer_shutdown
+ producer.kafka_client&.close
+ end
+
# :nodoc:
def self.execute(producer_class:, messages:)
Deimos.instrument(
'produce',
producer: producer_class,