Class: HTTPX::Connection::HTTP2

Inherits:
Object
  • Object
show all
Includes:
HTTPX::Callbacks, Loggable
Defined in:
lib/httpx/connection/http2.rb

Defined Under Namespace

Classes: Error, GoawayError, PingError

Constant Summary collapse

MAX_CONCURRENT_REQUESTS = 
::HTTP2::DEFAULT_MAX_CONCURRENT_STREAMS

Constants included from Loggable

Loggable::COLORS, Loggable::USE_DEBUG_LOG

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from Loggable

#log, #log_exception, #log_redact, #log_redact_body, #log_redact_headers

Methods included from HTTPX::Callbacks

#callbacks_for?, #emit, #on, #once

Constructor Details

#initialize(buffer, options) ⇒ HTTP2

Returns a new instance of HTTP2.



35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/httpx/connection/http2.rb', line 35

def initialize(buffer, options)
  @options = options
  @settings = @options.http2_settings
  @pending = []
  @streams = {}
  @drains = {}
  @pings = []
  @buffer = buffer
  @handshake_completed = false
  @wait_for_handshake = @settings.key?(:wait_for_handshake) ? @settings.delete(:wait_for_handshake) : true
  @max_concurrent_requests = @options.max_concurrent_requests || MAX_CONCURRENT_REQUESTS
  @max_requests = @options.max_requests
  init_connection
end

Instance Attribute Details

#pending ⇒ Object (readonly)

Returns the value of attribute pending.



33
34
35
 
# File 'lib/httpx/connection/http2.rb', line 33

def pending
  @pending
end

#streams ⇒ Object (readonly)

Returns the value of attribute streams.



33
34
35
 
# File 'lib/httpx/connection/http2.rb', line 33

def streams
  @streams
end

Instance Method Details

#<<(data) ⇒ Object 



105
106
107
 
# File 'lib/httpx/connection/http2.rb', line 105

def <<(data)
  @connection << data
end

#close ⇒ Object 



89
90
91
92
93
94
95
 
# File 'lib/httpx/connection/http2.rb', line 89

def close
  unless @connection.state == :closed
    @connection.goaway
    emit(:timeout, @options.timeout[:close_handshake_timeout])
  end
  emit(:close)
end

#consume ⇒ Object 



127
128
129
130
131
132
133
 
# File 'lib/httpx/connection/http2.rb', line 127

def consume
  @streams.each do |request, stream|
    next unless request.can_buffer?

    handle(request, stream)
  end
end

#empty? ⇒ Boolean

Returns:

  • (Boolean) 


97
98
99
 
# File 'lib/httpx/connection/http2.rb', line 97

def empty?
  @connection.state == :closed || @streams.empty?
end

#exhausted? ⇒ Boolean

Returns:

  • (Boolean) 


101
102
103
 
# File 'lib/httpx/connection/http2.rb', line 101

def exhausted?
  !@max_requests.positive?
end

#handle_error(ex, request = nil) ⇒ Object 



135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
 
# File 'lib/httpx/connection/http2.rb', line 135

def handle_error(ex, request = nil)
  if ex.is_a?(OperationTimeoutError) && !@handshake_completed && @connection.state != :closed
    @connection.goaway(:settings_timeout, "closing due to settings timeout")
    emit(:close_handshake)
    settings_ex = SettingsTimeoutError.new(ex.timeout, ex.message)
    settings_ex.set_backtrace(ex.backtrace)
    ex = settings_ex
  end
  @streams.each_key do |req|
    next if request && request == req

    emit(:error, req, ex)
  end
  while (req = @pending.shift)
    next if request && request == req

    emit(:error, req, ex)
  end
end

#interests ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
 
# File 'lib/httpx/connection/http2.rb', line 56

def interests
  if @connection.state == :closed
    return unless @handshake_completed

    return if @buffer.empty?

    return :w
  end

  unless @connection.state == :connected && @handshake_completed
    return @buffer.empty? ? :r : :rw
  end

  unless @connection.send_buffer.empty?
    return :rw unless @buffer.empty?

    # waiting for WINDOW_UPDATE frames
    return :r
  end

  return :w if !@pending.empty? && can_buffer_more_requests?

  return :w unless @drains.empty?

  if @buffer.empty?
    return if @streams.empty? && @pings.empty?

    :r
  else
    :w
  end
end

#ping ⇒ Object 



155
156
157
158
159
160
 
# File 'lib/httpx/connection/http2.rb', line 155

def ping
  ping = SecureRandom.gen_random(8)
  @connection.ping(ping.dup)
ensure
  @pings << ping
end

#send(request, head = false) ⇒ Object 



109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
 
# File 'lib/httpx/connection/http2.rb', line 109

def send(request, head = false)
  unless can_buffer_more_requests?
    head ? @pending.unshift(request) : @pending << request
    return false
  end
  unless (stream = @streams[request])
    stream = @connection.new_stream
    handle_stream(stream, request)
    @streams[request] = stream
    @max_requests -= 1
  end
  handle(request, stream)
  true
rescue ::HTTP2::Error::StreamLimitExceeded
  @pending.unshift(request)
  false
end

#timeout ⇒ Object 



50
51
52
53
54
 
# File 'lib/httpx/connection/http2.rb', line 50

def timeout
  return @options.timeout[:operation_timeout] if @handshake_completed

  @options.timeout[:settings_timeout]
end

#waiting_for_ping? ⇒ Boolean

Returns:

  • (Boolean) 


162
163
164
 
# File 'lib/httpx/connection/http2.rb', line 162

def waiting_for_ping?
  @pings.any?
end