Class: Twitter::Streaming::Connection

Inherits:
Object
  • Object
show all
Defined in:
lib/twitter/streaming/connection.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(options = {}) ⇒ Connection

Returns a new instance of Connection.


10
11
12
13
14
15
# File 'lib/twitter/streaming/connection.rb', line 10

def initialize(options = {})
  @tcp_socket_class = options.fetch(:tcp_socket_class) { TCPSocket }
  @ssl_socket_class = options.fetch(:ssl_socket_class) { OpenSSL::SSL::SSLSocket }
  @using_ssl        = options.fetch(:using_ssl)        { false }
  @write_pipe = nil
end

Instance Attribute Details

#ssl_socket_classObject (readonly)

Returns the value of attribute ssl_socket_class


8
9
10
# File 'lib/twitter/streaming/connection.rb', line 8

def ssl_socket_class
  @ssl_socket_class
end

#tcp_socket_classObject (readonly)

Returns the value of attribute tcp_socket_class


8
9
10
# File 'lib/twitter/streaming/connection.rb', line 8

def tcp_socket_class
  @tcp_socket_class
end

Instance Method Details

#closeObject


42
43
44
# File 'lib/twitter/streaming/connection.rb', line 42

def close
  @write_pipe&.write('q')
end

#connect(request) ⇒ Object


33
34
35
36
37
38
39
40
# File 'lib/twitter/streaming/connection.rb', line 33

def connect(request)
  client = new_tcp_socket(request.socket_host, request.socket_port)
  return client if !@using_ssl && request.using_proxy?

  client_context = OpenSSL::SSL::SSLContext.new
  ssl_client     = @ssl_socket_class.new(client, client_context)
  ssl_client.connect
end

#stream(request, response) ⇒ Object

rubocop:disable Metrics/MethodLength


17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# File 'lib/twitter/streaming/connection.rb', line 17

def stream(request, response) # rubocop:disable Metrics/MethodLength
  client = connect(request)
  request.stream(client)
  read_pipe, @write_pipe = IO.pipe
  loop do
    read_ios, _write_ios, _exception_ios = IO.select([read_pipe, client])
    case read_ios.first
    when client
      response << client.readpartial(1024)
    when read_pipe
      break
    end
  end
  client.close
end