lib/kafka_session.rb in kafka_session-0.4.0 vs lib/kafka_session.rb in kafka_session-0.4.1

@@ -1,17 +1,28 @@
 require 'kafka_session/producer'
+require 'kafka_session/mock_producer'
 require "kafka_session/version"
 
 require 'kafka_session/delivered_message'
 require 'kafka_session/message'
 
 require 'securerandom'
 require 'active_support/time'
+require 'active_support/core_ext/hash'
 
 class KafkaSession
   class << self
-    attr_writer :clock
-    attr_accessor :producer
+    attr_writer :clock, :producer
+  end
+
+  def self.configure(options = {})
+    producer_options = options.slice(:name, :brokers)
+
+    @producer = Producer.new(producer_options)
+  end
+
+  def self.producer
+    @producer ||= MockProducer.new
   end
 
   def self.session_id
     Thread.current[:kafka_session_session_id] ||= SecureRandom.uuid
   end