class Segment::Analytics::Worker

Public Class Methods

new(queue, app_id, options = {}) click to toggle source

public: Creates a new worker

The worker continuously takes messages off the queue and makes requests to the segment.io api

queue - Queue synchronized between client and worker app_id - String of the project’s app_id options - Hash of worker options

batch_size - Fixnum of how many items to send in a batch
on_error   - Proc of what to do on an error
# File lib/segment/analytics/worker.rb, line 23
def initialize(queue, app_id, options = {})
  symbolize_keys! options
  @queue = queue
  @app_id = app_id
  @batch_size = options[:batch_size] || Queue::BATCH_SIZE
  @on_error = options[:on_error] || Proc.new { |status, error| }
  @batch = []
  @lock = Mutex.new
end

Public Instance Methods

is_requesting?() click to toggle source

public: Check whether we have outstanding requests.

# File lib/segment/analytics/worker.rb, line 55
def is_requesting?
  @lock.synchronize { !@batch.empty? }
end
run() click to toggle source

public: Continuously runs the loop to check for new events

# File lib/segment/analytics/worker.rb, line 35
def run
  until Thread.current[:should_exit]
    return if @queue.empty?

    @lock.synchronize do
      until @batch.length >= @batch_size || @queue.empty?
        @batch << @queue.pop
      end
    end

    res = Request.new.post @app_id, @batch

    @lock.synchronize { @batch.clear }

    @on_error.call res.status, res.error unless res.status == 200
  end
end