lib/kafka_session.rb in kafka_session-0.1.0 vs lib/kafka_session.rb in kafka_session-0.2.0

@@ -5,15 +5,19 @@
 require 'securerandom'
 require 'active_support/time'
 
 class KafkaSession
   class << self
-    attr_writer :session_id, :clock
+    attr_writer :clock
     attr_accessor :producer
   end
 
   def self.session_id
-    @session_id ||= SecureRandom.uuid
+    Thread.current[:kafka_session_session_id] ||= SecureRandom.uuid
+  end
+
+  def self.session_id=(session_id)
+    Thread.current[:kafka_session_session_id] = session_id
   end
 
   def self.clock
     @clock ||= Time.public_method(:current)
   end