Add optional name parameter for thread naming

Author: flivni

lib/batch_queue/batch_queue.rb

@@ -1,21 +1,24 @@
 class BatchQueue
+  attr_reader :name
   attr_reader :max_batch_size
   attr_reader :max_interval_seconds
 
   # starts the queue
   # either max_batch_size or interval_milliseconds or both must be set
-  def initialize(max_batch_size: nil, max_interval_seconds: nil, &block)
+  def initialize(name: nil, max_batch_size: nil, max_interval_seconds: nil, &block)
     if max_batch_size.nil? && max_interval_seconds.nil?
       raise 'either max_batch_size or max_interval_seconds or both must be set'
     end
     @is_running = true
     @queue = Queue.new
     @block = block
+    @name = name
     @max_batch_size = max_batch_size
     @max_interval_seconds = max_interval_seconds
     @mutex = Mutex.new
     @cond_var = ConditionVariable.new
     @runner = Thread.new { run }
+    @runner.name = name if !name.nil? && @runner.respond_to?(:name=)
     @on_error_callback = nil
 
     at_exit do

lib/batch_queue/version.rb

@@ -1,3 +1,3 @@
 class BatchQueue
-  VERSION = "1.1.0"
+  VERSION = "1.2.0"
 end

test/batch_queue_test.rb

@@ -80,4 +80,16 @@ def test_it_handles_close
     assert_equal 0, bq.size
     assert_equal 1, processed_count
   end
+
+  def test_it_sets_thread_name
+    thread_name = nil
+    bq = BatchQueue.new(name: 'test-thread', max_batch_size: 1) do |arr|
+      thread_name = Thread.current.name
+    end
+
+    bq << 'test'
+    sleep(0.2)
+    assert_equal 'test-thread', thread_name
+    bq.stop
+  end
 end