1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
| | # -*- encoding: binary -*-
# Copyright (C) 2016 all contributors <yahns-public@yhbt.net>
# License: GPL-3.0+ <https://www.gnu.org/licenses/gpl-3.0.txt>
# frozen_string_literal: true
require_relative 'wbuf'
# This is only used for "proxy_buffering: false"
class Yahns::WbufLite < Yahns::Wbuf # :nodoc:
attr_reader :busy
attr_writer :req_res
def initialize(req_res)
alive = req_res.alive
super(nil, alive ? :ignore : false)
@req_res = req_res
end
def wbuf_write(client, buf)
super
rescue
@req_res = @req_res.close if @req_res
raise
end
def wbuf_flush(client)
super
rescue
@req_res = @req_res.close if @req_res
raise
end
# called by Yahns::HttpClient#step_write
def wbuf_close(client)
wbuf_abort
# resume the event loop when @blocked is empty
# The actual Yahns::ReqRes#yahns_step is actually read/write-event
# agnostic, and we should actually watch for writability here since
# the req_res socket itself could be completely drained of readable
# data and just waiting for another request (which we don't support, yet)
if @req_res
client.hijack_cleanup
Thread.current[:yahns_queue].queue_mod(@req_res, Yahns::Queue::QEV_WR)
return :ignore
end
@wbuf_persist
rescue
@req_res = @req_res.close if @req_res
raise
end
end
|