bugfix

Signed-off-by: Bing Wang <[email protected]>
fluent · Sep 29, 2024 · 42c6c22 · 42c6c22
1 parent f799f61
commit 42c6c22
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/fluent/plugin/kafka_producer_ext.rb b/lib/fluent/plugin/kafka_producer_ext.rb
@@ -254,9 +254,8 @@ def assign_partitions!
 
       @pending_message_queue.each do |message|
         partition = message.partition
-
+        partition_count = @cluster.partitions_for(message.topic).count
         begin
-          partition_count = @cluster.partitions_for(message.topic).count
 
           if partition.nil?
             partition = @partitioner.call(partition_count, message)

diff --git a/lib/fluent/plugin/out_kafka2.rb b/lib/fluent/plugin/out_kafka2.rb
@@ -403,6 +403,7 @@ def write(chunk)
       rescue Kafka::UnknownTopicOrPartition
         if @use_default_for_unknown_topic && topic != @default_topic
           log.warn "'#{topic}' topic not found. Retry with '#{default_topic}' topic"
+          producer.clear_buffer
           topic = @default_topic
           retry
         end