lib/rdkafka/producer/delivery_handle.rb in rdkafka-0.3.4 vs lib/rdkafka/producer/delivery_handle.rb in rdkafka-0.3.5

@@ -6,9 +6,19 @@
       layout :pending, :bool,
              :response, :int,
              :partition, :int,
              :offset, :int64
 
+      REGISTRY = {}
+
+      def self.register(address, handle)
+        REGISTRY[address] = handle
+      end
+
+      def self.remove(address)
+        REGISTRY.delete(address)
+      end
+
       # Whether the delivery handle is still pending.
       #
       # @return [Boolean]
       def pending?
         self[:pending]