Module: Plum::StreamUtils

Included in:
Stream
Defined in:
lib/plum/stream_utils.rb

Instance Method Summary collapse

Instance Method Details

#promise(headers) ⇒ Stream

Reserves a stream to server push. Sends PUSH_PROMISE and create new stream.

Parameters:

  • headers (Enumerable<String, String>) —

    The request headers. It must contain all of them: ‘:authority’, ‘:method’, ‘:scheme’ and ‘:path’.

Returns:

  • (Stream) —

    The stream to send push response.



9
10
11
12
13
14
15
 
# File 'lib/plum/stream_utils.rb', line 9

def promise(headers)
  stream = @connection.reserve_stream(weight: self.weight + 1, parent: self)
  encoded = @connection.hpack_encoder.encode(headers)
  frame = Frame.push_promise(id, stream.id, encoded, end_headers: true)
  send frame
  stream
end

#send_data(data = "", end_stream: true) ⇒ Object

Sends DATA frame. If the data is larger than MAX_FRAME_SIZE, DATA frame will be splitted.

Parameters:

  • data (String, IO) (defaults to: "") —

    The data to send.

  • end_stream (Boolean) (defaults to: true) —

    Set END_STREAM flag or not.



31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/plum/stream_utils.rb', line 31

def send_data(data = "", end_stream: true)
  max = @connection.remote_settings[:max_frame_size]
  if data.is_a?(IO)
    until data.eof?
      fragment = data.readpartial(max)
      send Frame.data(id, fragment, end_stream: end_stream && data.eof?)
    end
  else
    send Frame.data(id, data, end_stream: end_stream)
  end
  @state = :half_closed_local if end_stream
end

#send_headers(headers, end_stream:) ⇒ Object

Sends response headers. If the encoded frame is larger than MAX_FRAME_SIZE, the headers will be splitted into HEADERS frame and CONTINUATION frame(s).

Parameters:

  • headers (Enumerable<String, String>) —

    The response headers.

  • end_stream (Boolean) —

    Set END_STREAM flag or not.



20
21
22
23
24
25
26
 
# File 'lib/plum/stream_utils.rb', line 20

def send_headers(headers, end_stream:)
  max = @connection.remote_settings[:max_frame_size]
  encoded = @connection.hpack_encoder.encode(headers)
  frame = Frame.headers(id, encoded, end_headers: true, end_stream: end_stream)
  send frame
  @state = :half_closed_local if end_stream
end