class Gnip::GnipStream::Stream

Attributes

backfill_client[RW]
url[RW]
version[RW]

Public Class Methods

new(client) click to toggle source
# File lib/gnip/gnip-stream/stream.rb, line 11
def initialize(client)
  self.version = client.power_track_version
  case version
  when '1.0'
    @url = "https://stream.gnip.com:443/accounts/#{client.account}/publishers/#{client.publisher}/streams/track/#{client.label}.json"
  when '2.0'
    @url = "https://gnip-stream.twitter.com/stream/powertrack/accounts/#{client.account}/publishers/#{client.publisher}/#{client.label}.json"
  else
    raise Exception.new("version #{version} is not supported from this gem.")
  end
  @backfill_client = client.backfill_client
  @processor  = JsonDataBuffer.new("\r\n", Regexp.new(/^\{.*\}\r\n/))
  @headers    = { 'authorization' => [client.username, client.password], 'accept-encoding' => 'gzip, compressed' }
  @error_handler = ErrorReconnect.new(self, :consume)
  @connection_close_handler = ErrorReconnect.new(self, :consume)
  configure_handlers
end

Public Instance Methods

configure_handlers() click to toggle source
# File lib/gnip/gnip-stream/stream.rb, line 29
def configure_handlers
  on_error { |error| @error_handler.attempt_to_reconnect("Gnip Connection Error. Reason was: #{error.inspect}") }
  on_connection_close { @connection_close_handler.attempt_to_reconnect('Gnip Connection Closed') }
end
connect() click to toggle source
# File lib/gnip/gnip-stream/stream.rb, line 52
def connect
  EM.run do
    options = {}
    options = { query: { 'client' => backfill_client } } if backfill_client.present?
    http = EM::HttpRequest.new(url, inactivity_timeout: 45, connection_timeout: 75).get({ head: @headers }.merge!(options))
    http.stream { |chunk| process_chunk(chunk) }
    http.callback {
      handle_connection_close(http)
      EM.stop
    }
    http.errback {
      handle_error(http)
      EM.stop
    }
  end
end
consume(&block) click to toggle source
# File lib/gnip/gnip-stream/stream.rb, line 34
def consume(&block)
  @client_callback = block if block
  on_message(&@client_callback)
  connect
end
handle_connection_close(http_connection) click to toggle source
# File lib/gnip/gnip-stream/stream.rb, line 80
def handle_connection_close(http_connection)
  @on_connection_close.call(http_connection)
end
handle_error(http_connection) click to toggle source
# File lib/gnip/gnip-stream/stream.rb, line 76
def handle_error(http_connection)
  @on_error.call(http_connection)
end
on_connection_close(&block) click to toggle source
# File lib/gnip/gnip-stream/stream.rb, line 44
def on_connection_close(&block)
  @on_connection_close = block
end
on_error(&block) click to toggle source
# File lib/gnip/gnip-stream/stream.rb, line 48
def on_error(&block)
  @on_error = block
end
on_message(&block) click to toggle source
# File lib/gnip/gnip-stream/stream.rb, line 40
def on_message(&block)
  @on_message = block
end
process_chunk(chunk) click to toggle source
# File lib/gnip/gnip-stream/stream.rb, line 69
def process_chunk(chunk)
  @processor.process(chunk)
  @processor.complete_entries.each do |entry|
    EM.defer { @on_message.call(entry) }
  end
end